Fix multilocale infinite loop

Bears-R-Us · Jun 11, 2024 · 290f1e3 · 290f1e3
1 parent 9857968
commit 290f1e3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/ArrowFunctions.cpp b/src/ArrowFunctions.cpp
@@ -834,7 +834,7 @@ int cpp_readColumnByName(const char* filename, void* chpl_arr, bool* where_null_
         int totalProcessed = 0;
         std::vector<parquet::ByteArray> values(batchSize);
         std::vector<int16_t> definition_levels(batchSize);
-        while (reader->HasNext()) {
+        while (reader->HasNext() && totalProcessed < numElems) {
           if((numElems - totalProcessed) < batchSize) // adjust batchSize if needed
               batchSize = numElems - totalProcessed;