DRILL-7937: INTERNAL_ERROR when querying Parquet File with decimals

logicdrop · Jun 4, 2021 · f86a7cb · f86a7cb
1 parent 36ac8f3
commit f86a7cb
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/...va/org/apache/drill/exec/store/parquet/columnreaders/NullableFixedByteAlignedReaders.java b/...va/org/apache/drill/exec/store/parquet/columnreaders/NullableFixedByteAlignedReaders.java
@@ -321,6 +321,8 @@ protected void readField(long recordsToReadInThisPass) {
         case FIXED_LEN_BYTE_ARRAY:
         case BINARY:
           if (usingDictionary) {
+            recordsReadInThisIteration = Math.min(pageReader.currentPageCount
+                - pageReader.valuesRead, recordsToReadInThisPass - valuesReadInCurrentPass);
             NullableVarDecimalVector.Mutator mutator = valueVec.getMutator();
             for (int i = 0; i < recordsReadInThisIteration; i++) {
               Binary currDictValToWrite = pageReader.dictionaryValueReader.readBytes();