YQ Connector: move tests from yql to ydb (OSS)

Перенос папки с тестами на Коннектор из папки yql в папку ydb (синхронизируется с github).

1 files changed, 55 insertions, 0 deletions

diff --git a/contrib/clickhouse/src/Interpreters/PredicateRewriteVisitor.h b/contrib/clickhouse/src/Interpreters/PredicateRewriteVisitor.h
new file mode 100644
index 0000000000..d2b9ece930
--- /dev/null
+++ b/contrib/clickhouse/src/Interpreters/PredicateRewriteVisitor.h
@@ -0,0 +1,55 @@
+#pragma once
+
+#include <Interpreters/Context_fwd.h>
+#include <Interpreters/DatabaseAndTableWithAlias.h>
+#include <Interpreters/InDepthNodeVisitor.h>
+#include <Parsers/IAST_fwd.h>
+
+namespace DB
+{
+
+class ASTSelectIntersectExceptQuery;
+class ASTSelectQuery;
+class ASTSelectWithUnionQuery;
+
+class PredicateRewriteVisitorData : WithContext
+{
+public:
+    bool is_rewrite = false;
+    using TypeToVisit = ASTSelectWithUnionQuery;
+
+    void visit(ASTSelectWithUnionQuery & union_select_query, ASTPtr &);
+
+    static bool needChild(const ASTPtr & node, const ASTPtr &)
+    {
+        return !(node && node->as<TypeToVisit>());
+    }
+
+    PredicateRewriteVisitorData(
+        ContextPtr context_,
+        const ASTs & predicates_,
+        const TableWithColumnNamesAndTypes & table_columns_,
+        bool optimize_final_,
+        bool optimize_with_);
+
+private:
+    const ASTs & predicates;
+    const TableWithColumnNamesAndTypes & table_columns;
+    bool optimize_final;
+    bool optimize_with;
+
+    void visitFirstInternalSelect(ASTSelectQuery & select_query, ASTPtr &);
+
+    void visitOtherInternalSelect(ASTSelectQuery & select_query, ASTPtr &);
+
+    void visit(ASTSelectIntersectExceptQuery & intersect_except_query, ASTPtr &);
+
+    bool rewriteSubquery(ASTSelectQuery & subquery, const Names & inner_columns);
+
+    void visitInternalSelect(size_t index, ASTSelectQuery & select_node, ASTPtr & node);
+};
+
+using PredicateRewriteMatcher = OneTypeMatcher<PredicateRewriteVisitorData, PredicateRewriteVisitorData::needChild>;
+using PredicateRewriteVisitor = InDepthNodeVisitor<PredicateRewriteMatcher, true>;
+
+}