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

[feature](function) support initcap string function (#13193)

Browse Source 
support `initcap` string function
This commit is contained in:
luozenglin
2022-10-13 21:31:44 +08:00
committed by GitHub
parent cb300b0b39
commit de4315c1c5
10 changed files with 165 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
be/src/vec/functions/function_string.cpp
View File

@ -265,6 +265,42 @@ struct TransferImpl {
}
};
// Capitalize first letter
struct NameToInitcap {
static constexpr auto name = "initcap";
};
struct InitcapImpl {
static Status vector(const ColumnString::Chars& data, const ColumnString::Offsets& offsets,
ColumnString::Chars& res_data, ColumnString::Offsets& res_offsets) {
size_t offset_size = offsets.size();
res_offsets.resize(offsets.size());
memcpy(res_offsets.data(), offsets.data(), offset_size * sizeof(offsets.data()));
size_t data_length = data.size();
res_data.resize(data_length);
simd::VStringFunctions::to_lower(data.data(), data_length, res_data.data());
bool need_capitalize = true;
for (size_t offset_index = 0, start_index = 0; offset_index < offset_size; ++offset_index) {
auto end_index = res_offsets[offset_index];
need_capitalize = true;
for (size_t i = start_index; i < end_index; ++i) {
if (!::isalnum(res_data[i])) {
need_capitalize = true;
} else if (need_capitalize) {
res_data[i] = ::toupper(res_data[i]);
need_capitalize = false;
}
}
start_index = end_index;
}
return Status::OK();
}
};
struct NameTrim {
static constexpr auto name = "trim";
};
@ -588,6 +624,8 @@ using FunctionToLower = FunctionStringToString<TransferImpl<::tolower>, NameToLo
using FunctionToUpper = FunctionStringToString<TransferImpl<::toupper>, NameToUpper>;
using FunctionToInitcap = FunctionStringToString<InitcapImpl, NameToInitcap>;
using FunctionLTrim = FunctionStringToString<TrimImpl<true, false>, NameLTrim>;
using FunctionRTrim = FunctionStringToString<TrimImpl<false, true>, NameRTrim>;
@ -619,6 +657,7 @@ void register_function_string(SimpleFunctionFactory& factory) {
factory.register_function<FunctionUnHex>();
factory.register_function<FunctionToLower>();
factory.register_function<FunctionToUpper>();
factory.register_function<FunctionToInitcap>();
factory.register_function<FunctionLTrim>();
factory.register_function<FunctionRTrim>();
factory.register_function<FunctionTrim>();