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

[feature-wip](array-type) add the array_join function (#11406)

Browse Source 
this pr is used to add the array_join function.
Co-authored-by: hucheng01 <hucheng01@baidu.com>
This commit is contained in:
carlvinhust2012
2022-08-15 11:43:17 +08:00
committed by GitHub
parent 77e241cbb0
commit 8c8f48c4c2
8 changed files with 512 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
be/src/vec/functions/array/function_array_aggregation.cpp
View File

@ -31,6 +31,7 @@
#include "vec/core/types.h"
#include "vec/data_types/data_type.h"
#include "vec/data_types/data_type_nullable.h"
#include "vec/functions/array/function_array_join.h"
#include "vec/functions/array/function_array_mapped.h"
#include "vec/functions/simple_function_factory.h"
@ -139,14 +140,20 @@ struct ArrayAggregateImpl {
using column_type = ColumnArray;
using data_type = DataTypeArray;
static DataTypePtr get_return_type(const DataTypeArray* data_type_array) {
static bool _is_variadic() { return false; }
static size_t _get_number_of_arguments() { return 1; }
static DataTypePtr get_return_type(const DataTypes& arguments) {
using Function = AggregateFunction<AggregateFunctionImpl<operation>>;
const DataTypeArray* data_type_array =
static_cast<const DataTypeArray*>(remove_nullable(arguments[0]).get());
auto function = Function::create(data_type_array->get_nested_type());
return function->get_return_type();
}
static Status execute(Block& block, size_t result, const DataTypeArray* data_type_array,
const ColumnArray& array) {
static Status execute(Block& block, const ColumnNumbers& arguments, size_t result,
const DataTypeArray* data_type_array, const ColumnArray& array) {
ColumnPtr res;
DataTypePtr type = data_type_array->get_nested_type();
const IColumn* data = array.get_data_ptr().get();
@ -275,12 +282,15 @@ using FunctionArrayAverage =
using FunctionArrayProduct =
FunctionArrayMapped<ArrayAggregateImpl<AggregateOperation::PRODUCT>, NameArrayProduct>;
using FunctionArrayJoin = FunctionArrayMapped<ArrayJoinImpl, NameArrayJoin>;
void register_function_array_aggregation(SimpleFunctionFactory& factory) {
factory.register_function<FunctionArrayMin>();
factory.register_function<FunctionArrayMax>();
factory.register_function<FunctionArraySum>();
factory.register_function<FunctionArrayAverage>();
factory.register_function<FunctionArrayProduct>();
factory.register_function<FunctionArrayJoin>();
}
} // namespace vectorized