[tmp] the jump count of NLJ is 2 for runtime filter prune on external…

… db (#29367)
apache · Jan 1, 2024 · 895f1d4 · 895f1d4
1 parent 8a836be
commit 895f1d4
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/...ain/java/org/apache/doris/nereids/processor/post/RuntimeFilterPrunerForExternalTable.java b/...ain/java/org/apache/doris/nereids/processor/post/RuntimeFilterPrunerForExternalTable.java
@@ -24,6 +24,7 @@
 import org.apache.doris.nereids.trees.plans.physical.AbstractPhysicalJoin;
 import org.apache.doris.nereids.trees.plans.physical.PhysicalFileScan;
 import org.apache.doris.nereids.trees.plans.physical.PhysicalHashJoin;
+import org.apache.doris.nereids.trees.plans.physical.PhysicalNestedLoopJoin;
 import org.apache.doris.nereids.trees.plans.physical.PhysicalRelation;
 import org.apache.doris.nereids.trees.plans.physical.RuntimeFilter;
 import org.apache.doris.nereids.util.MutableState;
@@ -104,6 +105,20 @@ public PhysicalHashJoin visitPhysicalHashJoin(PhysicalHashJoin<? extends Plan, ?
         return join;
     }
 
+    @Override
+    public PhysicalNestedLoopJoin visitPhysicalNestedLoopJoin(
+            PhysicalNestedLoopJoin<? extends Plan, ? extends Plan> join,
+            CascadesContext context) {
+        join.right().accept(this, context);
+        join.right().setMutableState(MutableState.KEY_PARENT, join);
+        // nested loop join is slow, so jump add 2
+        join.setMutableState(MutableState.KEY_RF_JUMP,
+            (Integer) join.right().getMutableState(MutableState.KEY_RF_JUMP).get() + 2);
+        join.left().accept(this, context);
+        join.left().setMutableState(MutableState.KEY_PARENT, join);
+        return join;
+    }
+
     private List<Plan> getAncestors(Plan plan) {
         List<Plan> ancestors = Lists.newArrayList();
         ancestors.add(plan);