update the allocation length

mapleFU · Nov 23, 2023 · 51e6fb7 · 51e6fb7
1 parent 5f12876
commit 51e6fb7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cpp/src/parquet/encoding.cc b/cpp/src/parquet/encoding.cc
@@ -1921,7 +1921,7 @@ class DictByteArrayDecoderImpl : public DictDecoderImpl<ByteArrayType>,
     constexpr int32_t kBufferSize = 1024;
     int32_t indices[kBufferSize];
 
-    ArrowBinaryHelper<ByteArrayType> helper(out, num_values);
+    ArrowBinaryHelper<ByteArrayType> helper(out, std::min(num_values, this->num_values_));
     // The `len_` in the ByteArrayDictDecoder is the total length of the
     // RLE/Bit-pack encoded data size, so, we cannot use `len_` to reserve
     // space for binary data.
@@ -1992,7 +1992,7 @@ class DictByteArrayDecoderImpl : public DictDecoderImpl<ByteArrayType>,
     int32_t indices[kBufferSize];
     int values_decoded = 0;
 
-    ArrowBinaryHelper<ByteArrayType> helper(out, num_values);
+    ArrowBinaryHelper<ByteArrayType> helper(out, std::min(num_values, this->num_values_));
     // The `len_` in the ByteArrayDictDecoder is the total length of the
     // RLE/Bit-pack encoded data size, so, we cannot use `len_` to reserve
     // space for binary data.