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

[feature](vectorization) Support Vectorized Exec Engine In Doris (#7785)

Browse Source 
# 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
2022-01-18 10:07:15 +08:00
committed by GitHub
parent ebc27a40d7
commit e1d7233e9c
498 changed files with 69593 additions and 479 deletions
Download Patch File Download Diff File Expand all files Collapse all files

163
be/src/vec/functions/function_conv.cpp Normal file
View File

@ -0,0 +1,163 @@
// 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 "exprs/math_functions.h"
#include "vec/data_types/data_type_number.h"
#include "vec/data_types/data_type_string.h"
#include "vec/functions/simple_function_factory.h"
#include "vec/utils/util.hpp"
namespace doris::vectorized {
template <typename Impl>
class FunctionConv : public IFunction {
public:
static constexpr auto name = "conv";
String get_name() const override { return name; }
static FunctionPtr create() { return std::make_shared<FunctionConv<Impl>>(); }
DataTypePtr get_return_type_impl(const DataTypes& arguments) const override {
return make_nullable(std::make_shared<DataTypeString>());
}
DataTypes get_variadic_argument_types_impl() const override {
return {std::make_shared<typename Impl::DataType>(), std::make_shared<DataTypeInt8>(),
std::make_shared<DataTypeInt8>()};
}
size_t get_number_of_arguments() const override {
return get_variadic_argument_types_impl().size();
}
bool use_default_implementation_for_nulls() const override { return false; }
bool use_default_implementation_for_constants() const override { return true; }
Status execute_impl(FunctionContext* context, Block& block, const ColumnNumbers& arguments,
size_t result, size_t input_rows_count) override {
auto result_column = ColumnString::create();
auto result_null_map_column = ColumnUInt8::create(input_rows_count, 0);
ColumnPtr argument_columns[3];
for (int i = 0; i < 3; ++i) {
argument_columns[i] = block.get_by_position(arguments[i]).column;
if (auto* nullable = check_and_get_column<ColumnNullable>(*argument_columns[i])) {
argument_columns[i] = nullable->get_nested_column_ptr();
VectorizedUtils::update_null_map(result_null_map_column->get_data(),
nullable->get_null_map_data());
}
}
execute_straight(
context,
assert_cast<const typename Impl::DataType::ColumnType*>(argument_columns[0].get()),
assert_cast<const ColumnInt8*>(argument_columns[1].get()),
assert_cast<const ColumnInt8*>(argument_columns[2].get()),
assert_cast<ColumnString*>(result_column.get()),
assert_cast<ColumnUInt8*>(result_null_map_column.get())->get_data(),
input_rows_count);
block.get_by_position(result).column =
ColumnNullable::create(std::move(result_column), std::move(result_null_map_column));
return Status::OK();
}
private:
void execute_straight(FunctionContext* context,
const typename Impl::DataType::ColumnType* data_column,
const ColumnInt8* src_base_column, const ColumnInt8* dst_base_column,
ColumnString* result_column, NullMap& result_null_map,
size_t input_rows_count) {
for (size_t i = 0; i < input_rows_count; i++) {
if (result_null_map[i]) {
result_column->insert_default();
continue;
}
Int8 src_base = src_base_column->get_element(i);
Int8 dst_base = dst_base_column->get_element(i);
if (std::abs(src_base) < MathFunctions::MIN_BASE ||
std::abs(src_base) > MathFunctions::MAX_BASE ||
std::abs(dst_base) < MathFunctions::MIN_BASE ||
std::abs(dst_base) > MathFunctions::MAX_BASE) {
result_null_map[i] = true;
result_column->insert_default();
continue;
}
Impl::calculate_cell(context, data_column, src_base, dst_base, result_column,
result_null_map, i);
}
}
};
struct ConvInt64Impl {
using DataType = DataTypeInt64;
static void calculate_cell(FunctionContext* context, const DataType::ColumnType* data_column,
const Int8 src_base, const Int8 dst_base,
ColumnString* result_column, NullMap& result_null_map,
size_t index) {
Int64 num = data_column->get_element(index);
if (src_base < 0 && num >= 0) {
result_null_map[index] = true;
result_column->insert_default();
return;
}
int64_t decimal_num = num;
if (src_base != 10) {
if (!MathFunctions::decimal_in_base_to_decimal(num, src_base, &decimal_num)) {
MathFunctions::handle_parse_result(dst_base, &decimal_num,
StringParser::PARSE_OVERFLOW);
}
}
StringVal str = MathFunctions::decimal_to_base(context, decimal_num, dst_base);
result_column->insert_data(reinterpret_cast<const char*>(str.ptr), str.len);
}
};
struct ConvStringImpl {
using DataType = DataTypeString;
static void calculate_cell(FunctionContext* context, const DataType::ColumnType* data_column,
const Int8 src_base, const Int8 dst_base,
ColumnString* result_column, NullMap& result_null_map,
size_t index) {
StringRef str = data_column->get_data_at(index);
StringParser::ParseResult parse_res;
int64_t decimal_num =
StringParser::string_to_int<int64_t>(str.data, str.size, src_base, &parse_res);
if (src_base < 0 && decimal_num >= 0) {
result_null_map[index] = true;
result_column->insert_default();
return;
}
if (!MathFunctions::handle_parse_result(dst_base, &decimal_num, parse_res)) {
result_column->insert_data("0", 1);
} else {
StringVal str = MathFunctions::decimal_to_base(context, decimal_num, dst_base);
result_column->insert_data(reinterpret_cast<const char*>(str.ptr), str.len);
}
}
};
void register_function_conv(SimpleFunctionFactory& factory) {
factory.register_function<FunctionConv<ConvInt64Impl>>();
factory.register_function<FunctionConv<ConvStringImpl>>();
}
} // namespace doris::vectorized