contrib/clickhouse/src/Storages/MergeTree/MergeTreeIndexUtils.cpp


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47

#include <Storages/MergeTree/MergeTreeIndexUtils.h>

#include <Parsers/ASTExpressionList.h>
#include <Parsers/ASTFunction.h>
#include <Parsers/ASTSelectQuery.h>

namespace DB
{

ASTPtr buildFilterNode(const ASTPtr & select_query, ASTs additional_filters)
{
    auto & select_query_typed = select_query->as<ASTSelectQuery &>();

    ASTs filters;
    if (select_query_typed.where())
        filters.push_back(select_query_typed.where());

    if (select_query_typed.prewhere())
        filters.push_back(select_query_typed.prewhere());

    filters.insert(filters.end(), additional_filters.begin(), additional_filters.end());

    if (filters.empty())
        return nullptr;

    ASTPtr filter_node;

    if (filters.size() == 1)
    {
        filter_node = filters.front();
    }
    else
    {
        auto function = std::make_shared<ASTFunction>();

        function->name = "and";
        function->arguments = std::make_shared<ASTExpressionList>();
        function->children.push_back(function->arguments);
        function->arguments->children = std::move(filters);

        filter_node = std::move(function);
    }

    return filter_node;
}

}