From 1a1bb75c9d2551f580c7e65de1e0b9014e59d987 Mon Sep 17 00:00:00 2001 From: Liang-Chi Hsieh Date: Tue, 4 Mar 2025 00:37:11 -0800 Subject: [PATCH] MINOR: Add explicit exception when no more buffer can be read when loading buffers --- .../src/main/java/org/apache/arrow/vector/VectorLoader.java | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/vector/src/main/java/org/apache/arrow/vector/VectorLoader.java b/vector/src/main/java/org/apache/arrow/vector/VectorLoader.java index ecd3fb9124..9b9a890346 100644 --- a/vector/src/main/java/org/apache/arrow/vector/VectorLoader.java +++ b/vector/src/main/java/org/apache/arrow/vector/VectorLoader.java @@ -122,6 +122,10 @@ private void loadBuffers( (int) (variadicBufferLayoutCount + TypeLayout.getTypeBufferCount(field.getType())); List ownBuffers = new ArrayList<>(bufferLayoutCount); for (int j = 0; j < bufferLayoutCount; j++) { + if (!buffers.hasNext()) { + throw new IllegalArgumentException( + "no more buffers for field " + field + ". Expected " + bufferLayoutCount); + } ArrowBuf nextBuf = buffers.next(); // for vectors without nulls, the buffer is empty, so there is no need to decompress it. ArrowBuf bufferToAdd =