Fix the SortBuffer's noMoreInput called twice

facebookincubator · Aug 6, 2024 · 5735f21 · 5735f21
1 parent c0fa8f2
commit 5735f21
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/velox/exec/MergeJoin.cpp b/velox/exec/MergeJoin.cpp
@@ -662,7 +662,13 @@ RowVectorPtr MergeJoin::getOutput() {
         // No rows survived the filter. Get more rows.
         continue;
       } else if (isAntiJoin(joinType_)) {
-        return filterOutputForAntiJoin(output);
+        auto result = filterOutputForAntiJoin(output);
+        if (!result) {
+          // Make sure needsInput() method return true. Otherwise, Driver
+          // assumes the operator is finished.
+          input_ = nullptr;
+        }
+        return result;
       } else {
         return output;
       }