[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
111
be/src/vec/exec/vintersect_node.cpp
Normal file
111
be/src/vec/exec/vintersect_node.cpp
Normal file
@ -0,0 +1,111 @@
|
||||
// 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/exec/vintersect_node.h"
|
||||
|
||||
#include "gen_cpp/PlanNodes_types.h"
|
||||
#include "runtime/runtime_state.h"
|
||||
#include "util/runtime_profile.h"
|
||||
#include "vec/core/block.h"
|
||||
#include "vec/exec/vset_operation_node.h"
|
||||
#include "vec/exprs/vexpr.h"
|
||||
#include "vec/exprs/vexpr_context.h"
|
||||
namespace doris {
|
||||
namespace vectorized {
|
||||
|
||||
VIntersectNode::VIntersectNode(ObjectPool* pool, const TPlanNode& tnode, const DescriptorTbl& descs)
|
||||
: VSetOperationNode(pool, tnode, descs) {}
|
||||
|
||||
Status VIntersectNode::init(const TPlanNode& tnode, RuntimeState* state) {
|
||||
RETURN_IF_ERROR(VSetOperationNode::init(tnode, state));
|
||||
DCHECK(tnode.__isset.intersect_node);
|
||||
return Status::OK();
|
||||
}
|
||||
|
||||
Status VIntersectNode::prepare(RuntimeState* state) {
|
||||
RETURN_IF_ERROR(VSetOperationNode::prepare(state));
|
||||
return Status::OK();
|
||||
}
|
||||
|
||||
Status VIntersectNode::open(RuntimeState* state) {
|
||||
RETURN_IF_ERROR(VSetOperationNode::open(state));
|
||||
bool eos = false;
|
||||
Status st;
|
||||
|
||||
for (int i = 1; i < _children.size(); ++i) {
|
||||
if (i > 1) {
|
||||
refresh_hash_table<true>();
|
||||
}
|
||||
|
||||
_valid_element_in_hash_tbl = 0;
|
||||
RETURN_IF_ERROR(child(i)->open(state));
|
||||
eos = false;
|
||||
_probe_columns.resize(_child_expr_lists[i].size());
|
||||
|
||||
while (!eos) {
|
||||
RETURN_IF_ERROR(process_probe_block(state, i, &eos));
|
||||
if (_probe_rows == 0) continue;
|
||||
|
||||
std::visit(
|
||||
[&](auto&& arg) {
|
||||
using HashTableCtxType = std::decay_t<decltype(arg)>;
|
||||
if constexpr (!std::is_same_v<HashTableCtxType, std::monostate>) {
|
||||
HashTableProbe<HashTableCtxType, true> process_hashtable_ctx(
|
||||
this, state->batch_size(), _probe_rows);
|
||||
st = process_hashtable_ctx.mark_data_in_hashtable(arg);
|
||||
|
||||
} else {
|
||||
LOG(FATAL) << "FATAL: uninited hash table";
|
||||
}
|
||||
},
|
||||
_hash_table_variants);
|
||||
}
|
||||
}
|
||||
return st;
|
||||
}
|
||||
|
||||
Status VIntersectNode::get_next(RuntimeState* state, Block* output_block, bool* eos) {
|
||||
SCOPED_TIMER(_probe_timer);
|
||||
create_mutable_cols(output_block);
|
||||
Status st;
|
||||
|
||||
std::visit(
|
||||
[&](auto&& arg) {
|
||||
using HashTableCtxType = std::decay_t<decltype(arg)>;
|
||||
if constexpr (!std::is_same_v<HashTableCtxType, std::monostate>) {
|
||||
HashTableProbe<HashTableCtxType, true> process_hashtable_ctx(
|
||||
this, state->batch_size(), _probe_rows);
|
||||
st = process_hashtable_ctx.get_data_in_hashtable(arg, _mutable_cols,
|
||||
output_block, eos);
|
||||
|
||||
} else {
|
||||
LOG(FATAL) << "FATAL: uninited hash table";
|
||||
}
|
||||
},
|
||||
_hash_table_variants);
|
||||
|
||||
RETURN_IF_ERROR(VExprContext::filter_block(_vconjunct_ctx_ptr, output_block, output_block->columns()));
|
||||
reached_limit(output_block, eos);
|
||||
|
||||
return st;
|
||||
}
|
||||
|
||||
Status VIntersectNode::close(RuntimeState* state) {
|
||||
return VSetOperationNode::close(state);
|
||||
}
|
||||
} // namespace vectorized
|
||||
} // namespace doris
|
||||
Reference in New Issue
Block a user