[feature](vectorization) Support Vectorized Exec Engine In Doris (#7785)
# Proposed changes Issue Number: close #6238 Co-authored-by: HappenLee <happenlee@hotmail.com> Co-authored-by: stdpain <34912776+stdpain@users.noreply.github.com> Co-authored-by: Zhengguo Yang <yangzhgg@gmail.com> Co-authored-by: wangbo <506340561@qq.com> Co-authored-by: emmymiao87 <522274284@qq.com> Co-authored-by: Pxl <952130278@qq.com> Co-authored-by: zhangstar333 <87313068+zhangstar333@users.noreply.github.com> Co-authored-by: thinker <zchw100@qq.com> Co-authored-by: Zeno Yang <1521564989@qq.com> Co-authored-by: Wang Shuo <wangshuo128@gmail.com> Co-authored-by: zhoubintao <35688959+zbtzbtzbt@users.noreply.github.com> Co-authored-by: Gabriel <gabrielleebuaa@gmail.com> Co-authored-by: xinghuayu007 <1450306854@qq.com> Co-authored-by: weizuo93 <weizuo@apache.org> Co-authored-by: yiguolei <guoleiyi@tencent.com> Co-authored-by: anneji-dev <85534151+anneji-dev@users.noreply.github.com> Co-authored-by: awakeljw <993007281@qq.com> Co-authored-by: taberylyang <95272637+taberylyang@users.noreply.github.com> Co-authored-by: Cui Kaifeng <48012748+azurenake@users.noreply.github.com> ## Problem Summary: ### 1. Some code from clickhouse **ClickHouse is an excellent implementation of the vectorized execution engine database, so here we have referenced and learned a lot from its excellent implementation in terms of data structure and function implementation. We are based on ClickHouse v19.16.2.2 and would like to thank the ClickHouse community and developers.** The following comment has been added to the code from Clickhouse, eg: // This file is copied from // https://github.com/ClickHouse/ClickHouse/blob/master/src/Interpreters/AggregationCommon.h // and modified by Doris ### 2. Support exec node and query: * vaggregation_node * vanalytic_eval_node * vassert_num_rows_node * vblocking_join_node * vcross_join_node * vempty_set_node * ves_http_scan_node * vexcept_node * vexchange_node * vintersect_node * vmysql_scan_node * vodbc_scan_node * volap_scan_node * vrepeat_node * vschema_scan_node * vselect_node * vset_operation_node * vsort_node * vunion_node * vhash_join_node You can run exec engine of SSB/TPCH and 70% TPCDS stand query test set. ### 3. Data Model Vec Exec Engine Support **Dup/Agg/Unq** table, Support Block Reader Vectorized. Segment Vec is working in process. ### 4. How to use 1. Set the environment variable `set enable_vectorized_engine = true; `(required) 2. Set the environment variable `set batch_size = 4096; ` (recommended) ### 5. Some diff from origin exec engine https://github.com/doris-vectorized/doris-vectorized/issues/294 ## Checklist(Required) 1. Does it affect the original behavior: (No) 2. Has unit tests been added: (Yes) 3. Has document been added or modified: (No) 4. Does it need to update dependencies: (No) 5. Are there any changes that cannot be rolled back: (Yes)
This commit is contained in:
HappenLee
87
be/src/vec/aggregate_functions/aggregate_function_bitmap.cpp
Normal file
87
be/src/vec/aggregate_functions/aggregate_function_bitmap.cpp
Normal file
@ -0,0 +1,87 @@
|
||||
// Licensed to the Apache Software Foundation (ASF) under one
|
||||
// or more contributor license agreements. See the NOTICE file
|
||||
// distributed with this work for additional information
|
||||
// regarding copyright ownership. The ASF licenses this file
|
||||
// to you under the Apache License, Version 2.0 (the
|
||||
// "License"); you may not use this file except in compliance
|
||||
// with the License. You may obtain a copy of the License at
|
||||
//
|
||||
// http://www.apache.org/licenses/LICENSE-2.0
|
||||
//
|
||||
// Unless required by applicable law or agreed to in writing,
|
||||
// software distributed under the License is distributed on an
|
||||
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
||||
// KIND, either express or implied. See the License for the
|
||||
// specific language governing permissions and limitations
|
||||
// under the License.
|
||||
|
||||
#include "vec/aggregate_functions/aggregate_function_bitmap.h"
|
||||
|
||||
#include "vec/aggregate_functions/aggregate_function_simple_factory.h"
|
||||
|
||||
namespace doris::vectorized {
|
||||
|
||||
template <bool nullable, template <bool, typename> class AggregateFunctionTemplate>
|
||||
static IAggregateFunction* createWithIntDataType(const DataTypes& argument_type) {
|
||||
auto type = argument_type[0].get();
|
||||
if (type->is_nullable()) {
|
||||
type = assert_cast<const DataTypeNullable*>(type)->get_nested_type().get();
|
||||
}
|
||||
WhichDataType which(type);
|
||||
if (which.idx == TypeIndex::Int8)
|
||||
return new AggregateFunctionTemplate<nullable, ColumnVector<Int8>>(argument_type);
|
||||
if (which.idx == TypeIndex::Int16)
|
||||
return new AggregateFunctionTemplate<nullable, ColumnVector<Int16>>(argument_type);
|
||||
if (which.idx == TypeIndex::Int32)
|
||||
return new AggregateFunctionTemplate<nullable, ColumnVector<Int32>>(argument_type);
|
||||
if (which.idx == TypeIndex::Int64)
|
||||
return new AggregateFunctionTemplate<nullable, ColumnVector<Int64>>(argument_type);
|
||||
return nullptr;
|
||||
}
|
||||
|
||||
AggregateFunctionPtr create_aggregate_function_bitmap_union(const std::string& name,
|
||||
const DataTypes& argument_types,
|
||||
const Array& parameters,
|
||||
const bool result_is_nullable) {
|
||||
return std::make_shared<AggregateFunctionBitmapOp<AggregateFunctionBitmapUnionOp>>(
|
||||
argument_types);
|
||||
}
|
||||
|
||||
AggregateFunctionPtr create_aggregate_function_bitmap_intersect(const std::string& name,
|
||||
const DataTypes& argument_types,
|
||||
const Array& parameters,
|
||||
const bool result_is_nullable) {
|
||||
return std::make_shared<AggregateFunctionBitmapOp<AggregateFunctionBitmapIntersectOp>>(
|
||||
argument_types);
|
||||
}
|
||||
template <bool nullable>
|
||||
AggregateFunctionPtr create_aggregate_function_bitmap_union_count(const std::string& name,
|
||||
const DataTypes& argument_types,
|
||||
const Array& parameters,
|
||||
const bool result_is_nullable) {
|
||||
return std::make_shared<AggregateFunctionBitmapCount<nullable, ColumnBitmap>>(argument_types);
|
||||
}
|
||||
|
||||
template <bool nullable>
|
||||
AggregateFunctionPtr create_aggregate_function_bitmap_union_int(const std::string& name,
|
||||
const DataTypes& argument_types,
|
||||
const Array& parameters,
|
||||
const bool result_is_nullable) {
|
||||
return std::shared_ptr<IAggregateFunction>(
|
||||
createWithIntDataType<nullable, AggregateFunctionBitmapCount>(argument_types));
|
||||
}
|
||||
|
||||
void register_aggregate_function_bitmap(AggregateFunctionSimpleFactory& factory) {
|
||||
factory.register_function("bitmap_union", create_aggregate_function_bitmap_union);
|
||||
factory.register_function("bitmap_intersect", create_aggregate_function_bitmap_intersect);
|
||||
factory.register_function("bitmap_union_count",
|
||||
create_aggregate_function_bitmap_union_count<false>);
|
||||
factory.register_function("bitmap_union_count",
|
||||
create_aggregate_function_bitmap_union_count<true>, true);
|
||||
|
||||
factory.register_function("bitmap_union_int",
|
||||
create_aggregate_function_bitmap_union_int<false>);
|
||||
factory.register_function("bitmap_union_int", create_aggregate_function_bitmap_union_int<true>,
|
||||
true);
|
||||
}
|
||||
} // namespace doris::vectorized
|
||||
Reference in New Issue
Block a user