Examples with RedirectionAwareTableHandle io.trino.metadata.RedirectionAwareTableHandle used on opensource projects

Example 6 with RedirectionAwareTableHandle

use of io.trino.metadata.RedirectionAwareTableHandle in project trino by trinodb.

the class CommentTask method execute.

@Override
public ListenableFuture<Void> execute(Comment statement, QueryStateMachine stateMachine, List<Expression> parameters, WarningCollector warningCollector) {
    Session session = stateMachine.getSession();
    if (statement.getType() == Comment.Type.TABLE) {
        QualifiedObjectName originalTableName = createQualifiedObjectName(session, statement, statement.getName());
        RedirectionAwareTableHandle redirectionAwareTableHandle = metadata.getRedirectionAwareTableHandle(session, originalTableName);
        if (redirectionAwareTableHandle.getTableHandle().isEmpty()) {
            throw semanticException(TABLE_NOT_FOUND, statement, "Table does not exist: %s", originalTableName);
        }
        accessControl.checkCanSetTableComment(session.toSecurityContext(), redirectionAwareTableHandle.getRedirectedTableName().orElse(originalTableName));
        TableHandle tableHandle = redirectionAwareTableHandle.getTableHandle().get();
        metadata.setTableComment(session, tableHandle, statement.getComment());
    } else if (statement.getType() == Comment.Type.COLUMN) {
        Optional<QualifiedName> prefix = statement.getName().getPrefix();
        if (prefix.isEmpty()) {
            throw semanticException(MISSING_TABLE, statement, "Table must be specified");
        }
        QualifiedObjectName originalTableName = createQualifiedObjectName(session, statement, prefix.get());
        RedirectionAwareTableHandle redirectionAwareTableHandle = metadata.getRedirectionAwareTableHandle(session, originalTableName);
        if (redirectionAwareTableHandle.getTableHandle().isEmpty()) {
            throw semanticException(TABLE_NOT_FOUND, statement, "Table does not exist: " + originalTableName);
        }
        TableHandle tableHandle = redirectionAwareTableHandle.getTableHandle().get();
        String columnName = statement.getName().getSuffix();
        Map<String, ColumnHandle> columnHandles = metadata.getColumnHandles(session, tableHandle);
        if (!columnHandles.containsKey(columnName)) {
            throw semanticException(COLUMN_NOT_FOUND, statement, "Column does not exist: " + columnName);
        }
        accessControl.checkCanSetColumnComment(session.toSecurityContext(), redirectionAwareTableHandle.getRedirectedTableName().orElse(originalTableName));
        metadata.setColumnComment(session, tableHandle, columnHandles.get(columnName), statement.getComment());
    } else {
        throw semanticException(NOT_SUPPORTED, statement, "Unsupported comment type: %s", statement.getType());
    }
    return immediateVoidFuture();
}