fix

englefly · Sep 11, 2024 · d1b8292 · d1b8292
1 parent 2e4942b
commit d1b8292
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 6 deletions.
diff --git a/.../main/java/org/apache/doris/nereids/processor/post/ProjectAggregateExpressionsForCse.java b/.../main/java/org/apache/doris/nereids/processor/post/ProjectAggregateExpressionsForCse.java
@@ -23,6 +23,7 @@
 import org.apache.doris.nereids.trees.expressions.Alias;
 import org.apache.doris.nereids.trees.expressions.Expression;
 import org.apache.doris.nereids.trees.expressions.NamedExpression;
+import org.apache.doris.nereids.trees.expressions.OrderExpression;
 import org.apache.doris.nereids.trees.expressions.Slot;
 import org.apache.doris.nereids.trees.expressions.SlotReference;
 import org.apache.doris.nereids.trees.expressions.functions.agg.AggregateFunction;
@@ -130,7 +131,7 @@ public Plan visitPhysicalHashAggregate(PhysicalHashAggregate<? extends Plan> agg
     private void getCseCandidatesFromAggregateFunction(Expression expr, Map<Expression, Alias> result) {
         if (expr instanceof AggregateFunction) {
             for (Expression child : expr.children()) {
-                if (!(child instanceof SlotReference) && !child.isConstant()) {
+                if (!(child instanceof SlotReference) && !child.isConstant() && !(child instanceof OrderExpression)) {
                     if (child instanceof Alias) {
                         result.put(child, (Alias) child);
                     } else {

diff --git a/regression-test/data/nereids_tpch_p0/tpch/agg_cse.out b/regression-test/data/nereids_tpch_p0/tpch/agg_cse.out
@@ -1,10 +1,10 @@
 -- This file is automatically generated. You should know what you did if you want to edit this
--- !select --
-2	2.0	3
-8	8.0	9
+-- !1 --
 0	0.0	1
+2	2.0	3
 4	4.0	5
 6	6.0	7
+8	8.0	9
 
 -- !agg_cse_subquery --
 0	0.0	1

diff --git a/regression-test/suites/nereids_tpch_p0/tpch/agg_cse.groovy b/regression-test/suites/nereids_tpch_p0/tpch/agg_cse.groovy
@@ -23,9 +23,9 @@ suite("agg_cse") {
     sql 'set enable_nereids_planner=true'
     sql 'set enable_fallback_to_original_planner=false'
 
-    qt_select """
+    qt_1 """
     select sum(r_regionkey + r_regionkey), avg(r_regionkey + r_regionkey), sum(r_regionkey + (r_regionkey+1)) 
-    from region group by r_name;
+    from region group by r_name order by 1, 2, 3;
     """
     explain{
         sql """