Erase the previous processed window partition when begin handle the …

…next partition
facebookincubator · Sep 25, 2024 · 4e2217e · 4e2217e
1 parent 4e45bc5
commit 4e2217e
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/velox/exec/RowsStreamingWindowBuild.cpp b/velox/exec/RowsStreamingWindowBuild.cpp
@@ -102,6 +102,15 @@ void RowsStreamingWindowBuild::noMoreInput() {
 
 std::shared_ptr<WindowPartition> RowsStreamingWindowBuild::nextPartition() {
   VELOX_CHECK(hasNextPartition());
+
+  if (outputPartition_ > 0) {
+    windowPartitions_.erase(
+        windowPartitions_.begin(), windowPartitions_.begin() + 1);
+    windowPartitions_.shrink_to_fit();
+    --outputPartition_;
+    --inputPartition_;
+  }
+
   return windowPartitions_[++outputPartition_];
 }