Examples with ColumnDefinition io.trino.sql.tree.ColumnDefinition used on opensource projects

Example 6 with ColumnDefinition

use of io.trino.sql.tree.ColumnDefinition in project trino by trinodb.

the class TestCreateTableTask method testCreateTableNotExistsTrue.

@Test
public void testCreateTableNotExistsTrue() {
    CreateTable statement = new CreateTable(QualifiedName.of("test_table"), ImmutableList.of(new ColumnDefinition(identifier("a"), toSqlType(BIGINT), true, emptyList(), Optional.empty())), true, ImmutableList.of(), Optional.empty());
    CreateTableTask createTableTask = new CreateTableTask(plannerContext, new AllowAllAccessControl(), columnPropertyManager, tablePropertyManager);
    getFutureValue(createTableTask.internalExecute(statement, testSession, emptyList(), output -> {
    }));
    assertEquals(metadata.getCreateTableCallCount(), 1);
}

Example 7 with ColumnDefinition

use of io.trino.sql.tree.ColumnDefinition in project trino by trinodb.

the class TestCreateTableTask method testCreateWithUnsupportedConnectorThrowsWhenNotNull.

@Test
public void testCreateWithUnsupportedConnectorThrowsWhenNotNull() {
    List<TableElement> inputColumns = ImmutableList.of(new ColumnDefinition(identifier("a"), toSqlType(DATE), true, emptyList(), Optional.empty()), new ColumnDefinition(identifier("b"), toSqlType(VARCHAR), false, emptyList(), Optional.empty()), new ColumnDefinition(identifier("c"), toSqlType(VARBINARY), false, emptyList(), Optional.empty()));
    CreateTable statement = new CreateTable(QualifiedName.of("test_table"), inputColumns, true, ImmutableList.of(), Optional.empty());
    CreateTableTask createTableTask = new CreateTableTask(plannerContext, new AllowAllAccessControl(), columnPropertyManager, tablePropertyManager);
    assertTrinoExceptionThrownBy(() -> getFutureValue(createTableTask.internalExecute(statement, testSession, emptyList(), output -> {
    }))).hasErrorCode(NOT_SUPPORTED).hasMessage("Catalog 'catalog' does not support non-null column for column name 'b'");
}

Example 8 with ColumnDefinition

use of io.trino.sql.tree.ColumnDefinition in project trino by trinodb.

the class AddColumnTask method execute.

@Override
public ListenableFuture<Void> execute(AddColumn statement, QueryStateMachine stateMachine, List<Expression> parameters, WarningCollector warningCollector) {
    Session session = stateMachine.getSession();
    QualifiedObjectName originalTableName = createQualifiedObjectName(session, statement, statement.getName());
    RedirectionAwareTableHandle redirectionAwareTableHandle = plannerContext.getMetadata().getRedirectionAwareTableHandle(session, originalTableName);
    if (redirectionAwareTableHandle.getTableHandle().isEmpty()) {
        if (!statement.isTableExists()) {
            throw semanticException(TABLE_NOT_FOUND, statement, "Table '%s' does not exist", originalTableName);
        }
        return immediateVoidFuture();
    }
    TableHandle tableHandle = redirectionAwareTableHandle.getTableHandle().get();
    CatalogName catalogName = getRequiredCatalogHandle(plannerContext.getMetadata(), session, statement, tableHandle.getCatalogName().getCatalogName());
    accessControl.checkCanAddColumns(session.toSecurityContext(), redirectionAwareTableHandle.getRedirectedTableName().orElse(originalTableName));
    Map<String, ColumnHandle> columnHandles = plannerContext.getMetadata().getColumnHandles(session, tableHandle);
    ColumnDefinition element = statement.getColumn();
    Type type;
    try {
        type = plannerContext.getTypeManager().getType(toTypeSignature(element.getType()));
    } catch (TypeNotFoundException e) {
        throw semanticException(TYPE_NOT_FOUND, element, "Unknown type '%s' for column '%s'", element.getType(), element.getName());
    }
    if (type.equals(UNKNOWN)) {
        throw semanticException(COLUMN_TYPE_UNKNOWN, element, "Unknown type '%s' for column '%s'", element.getType(), element.getName());
    }
    if (columnHandles.containsKey(element.getName().getValue().toLowerCase(ENGLISH))) {
        if (!statement.isColumnNotExists()) {
            throw semanticException(COLUMN_ALREADY_EXISTS, statement, "Column '%s' already exists", element.getName());
        }
        return immediateVoidFuture();
    }
    if (!element.isNullable() && !plannerContext.getMetadata().getConnectorCapabilities(session, catalogName).contains(NOT_NULL_COLUMN_CONSTRAINT)) {
        throw semanticException(NOT_SUPPORTED, element, "Catalog '%s' does not support NOT NULL for column '%s'", catalogName.getCatalogName(), element.getName());
    }
    Map<String, Object> columnProperties = columnPropertyManager.getProperties(catalogName, element.getProperties(), session, plannerContext, accessControl, parameterExtractor(statement, parameters), true);
    ColumnMetadata column = ColumnMetadata.builder().setName(element.getName().getValue()).setType(type).setNullable(element.isNullable()).setComment(element.getComment()).setProperties(columnProperties).build();
    plannerContext.getMetadata().addColumn(session, tableHandle, column);
    return immediateVoidFuture();
}

Example 9 with ColumnDefinition

use of io.trino.sql.tree.ColumnDefinition in project trino by trinodb.

the class TestSqlParser method testAddColumn.

@Test
public void testAddColumn() {
    assertThat(statement("ALTER TABLE foo.t ADD COLUMN c bigint")).ignoringLocation().isEqualTo(new AddColumn(QualifiedName.of("foo", "t"), new ColumnDefinition(identifier("c"), simpleType(location(1, 31), "bigint"), true, emptyList(), Optional.empty()), false, false));
    assertThat(statement("ALTER TABLE foo.t ADD COLUMN d double NOT NULL")).ignoringLocation().isEqualTo(new AddColumn(QualifiedName.of("foo", "t"), new ColumnDefinition(identifier("d"), simpleType(location(1, 31), "double"), false, emptyList(), Optional.empty()), false, false));
    assertThat(statement("ALTER TABLE IF EXISTS foo.t ADD COLUMN d double NOT NULL")).ignoringLocation().isEqualTo(new AddColumn(QualifiedName.of("foo", "t"), new ColumnDefinition(identifier("d"), simpleType(location(1, 31), "double"), false, emptyList(), Optional.empty()), true, false));
    assertThat(statement("ALTER TABLE foo.t ADD COLUMN IF NOT EXISTS d double NOT NULL")).ignoringLocation().isEqualTo(new AddColumn(QualifiedName.of("foo", "t"), new ColumnDefinition(identifier("d"), simpleType(location(1, 31), "double"), false, emptyList(), Optional.empty()), false, true));
    assertThat(statement("ALTER TABLE IF EXISTS foo.t ADD COLUMN IF NOT EXISTS d double NOT NULL")).ignoringLocation().isEqualTo(new AddColumn(QualifiedName.of("foo", "t"), new ColumnDefinition(identifier("d"), simpleType(location(1, 31), "double"), false, emptyList(), Optional.empty()), true, true));
}