[fix](spill) SpillStream's writer maybe may not have been finalized (#…

…32931)
apache · Mar 28, 2024 · ceef19a · ceef19a
1 parent b5dc1e4
commit ceef19a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/be/src/pipeline/exec/partitioned_aggregation_sink_operator.cpp b/be/src/pipeline/exec/partitioned_aggregation_sink_operator.cpp
@@ -170,6 +170,9 @@ Status PartitionedAggSinkOperatorX::sink(doris::RuntimeState* state, vectorized:
         if (revocable_mem_size(state) > 0) {
             RETURN_IF_ERROR(revoke_memory(state));
         } else {
+            for (auto& partition : local_state._shared_state->spill_partitions) {
+                RETURN_IF_ERROR(partition->finish_current_spilling(eos));
+            }
             local_state._dependency->set_ready_to_read();
         }
     }