Init vectors once

Bears-R-Us · May 21, 2024 · 19153b0 · 19153b0
1 parent f65b440
commit 19153b0
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/ArrowFunctions.cpp b/src/ArrowFunctions.cpp
@@ -846,12 +846,11 @@ int cpp_readColumnByName(const char* filename, void* chpl_arr, bool* where_null_
           static_cast<parquet::ByteArrayReader*>(column_reader.get());
 
         int totalProcessed = 0;
-
+        std::vector<parquet::ByteArray> values(batchSize);
+        std::vector<int16_t> definition_levels(batchSize);
         while (reader->HasNext()) {
           if((numElems - totalProcessed) < batchSize) // adjust batchSize if needed
               batchSize = numElems - totalProcessed;
-          std::vector<parquet::ByteArray> values(batchSize);
-          std::vector<int16_t> definition_levels(batchSize);
 
           (void)reader->ReadBatch(batchSize, definition_levels.data(), nullptr, values.data(), &values_read);
           totalProcessed += values_read;