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

prestodb · Oct 7, 2024 · 6f60355 · 6f60355
1 parent 88c03ab
commit 6f60355
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/...cebook/presto/parquet/batchreader/decoders/delta/BinaryLongDecimalDeltaValuesDecoder.java b/...cebook/presto/parquet/batchreader/decoders/delta/BinaryLongDecimalDeltaValuesDecoder.java
@@ -49,7 +49,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/...cebook/presto/parquet/batchreader/decoders/plain/BinaryLongDecimalPlainValuesDecoder.java b/...cebook/presto/parquet/batchreader/decoders/plain/BinaryLongDecimalPlainValuesDecoder.java
@@ -45,7 +45,7 @@ public void readNext(long[] values, int offset, int length)
         PlainValueBuffer valueBuffer = (PlainValueBuffer) 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/...cebook/presto/parquet/batchreader/decoders/rle/LongDecimalRLEDictionaryValuesDecoder.java b/...cebook/presto/parquet/batchreader/decoders/rle/LongDecimalRLEDictionaryValuesDecoder.java
@@ -46,7 +46,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++) {