aboutsummaryrefslogtreecommitdiffstats
path: root/contrib/clickhouse/src/Interpreters/GatherFunctionQuantileVisitor.h
diff options
context:
space:
mode:
authorvitalyisaev <vitalyisaev@ydb.tech>2023-11-14 09:58:56 +0300
committervitalyisaev <vitalyisaev@ydb.tech>2023-11-14 10:20:20 +0300
commitc2b2dfd9827a400a8495e172a56343462e3ceb82 (patch)
treecd4e4f597d01bede4c82dffeb2d780d0a9046bd0 /contrib/clickhouse/src/Interpreters/GatherFunctionQuantileVisitor.h
parentd4ae8f119e67808cb0cf776ba6e0cf95296f2df7 (diff)
downloadydb-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.h37
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>;
+
+}