Examples with ColumnConstraint com.facebook.presto.sql.planner.planPrinter.IOPlanPrinter.ColumnConstraint used on opensource projects

Example 1 with ColumnConstraint

use of com.facebook.presto.sql.planner.planPrinter.IOPlanPrinter.ColumnConstraint in project presto by prestodb.

the class TestHiveIntegrationSmokeTest method testIOExplain.

@Test
public void testIOExplain() {
    // Test IO explain with small number of discrete components.
    computeActual("CREATE TABLE test_orders WITH (partitioned_by = ARRAY['orderkey', 'processing']) AS SELECT custkey, orderkey, orderstatus = 'P' processing FROM orders WHERE orderkey < 3");
    MaterializedResult result = computeActual("EXPLAIN (TYPE IO, FORMAT JSON) INSERT INTO test_orders SELECT custkey, orderkey, processing FROM test_orders where custkey <= 10");
    ImmutableSet.Builder<ColumnConstraint> expectedConstraints = ImmutableSet.builder();
    expectedConstraints.add(new ColumnConstraint("orderkey", BIGINT.getTypeSignature(), new FormattedDomain(false, ImmutableSet.of(new FormattedRange(new FormattedMarker(Optional.of("1"), EXACTLY), new FormattedMarker(Optional.of("1"), EXACTLY)), new FormattedRange(new FormattedMarker(Optional.of("2"), EXACTLY), new FormattedMarker(Optional.of("2"), EXACTLY))))));
    expectedConstraints.add(new ColumnConstraint("processing", BOOLEAN.getTypeSignature(), new FormattedDomain(false, ImmutableSet.of(new FormattedRange(new FormattedMarker(Optional.of("false"), EXACTLY), new FormattedMarker(Optional.of("false"), EXACTLY))))));
    assertEquals(jsonCodec(IOPlan.class).fromJson((String) getOnlyElement(result.getOnlyColumnAsSet())), new IOPlan(ImmutableSet.of(new TableColumnInfo(new CatalogSchemaTableName(catalog, "tpch", "test_orders"), expectedConstraints.build())), Optional.of(new CatalogSchemaTableName(catalog, "tpch", "test_orders"))));
    assertUpdate("DROP TABLE test_orders");
    // Test IO explain with large number of discrete components where Domain::simplify comes into play.
    computeActual("CREATE TABLE test_orders WITH (partitioned_by = ARRAY['orderkey']) AS select custkey, orderkey FROM orders where orderkey < 200");
    expectedConstraints = ImmutableSet.builder();
    expectedConstraints.add(new ColumnConstraint("orderkey", BIGINT.getTypeSignature(), new FormattedDomain(false, ImmutableSet.of(new FormattedRange(new FormattedMarker(Optional.of("1"), EXACTLY), new FormattedMarker(Optional.of("199"), EXACTLY))))));
    result = computeActual("EXPLAIN (TYPE IO, FORMAT JSON) INSERT INTO test_orders SELECT custkey, orderkey + 10 FROM test_orders where custkey <= 10");
    assertEquals(jsonCodec(IOPlan.class).fromJson((String) getOnlyElement(result.getOnlyColumnAsSet())), new IOPlan(ImmutableSet.of(new TableColumnInfo(new CatalogSchemaTableName(catalog, "tpch", "test_orders"), expectedConstraints.build())), Optional.of(new CatalogSchemaTableName(catalog, "tpch", "test_orders"))));
    assertUpdate("DROP TABLE test_orders");
}

Example 2 with ColumnConstraint

use of com.facebook.presto.sql.planner.planPrinter.IOPlanPrinter.ColumnConstraint in project presto by prestodb.

the class TestLocalQueries method testIOExplain.

@Test
public void testIOExplain() {
    String query = "SELECT * FROM orders";
    MaterializedResult result = computeActual("EXPLAIN (TYPE IO, FORMAT JSON) " + query);
    TableColumnInfo input = new TableColumnInfo(new CatalogSchemaTableName("local", "sf0.01", "orders"), ImmutableSet.of(new ColumnConstraint("orderstatus", createVarcharType(1).getTypeSignature(), new FormattedDomain(false, ImmutableSet.of(new FormattedRange(new FormattedMarker(Optional.of("F"), EXACTLY), new FormattedMarker(Optional.of("F"), EXACTLY)), new FormattedRange(new FormattedMarker(Optional.of("O"), EXACTLY), new FormattedMarker(Optional.of("O"), EXACTLY)), new FormattedRange(new FormattedMarker(Optional.of("P"), EXACTLY), new FormattedMarker(Optional.of("P"), EXACTLY)))))));
    assertEquals(jsonCodec(IOPlan.class).fromJson((String) getOnlyElement(result.getOnlyColumnAsSet())), new IOPlan(ImmutableSet.of(input), Optional.empty()));
}