Fix phased execution for merge join

visitMergeJoin function was missing from PhasedExecutionSchedule which makes it throw exceptions when doing phased execution schedule
prestodb · May 16, 2022 · 47b1fe3 · 47b1fe3
1 parent 497e3d5
commit 47b1fe3
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/...o-main/src/main/java/com/facebook/presto/execution/scheduler/PhasedExecutionSchedule.java b/...o-main/src/main/java/com/facebook/presto/execution/scheduler/PhasedExecutionSchedule.java
@@ -22,6 +22,7 @@
 import com.facebook.presto.sql.planner.plan.IndexJoinNode;
 import com.facebook.presto.sql.planner.plan.InternalPlanVisitor;
 import com.facebook.presto.sql.planner.plan.JoinNode;
+import com.facebook.presto.sql.planner.plan.MergeJoinNode;
 import com.facebook.presto.sql.planner.plan.PlanFragmentId;
 import com.facebook.presto.sql.planner.plan.RemoteSourceNode;
 import com.facebook.presto.sql.planner.plan.SemiJoinNode;
@@ -226,6 +227,12 @@ public Set<PlanFragmentId> visitIndexJoin(IndexJoinNode node, PlanFragmentId cur
             return processJoin(node.getIndexSource(), node.getProbeSource(), currentFragmentId);
         }
 
+        @Override
+        public Set<PlanFragmentId> visitMergeJoin(MergeJoinNode node, PlanFragmentId currentFragmentId)
+        {
+            return processJoin(node.getRight(), node.getLeft(), currentFragmentId);
+        }
+
         private Set<PlanFragmentId> processJoin(PlanNode build, PlanNode probe, PlanFragmentId currentFragmentId)
         {
             Set<PlanFragmentId> buildSources = build.accept(this, currentFragmentId);