[BugFix] support SqlToScalarOperatorTranslator visit lambda functions more than one times (backport #19843)

fe/fe-core/src/main/java/com/starrocks/sql/ast/LambdaArgument.java

@@ -8,12 +8,23 @@
 import com.starrocks.common.AnalysisException;
 import com.starrocks.sql.common.ErrorType;
 import com.starrocks.sql.common.StarRocksPlannerException;
+import com.starrocks.sql.optimizer.operator.scalar.ColumnRefOperator;
 import com.starrocks.thrift.TExprNode;
 
 public class LambdaArgument extends Expr {
     private String name;
     private boolean nullable;
 
+    ColumnRefOperator transformedOp = null;
+
+    public ColumnRefOperator getTransformed() {
+        return transformedOp;
+    }
+
+    public void setTransformed(ColumnRefOperator op) {
+        transformedOp = op;
+    }
+
     public LambdaArgument(String name) {
         this.name = name;
     }
@@ -22,7 +33,7 @@ public LambdaArgument(LambdaArgument rhs) {
         super(rhs);
         name = rhs.getName();
     }
-
+    
     public String getName() {
         return name;
     }

fe/fe-core/src/main/java/com/starrocks/sql/ast/LambdaFunctionExpr.java

@@ -7,15 +7,18 @@
 import com.starrocks.analysis.Expr;
 import com.starrocks.analysis.SlotRef;
 import com.starrocks.common.AnalysisException;
+<<<<<<< HEAD
 import com.starrocks.sql.optimizer.operator.scalar.LambdaFunctionOperator;
+=======
+import com.starrocks.sql.parser.NodePosition;
+>>>>>>> 863c40ca1 ([BugFix] support SqlToScalarOperatorTranslator visit lambda functions more than one times (#19843))
 import com.starrocks.thrift.TExprNode;
 import com.starrocks.thrift.TExprNodeType;
 
 import java.util.List;
 import java.util.Map;
 
 public class LambdaFunctionExpr extends Expr {
-    private LambdaFunctionOperator transformedOp = null;
     private int commonSubOperatorNum = 0;
 
     public LambdaFunctionExpr(List<Expr> arguments) {
@@ -37,14 +40,6 @@ public LambdaFunctionExpr(LambdaFunctionExpr rhs) {
         super(rhs);
     }
 
-    public LambdaFunctionOperator getTransformed() {
-        return transformedOp;
-    }
-
-    public void setTransformed(LambdaFunctionOperator op) {
-        transformedOp = op;
-    }
-
     @Override
     protected void analyzeImpl(Analyzer analyzer) throws AnalysisException {
         Preconditions.checkState(false, "unreachable");

fe/fe-core/src/main/java/com/starrocks/sql/optimizer/transformer/SqlToScalarOperatorTranslator.java

@@ -325,10 +325,6 @@ public ScalarOperator visitArrowExpr(ArrowExpr node, Context context) {
         @Override
         public ScalarOperator visitLambdaFunctionExpr(LambdaFunctionExpr node,
                                                       Context context) {
-            // To avoid the ids of lambda arguments are different after each visit()
-            if (node.getTransformed() != null) {
-                return node.getTransformed();
-            }
             Preconditions.checkArgument(node.getChildren().size() >= 2);
             List<ColumnRefOperator> refs = Lists.newArrayList();
             List<LambdaArgument> args = Lists.newArrayList();
@@ -341,13 +337,16 @@ public ScalarOperator visitLambdaFunctionExpr(LambdaFunctionExpr node,
             expressionMapping = new ExpressionMapping(scope, refs, expressionMapping);
             ScalarOperator lambda = visit(node.getChild(0), context.clone(node));
             expressionMapping = old; // recover it
-            node.setTransformed(new LambdaFunctionOperator(refs, lambda, Type.FUNCTION));
-            return node.getTransformed();
+            return new LambdaFunctionOperator(refs, lambda, Type.FUNCTION);
         }
 
         @Override
         public ScalarOperator visitLambdaArguments(LambdaArgument node, Context context) {
-            return columnRefFactory.create(node.getName(), node.getType(), node.isNullable(), true);
+            // To avoid the ids of lambda arguments are different after each visit()
+            if (node.getTransformed() == null) {
+                node.setTransformed(columnRefFactory.create(node.getName(), node.getType(), node.isNullable(), true));
+            }
+            return node.getTransformed();
         }
 
         @Override

fe/fe-core/src/test/java/com/starrocks/sql/analyzer/AnalyzeExprTest.java

@@ -114,6 +114,7 @@ public void testLambdaFunction() {
         analyzeSuccess("select array_map(x-> x +  count(v1) over (partition by v1 order by v2),[111]) from tarray");
         analyzeSuccess("select v1, v2, count(v1) over (partition by v1 order by v2) from tarray");
         analyzeSuccess("select v1, v2, count(v1) over (partition by array_sum(array_map(x->x+1, [1])) order by v2) from tarray");
+        analyzeSuccess("with x2 as (select array_map((ss) -> ss * v1, v3) from tarray) select * from x2;");
 
         analyzeFail("select array_map(x,y -> x + y, [], [])"); // should be (x,y)
         analyzeFail("select array_map((x,y,z) -> x + y, [], [])");