diff options
author | vitalyisaev <vitalyisaev@ydb.tech> | 2023-11-14 09:58:56 +0300 |
---|---|---|
committer | vitalyisaev <vitalyisaev@ydb.tech> | 2023-11-14 10:20:20 +0300 |
commit | c2b2dfd9827a400a8495e172a56343462e3ceb82 (patch) | |
tree | cd4e4f597d01bede4c82dffeb2d780d0a9046bd0 /contrib/clickhouse/src/Interpreters/GatherFunctionQuantileVisitor.h | |
parent | d4ae8f119e67808cb0cf776ba6e0cf95296f2df7 (diff) | |
download | ydb-c2b2dfd9827a400a8495e172a56343462e3ceb82.tar.gz |
YQ Connector: move tests from yql to ydb (OSS)
Перенос папки с тестами на Коннектор из папки yql в папку ydb (синхронизируется с github).
Diffstat (limited to 'contrib/clickhouse/src/Interpreters/GatherFunctionQuantileVisitor.h')
-rw-r--r-- | contrib/clickhouse/src/Interpreters/GatherFunctionQuantileVisitor.h | 37 |
1 files changed, 37 insertions, 0 deletions
diff --git a/contrib/clickhouse/src/Interpreters/GatherFunctionQuantileVisitor.h b/contrib/clickhouse/src/Interpreters/GatherFunctionQuantileVisitor.h new file mode 100644 index 0000000000..74388e7161 --- /dev/null +++ b/contrib/clickhouse/src/Interpreters/GatherFunctionQuantileVisitor.h @@ -0,0 +1,37 @@ +#pragma once + +#include <Interpreters/InDepthNodeVisitor.h> +#include <Parsers/IAST_fwd.h> + +namespace DB +{ + +class ASTFunction; + +/// Gather all the `quantile*` functions +class GatherFunctionQuantileData +{ +public: + struct FuseQuantileAggregatesData + { + std::unordered_map<String, std::vector<ASTPtr *>> arg_map_function; + + void addFuncNode(ASTPtr & ast); + }; + + using TypeToVisit = ASTFunction; + + std::unordered_map<String, FuseQuantileAggregatesData> fuse_quantile; + + void visit(ASTFunction & function, ASTPtr & ast); + + static String toFusedNameOrSelf(const String & func_name); + + static String getFusedName(const String & func_name); + + static bool needChild(const ASTPtr & node, const ASTPtr &); +}; + +using GatherFunctionQuantileVisitor = InDepthNodeVisitor<OneTypeMatcher<GatherFunctionQuantileData, GatherFunctionQuantileData::needChild>, true>; + +} |