[vectorized](function) support array_map function (#17581)

be/src/vec/exprs/vlambda_function_call_expr.h

@@ -0,0 +1,86 @@
+// Licensed to the Apache Software Foundation (ASF) under one
+// or more contributor license agreements.  See the NOTICE file
+// distributed with this work for additional information
+// regarding copyright ownership.  The ASF licenses this file
+// to you under the Apache License, Version 2.0 (the
+// "License"); you may not use this file except in compliance
+// with the License.  You may obtain a copy of the License at
+//
+//   http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing,
+// software distributed under the License is distributed on an
+// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+// KIND, either express or implied.  See the License for the
+// specific language governing permissions and limitations
+// under the License.
+
+#pragma once
+
+#include "common/status.h"
+#include "fmt/format.h"
+#include "fmt/ranges.h"
+#include "vec/exprs/lambda_function/lambda_function.h"
+#include "vec/exprs/lambda_function/lambda_function_factory.h"
+#include "vec/exprs/vexpr.h"
+#include "vec/exprs/vlambda_function_expr.h"
+
+namespace doris::vectorized {
+
+class VLambdaFunctionCallExpr : public VExpr {
+public:
+    VLambdaFunctionCallExpr(const TExprNode& node) : VExpr(node) {}
+    ~VLambdaFunctionCallExpr() override = default;
+
+    VExpr* clone(ObjectPool* pool) const override {
+        return pool->add(new VLambdaFunctionCallExpr(*this));
+    }
+
+    doris::Status prepare(doris::RuntimeState* state, const doris::RowDescriptor& desc,
+                          VExprContext* context) override {
+        RETURN_IF_ERROR_OR_PREPARED(VExpr::prepare(state, desc, context));
+
+        std::vector<std::string_view> child_expr_name;
+        for (auto child : _children) {
+            child_expr_name.emplace_back(child->expr_name());
+        }
+        _expr_name = fmt::format("{}({})", _fn.name.function_name, child_expr_name);
+
+        _lambda_function = LambdaFunctionFactory::instance().get_function(_fn.name.function_name);
+        if (_lambda_function == nullptr) {
+            return Status::InternalError("Lambda Function {} is not implemented.",
+                                         _fn.name.function_name);
+        }
+        return Status::OK();
+    }
+
+    const std::string& expr_name() const override { return _expr_name; }
+
+    Status execute(VExprContext* context, doris::vectorized::Block* block,
+                   int* result_column_id) override {
+        return _lambda_function->execute(context, block, result_column_id, _data_type, _children);
+    }
+
+    std::string debug_string() const override {
+        std::stringstream out;
+        out << "VLambdaFunctionCallExpr[";
+        out << _expr_name;
+        out << "]{";
+        bool first = true;
+        for (VExpr* input_expr : children()) {
+            if (first) {
+                first = false;
+            } else {
+                out << ",";
+            }
+            out << "\n" << input_expr->debug_string();
+        }
+        out << "}";
+        return out.str();
+    }
+
+private:
+    std::string _expr_name;
+    LambdaFunctionPtr _lambda_function;
+};
+} // namespace doris::vectorized