database/doris
2
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

[Vectorized][UDF] support java-udaf (#9930)

Browse Source
This commit is contained in:
zhangstar333
2022-06-15 10:53:44 +08:00
committed by GitHub
parent f4e2f78a1a
commit 4c24586865
15 changed files with 1415 additions and 178 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
be/src/vec/functions/function_java_udf.cpp
View File

@ -160,7 +160,7 @@ Status JavaFunctionCall::execute(FunctionContext* context, Block& block,
ColumnString::Offsets& offsets = \
const_cast<ColumnString::Offsets&>(str_col->get_offsets()); \
int increase_buffer_size = 0; \
int32_t buffer_size = JavaFunctionCall::IncreaseReservedBufferSize(increase_buffer_size); \
int32_t buffer_size = JniUtil::IncreaseReservedBufferSize(increase_buffer_size); \
chars.reserve(buffer_size); \
chars.resize(buffer_size); \
offsets.reserve(num_rows); \
@ -173,8 +173,7 @@ Status JavaFunctionCall::execute(FunctionContext* context, Block& block,
nullptr); \
while (jni_ctx->output_intermediate_state_ptr->row_idx < num_rows) { \
increase_buffer_size++; \
int32_t buffer_size = \
JavaFunctionCall::IncreaseReservedBufferSize(increase_buffer_size); \
int32_t buffer_size = JniUtil::IncreaseReservedBufferSize(increase_buffer_size); \
chars.resize(buffer_size); \
*(jni_ctx->output_value_buffer) = reinterpret_cast<int64_t>(chars.data()); \
jni_ctx->output_intermediate_state_ptr->buffer_size = buffer_size; \