fix PushDownFilterThroughSetOperation

feiniaofeiafei · Jan 3, 2025 · 22e4162 · 22e4162
1 parent 073fd68
commit 22e4162
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/...c/main/java/org/apache/doris/nereids/rules/rewrite/PushDownFilterThroughSetOperation.java b/...c/main/java/org/apache/doris/nereids/rules/rewrite/PushDownFilterThroughSetOperation.java
@@ -44,16 +44,16 @@ public Rule build() {
         return logicalFilter(logicalSetOperation()).then(f -> {
             LogicalSetOperation setOperation = f.child();
             List<Plan> newChildren = new ArrayList<>();
-            for (Plan child : setOperation.children()) {
+            for (int childIdx = 0; childIdx < setOperation.children().size(); ++childIdx) {
                 Map<Expression, Expression> replaceMap = new HashMap<>();
                 for (int i = 0; i < setOperation.getOutputs().size(); ++i) {
                     NamedExpression output = setOperation.getOutputs().get(i);
-                    replaceMap.put(output, child.getOutput().get(i));
+                    replaceMap.put(output, setOperation.getRegularChildOutput(childIdx).get(i));
                 }
 
                 Set<Expression> newFilterPredicates = f.getConjuncts().stream().map(conjunct ->
                         ExpressionUtils.replace(conjunct, replaceMap)).collect(ImmutableSet.toImmutableSet());
-                newChildren.add(new LogicalFilter<>(newFilterPredicates, child));
+                newChildren.add(new LogicalFilter<>(newFilterPredicates, setOperation.child(childIdx)));
             }
 
             return setOperation.withChildren(newChildren);