test

apache · Aug 11, 2024 · 5574bf5 · 5574bf5
1 parent aac200a
commit 5574bf5
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/common/src/main/java/org/apache/comet/parquet/BatchReader.java b/common/src/main/java/org/apache/comet/parquet/BatchReader.java
@@ -530,6 +530,15 @@ public void close() throws IOException {
       importer.close();
       importer = null;
     }
+    // As our native reader reuses the same memory buffer for all columns, we only
+    // need to release the buffers once when closing the reader.
+    if (currentBatch != null) {
+      for (int i = 0; i < currentBatch.numCols(); i++) {
+        if (currentBatch.column(i) != null) {
+          currentBatch.column(i).close();
+        }
+      }
+    }
   }
 
   @SuppressWarnings("deprecation")