Merge pull request apache#49 from rongma1997/modify-vectorloader

Modify VectorLoader to support native compression
jikunshang · May 18, 2020 · 6a278e7 · 6a278e7
2 parents 7a60602 + 3fad4b6
commit 6a278e7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/java/vector/src/main/java/org/apache/arrow/vector/VectorLoader.java b/java/vector/src/main/java/org/apache/arrow/vector/VectorLoader.java
@@ -53,12 +53,12 @@ public VectorLoader(VectorSchemaRoot root) {
    * @param recordBatch the batch to load
    */
   public void load(ArrowRecordBatch recordBatch) {
+    root.setRowCount(recordBatch.getLength());
     Iterator<ArrowBuf> buffers = recordBatch.getBuffers().iterator();
     Iterator<ArrowFieldNode> nodes = recordBatch.getNodes().iterator();
     for (FieldVector fieldVector : root.getFieldVectors()) {
       loadBuffers(fieldVector, fieldVector.getField(), buffers, nodes);
     }
-    root.setRowCount(recordBatch.getLength());
     if (nodes.hasNext() || buffers.hasNext()) {
       throw new IllegalArgumentException("not all nodes and buffers were consumed. nodes: " +
           Collections2.toList(nodes).toString() + " buffers: " + Collections2.toList(buffers).toString());