Add missing openForRead in LocalPartitionWriter

apache · Nov 28, 2024 · 4d8cee7 · 4d8cee7
1 parent 7bcaa67
commit 4d8cee7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/cpp/core/shuffle/LocalPartitionWriter.cc b/cpp/core/shuffle/LocalPartitionWriter.cc
@@ -457,6 +457,7 @@ arrow::Status LocalPartitionWriter::stop(ShuffleWriterMetrics* metrics) {
       }
     }
 
+    spill->openForRead(options_.shuffleFileBufferSize);
     for (auto pid = 0; pid < numPartitions_; ++pid) {
       while (auto payload = spill->nextPayload(pid)) {
         partitionLengths_[pid] += payload->rawSize();