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]