Fix the length of offsets array in parquet short decimal value decoders

unix280 · Nov 4, 2024 · 991791d · 991791d
1 parent 6cfa327
commit 991791d
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/...ebook/presto/parquet/batchreader/decoders/delta/BinaryShortDecimalDeltaValuesDecoder.java b/...ebook/presto/parquet/batchreader/decoders/delta/BinaryShortDecimalDeltaValuesDecoder.java
@@ -47,7 +47,7 @@ public void readNext(long[] values, int offset, int length)
         BinaryValuesDecoder.ValueBuffer valueBuffer = delegate.readNext(length);
         int bufferSize = valueBuffer.getBufferSize();
         byte[] byteBuffer = new byte[bufferSize];
-        int[] offsets = new int[bufferSize + 1];
+        int[] offsets = new int[length + 1];
         delegate.readIntoBuffer(byteBuffer, 0, offsets, 0, valueBuffer);
 
         for (int i = 0; i < length; i++) {

diff --git a/...ebook/presto/parquet/batchreader/decoders/plain/BinaryShortDecimalPlainValuesDecoder.java b/...ebook/presto/parquet/batchreader/decoders/plain/BinaryShortDecimalPlainValuesDecoder.java
@@ -41,7 +41,7 @@ public void readNext(long[] values, int offset, int length)
         ValueBuffer valueBuffer = delegate.readNext(length);
         int bufferSize = valueBuffer.getBufferSize();
         byte[] byteBuffer = new byte[bufferSize];
-        int[] offsets = new int[bufferSize + 1];
+        int[] offsets = new int[length + 1];
         delegate.readIntoBuffer(byteBuffer, 0, offsets, 0, valueBuffer);
 
         for (int i = 0; i < length; i++) {

diff --git a/...ebook/presto/parquet/batchreader/decoders/rle/ShortDecimalRLEDictionaryValuesDecoder.java b/...ebook/presto/parquet/batchreader/decoders/rle/ShortDecimalRLEDictionaryValuesDecoder.java
@@ -44,7 +44,7 @@ public void readNext(long[] values, int offset, int length)
         ValueBuffer valueBuffer = delegate.readNext(length);
         int bufferSize = valueBuffer.getBufferSize();
         byte[] byteBuffer = new byte[bufferSize];
-        int[] offsets = new int[bufferSize + 1];
+        int[] offsets = new int[length + 1];
         delegate.readIntoBuffer(byteBuffer, 0, offsets, 0, valueBuffer);
 
         for (int i = 0; i < length; i++) {