Search in sources :

Example 11 with FilterExpression

use of org.apache.drill.metastore.expressions.FilterExpression in project drill by apache.

the class TestBasicTablesRequestsRequestMetadata method testRequestMetadataWithInFilter.

@Test
public void testRequestMetadataWithInFilter() {
    List<String> locations = Arrays.asList("/tmp/dir0", "/tmp/dir1");
    List<String> metadataKeys = Arrays.asList("a", "b", "c");
    BasicTablesRequests.RequestMetadata requestMetadata = BasicTablesRequests.RequestMetadata.builder().locations(locations).metadataKeys(metadataKeys).build();
    FilterExpression expected = FilterExpression.and(FilterExpression.in(MetastoreColumn.LOCATION, locations), FilterExpression.in(MetastoreColumn.METADATA_KEY, metadataKeys));
    assertEquals(expected.toString(), requestMetadata.filter().toString());
}
Also used : FilterExpression(org.apache.drill.metastore.expressions.FilterExpression) Test(org.junit.Test) BaseTest(org.apache.drill.test.BaseTest) MetastoreTest(org.apache.drill.categories.MetastoreTest)

Example 12 with FilterExpression

use of org.apache.drill.metastore.expressions.FilterExpression in project drill by apache.

the class TestTablesOperationTransformer method testToDeleteOperation.

@Test
public void testToDeleteOperation() {
    Bson expected = Filters.and(Filters.eq(MetastoreColumn.STORAGE_PLUGIN.columnName(), "dfs"), Filters.eq(MetastoreColumn.WORKSPACE.columnName(), "tmp"));
    FilterExpression filter = FilterExpression.and(FilterExpression.equal(MetastoreColumn.STORAGE_PLUGIN, "dfs"), FilterExpression.equal(MetastoreColumn.WORKSPACE, "tmp"));
    Delete delete = Delete.builder().metadataType(MetadataType.ALL).filter(filter).build();
    MongoDelete operation = transformer.toDelete(delete);
    assertEquals(expected.toString(), operation.filter().toString());
}
Also used : MongoDelete(org.apache.drill.metastore.mongo.operate.MongoDelete) Delete(org.apache.drill.metastore.operate.Delete) FilterExpression(org.apache.drill.metastore.expressions.FilterExpression) MongoDelete(org.apache.drill.metastore.mongo.operate.MongoDelete) Bson(org.bson.conversions.Bson) Test(org.junit.Test)

Example 13 with FilterExpression

use of org.apache.drill.metastore.expressions.FilterExpression in project drill by apache.

the class TestTablesMetadataMapper method testToCondition.

@Test
public void testToCondition() {
    FilterExpression filterExpression = FilterExpression.equal(MetastoreColumn.STORAGE_PLUGIN, "dfs");
    Condition tablesCondition = TablesMetadataMapper.TableMapper.get().toCondition(filterExpression);
    assertEquals(Tables.TABLES.STORAGE_PLUGIN.eq("dfs"), tablesCondition);
    Condition segmentsCondition = TablesMetadataMapper.SegmentMapper.get().toCondition(filterExpression);
    assertEquals(Tables.SEGMENTS.STORAGE_PLUGIN.eq("dfs"), segmentsCondition);
    Condition filesCondition = TablesMetadataMapper.FileMapper.get().toCondition(filterExpression);
    assertEquals(Tables.FILES.STORAGE_PLUGIN.eq("dfs"), filesCondition);
    Condition rowGroupsCondition = TablesMetadataMapper.RowGroupMapper.get().toCondition(filterExpression);
    assertEquals(Tables.ROW_GROUPS.STORAGE_PLUGIN.eq("dfs"), rowGroupsCondition);
    Condition partitionsCondition = TablesMetadataMapper.PartitionMapper.get().toCondition(filterExpression);
    assertEquals(Tables.PARTITIONS.STORAGE_PLUGIN.eq("dfs"), partitionsCondition);
}
Also used : Condition(org.jooq.Condition) FilterExpression(org.apache.drill.metastore.expressions.FilterExpression) Test(org.junit.Test) RdbmsBaseTest(org.apache.drill.metastore.rdbms.RdbmsBaseTest)

Aggregations

FilterExpression (org.apache.drill.metastore.expressions.FilterExpression)13 Test (org.junit.Test)11 MetastoreTest (org.apache.drill.categories.MetastoreTest)6 BaseTest (org.apache.drill.test.BaseTest)6 IcebergBaseTest (org.apache.drill.metastore.iceberg.IcebergBaseTest)3 TableInfo (org.apache.drill.metastore.metadata.TableInfo)2 Expression (org.apache.iceberg.expressions.Expression)2 MetadataControllerContext (org.apache.drill.exec.metastore.analyze.MetadataControllerContext)1 MetastoreTableInfo (org.apache.drill.metastore.components.tables.MetastoreTableInfo)1 TableMetadataUnit (org.apache.drill.metastore.components.tables.TableMetadataUnit)1 Delete (org.apache.drill.metastore.iceberg.operate.Delete)1 MetadataInfo (org.apache.drill.metastore.metadata.MetadataInfo)1 MongoDelete (org.apache.drill.metastore.mongo.operate.MongoDelete)1 Delete (org.apache.drill.metastore.operate.Delete)1 RdbmsBaseTest (org.apache.drill.metastore.rdbms.RdbmsBaseTest)1 Bson (org.bson.conversions.Bson)1 Condition (org.jooq.Condition)1