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

[fix] (udf) fix check_fn and fn_call function name not same (#8132)

Browse Source
This commit is contained in:
Zhengguo Yang
2022-02-22 09:18:07 +08:00
committed by GitHub
parent c3348b8023
commit c47368f80c
5 changed files with 57 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
be/src/vec/functions/function_rpc.cpp
View File

@ -290,6 +290,24 @@ void convert_col_to_pvalue(const ColumnPtr& column, const DataTypePtr& data_type
}
break;
}
case TypeIndex::HLL: {
ptype->set_id(PGenericType::HLL);
arg->mutable_bytes_value()->Reserve(row_count);
for (size_t row_num = 0; row_num < row_count; ++row_num) {
if constexpr (nullable) {
if (column->is_null_at(row_num)) {
arg->add_bytes_value(nullptr);
} else {
StringRef data = column->get_data_at(row_num);
arg->add_bytes_value(data.data, data.size);
}
} else {
StringRef data = column->get_data_at(row_num);
arg->add_bytes_value(data.data, data.size);
}
}
break;
}
default:
LOG(INFO) << "unknown type: " << data_type->get_name();
ptype->set_id(PGenericType::UNKNOWN);
@ -469,6 +487,13 @@ void convert_to_column(MutableColumnPtr& column, const PValues& result) {
}
break;
}
case PGenericType::HLL: {
column->reserve(result.bytes_value_size());
for (int i = 0; i < result.bytes_value_size(); ++i) {
column->insert_data(result.bytes_value(i).c_str(), result.bytes_value(i).size());
}
break;
}
default: {
LOG(WARNING) << "unknown PGenericType: " << result.type().DebugString();
break;
@ -517,6 +542,10 @@ Status RPCFnCall::execute(FunctionContext* context, Block& block, const ColumnNu
fmt::format("call to rpc function {} failed: {}", _symbol, cntl.ErrorText())
.c_str());
}
if (!response.has_status() || !response.has_result()) {
return Status::InternalError(
fmt::format("call rpc function {} failed: status or result is not set.", _symbol));
}
if (response.status().status_code() != 0) {
return Status::InternalError(fmt::format("call to rpc function {} failed: {}", _symbol,
response.status().DebugString()));