[bugfix](nereids) prune partition bug in pattern ColA <> ColB #25769 (#…

…26224)
apache · Nov 1, 2023 · 81b5f74 · 81b5f74
1 parent a35de81
commit 81b5f74
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/...a/org/apache/doris/nereids/rules/expression/rules/TryEliminateUninterestedPredicates.java b/...a/org/apache/doris/nereids/rules/expression/rules/TryEliminateUninterestedPredicates.java
@@ -20,6 +20,7 @@
 import org.apache.doris.nereids.CascadesContext;
 import org.apache.doris.nereids.rules.expression.ExpressionRewriteContext;
 import org.apache.doris.nereids.rules.expression.rules.TryEliminateUninterestedPredicates.Context;
+import org.apache.doris.nereids.trees.expressions.And;
 import org.apache.doris.nereids.trees.expressions.Expression;
 import org.apache.doris.nereids.trees.expressions.Slot;
 import org.apache.doris.nereids.trees.expressions.literal.BooleanLiteral;
@@ -104,6 +105,27 @@ public Expression visit(Expression originExpr, Context parentContext) {
         return expr;
     }
 
+    @Override
+    public Expression visitAnd(And and, Context parentContext) {
+        Expression left = and.left();
+        Context leftContext = new Context();
+        Expression newLeft = this.visit(left, leftContext);
+        if (leftContext.childrenContainsNonInterestedSlots) {
+            newLeft = BooleanLiteral.TRUE;
+        }
+
+        Expression right = and.right();
+        Context rightContext = new Context();
+        Expression newRight = this.visit(right, rightContext);
+        if (rightContext.childrenContainsNonInterestedSlots) {
+            newRight = BooleanLiteral.TRUE;
+        }
+        Expression expr = new And(newLeft, newRight).accept(FoldConstantRuleOnFE.INSTANCE, expressionRewriteContext);
+        parentContext.childrenContainsInterestedSlots =
+                rightContext.childrenContainsInterestedSlots || leftContext.childrenContainsInterestedSlots;
+        return expr;
+    }
+
     @Override
     public Expression visitSlot(Slot slot, Context context) {
         boolean isInterestedSlot = interestedSlots.contains(slot);