From ffdb1388a28a1d2320a37916245337825b439ecb Mon Sep 17 00:00:00 2001
From: "xvyang.xy" <xvyang.xy@antfin.com>
Date: Thu, 30 Mar 2023 15:25:32 +0800
Subject: [PATCH] fix: multi arrow batch

---
 ceresdb-protocol/src/main/java/io/ceresdb/util/Utils.java | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/ceresdb-protocol/src/main/java/io/ceresdb/util/Utils.java b/ceresdb-protocol/src/main/java/io/ceresdb/util/Utils.java
index d95fba5..3bc304d 100644
--- a/ceresdb-protocol/src/main/java/io/ceresdb/util/Utils.java
+++ b/ceresdb-protocol/src/main/java/io/ceresdb/util/Utils.java
@@ -12,6 +12,7 @@
 import java.util.Collection;
 import java.util.Collections;
 import java.util.Iterator;
+import java.util.LinkedList;
 import java.util.List;
 import java.util.Map;
 import java.util.Properties;
@@ -455,10 +456,11 @@ private static List<Row> parseArrowBatch(ByteString batch, Storage.ArrowPayload.
             VectorSchemaRoot readRoot = arrowStreamReader.getVectorSchemaRoot();
             Schema schema = readRoot.getSchema();
 
+            List<Row> batchRows = new LinkedList<>();
             while (arrowStreamReader.loadNextBatch()) {
-                return parseArrowRecord(schema, readRoot);
+                batchRows.addAll(parseArrowRecord(schema, readRoot));
             }
-            return Collections.EMPTY_LIST;
+            return batchRows;
         } catch (IOException e) {
             return Collections.EMPTY_LIST;
         }