Fix reading of plain dictionary with zero length

aloneguid · Feb 23, 2021 · 9dda79f · 9dda79f
1 parent 8273012
commit 9dda79f
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/Parquet/File/DataColumnReader.cs b/src/Parquet/File/DataColumnReader.cs
@@ -265,8 +265,10 @@ private static int ReadPlainDictionary(BinaryReader reader, int maxReadCount, in
          int start = offset;
          int bitWidth = reader.ReadByte();
 
+         int length = GetRemainingLength(reader);
+
          //when bit width is zero reader must stop and just repeat zero maxValue number of times
-         if (bitWidth == 0)
+         if (bitWidth == 0 || length == 0)
          {
             for (int i = 0; i < maxReadCount; i++)
             {
@@ -275,7 +277,6 @@ private static int ReadPlainDictionary(BinaryReader reader, int maxReadCount, in
          }
          else
          {
-            int length = GetRemainingLength(reader);
             offset += RunLengthBitPackingHybridValuesReader.ReadRleBitpackedHybrid(reader, bitWidth, length, dest, offset, maxReadCount);
          }