This is an automated email from the ASF dual-hosted git repository.

gabriellee pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/doris.git


The following commit(s) were added to refs/heads/master by this push:
     new c04e5bac392 [bug](pipelineX) fix java-udaf failed with open pipelineX 
(#24939)
c04e5bac392 is described below

commit c04e5bac39203906f2ee8df15cc8074be4d3c7a1
Author: zhangstar333 <[email protected]>
AuthorDate: Wed Sep 27 13:14:10 2023 +0800

    [bug](pipelineX) fix java-udaf failed with open pipelineX (#24939)
---
 be/src/vec/exprs/vectorized_agg_fn.cpp | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/be/src/vec/exprs/vectorized_agg_fn.cpp 
b/be/src/vec/exprs/vectorized_agg_fn.cpp
index 06d44cfbc57..642b37cccc4 100644
--- a/be/src/vec/exprs/vectorized_agg_fn.cpp
+++ b/be/src/vec/exprs/vectorized_agg_fn.cpp
@@ -335,6 +335,21 @@ AggFnEvaluator::AggFnEvaluator(AggFnEvaluator& evaluator, 
RuntimeState* state)
           _function(evaluator._function),
           _expr_name(evaluator._expr_name),
           _agg_columns(evaluator._agg_columns) {
+    if (evaluator._fn.binary_type == TFunctionBinaryType::JAVA_UDF) {
+        DataTypes tmp_argument_types;
+        tmp_argument_types.reserve(evaluator._input_exprs_ctxs.size());
+        // prepare for argument
+        for (int i = 0; i < evaluator._input_exprs_ctxs.size(); ++i) {
+            auto data_type = 
evaluator._input_exprs_ctxs[i]->root()->data_type();
+            tmp_argument_types.emplace_back(data_type);
+        }
+        const DataTypes& argument_types =
+                _real_argument_types.empty() ? tmp_argument_types : 
_real_argument_types;
+        _function = AggregateJavaUdaf::create(evaluator._fn, argument_types, 
evaluator._data_type);
+        
static_cast<AggregateJavaUdaf*>(_function.get())->check_udaf(evaluator._fn);
+    }
+    DCHECK(_function != nullptr);
+
     _input_exprs_ctxs.resize(evaluator._input_exprs_ctxs.size());
     for (size_t i = 0; i < _input_exprs_ctxs.size(); i++) {
         WARN_IF_ERROR(evaluator._input_exprs_ctxs[i]->clone(state, 
_input_exprs_ctxs[i]), "");


---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]

Reply via email to