arrow: use Arrow::RecordBatchStreamReader#read_all

For empty record case.
groonga · Feb 6, 2024 · 5f68507 · 5f68507
1 parent ff37133
commit 5f68507
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/lib/grntest/test-runner.rb b/lib/grntest/test-runner.rb
@@ -714,9 +714,8 @@ def normalize_apache_arrow_content(content, options)
       Arrow::BufferInputStream.open(buffer) do |input|
         while input.tell < content.bytesize
           reader = Arrow::RecordBatchStreamReader.new(input)
-          schema = reader.schema
-          record_batches = reader.to_a
-          table = Arrow::Table.new(schema, record_batches)
+          table = reader.read_all
+          schema = table.schema
           unless normalized.empty?
             normalized << "=" * 40
             normalized << "\n"