From 32b5f69b9edc605e7df23cda8da069ba67da2353 Mon Sep 17 00:00:00 2001 From: Przemyslaw Maciolek Date: Tue, 31 May 2016 20:46:53 +0200 Subject: [PATCH] DRILL-4139: Add missing BIT support for Paruet partition pruning --- .../apache/drill/exec/store/parquet/ParquetGroupScan.java | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/exec/java-exec/src/main/java/org/apache/drill/exec/store/parquet/ParquetGroupScan.java b/exec/java-exec/src/main/java/org/apache/drill/exec/store/parquet/ParquetGroupScan.java index 5950b74caf1..2889e5e1153 100644 --- a/exec/java-exec/src/main/java/org/apache/drill/exec/store/parquet/ParquetGroupScan.java +++ b/exec/java-exec/src/main/java/org/apache/drill/exec/store/parquet/ParquetGroupScan.java @@ -61,6 +61,7 @@ import org.apache.drill.exec.store.schedule.EndpointByteMap; import org.apache.drill.exec.store.schedule.EndpointByteMapImpl; import org.apache.drill.exec.util.ImpersonationUtil; +import org.apache.drill.exec.vector.NullableBitVector; import org.apache.drill.exec.vector.NullableBigIntVector; import org.apache.drill.exec.vector.NullableDateVector; import org.apache.drill.exec.vector.NullableDecimal18Vector; @@ -386,6 +387,12 @@ public void populatePruningVector(ValueVector v, int index, SchemaPath column, S String f = Path.getPathWithoutSchemeAndAuthority(new Path(file)).toString(); MinorType type = getTypeForColumn(column).getMinorType(); switch (type) { + case BIT: { + NullableBitVector bitVector = (NullableBitVector) v; + Boolean value = (Boolean) partitionValueMap.get(f).get(column); + bitVector.getMutator().setSafe(index, value ? 1 : 0); + return; + } case INT: { NullableIntVector intVector = (NullableIntVector) v; Integer value = (Integer) partitionValueMap.get(f).get(column);