blob: 48b4ae3c38de25eb308b8c067452b1d7ded2d126 (
plain) (
blame)
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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
#include <Parsers/ASTSelectWithUnionQuery.h>
#include <Parsers/ASTSubquery.h>
#include <Common/typeid_cast.h>
#include <Parsers/SelectUnionMode.h>
#include <IO/Operators.h>
#include <Parsers/ASTSelectQuery.h>
namespace DB
{
ASTPtr ASTSelectWithUnionQuery::clone() const
{
auto res = std::make_shared<ASTSelectWithUnionQuery>(*this);
res->children.clear();
res->list_of_selects = list_of_selects->clone();
res->children.push_back(res->list_of_selects);
res->union_mode = union_mode;
res->list_of_modes = list_of_modes;
res->set_of_modes = set_of_modes;
cloneOutputOptions(*res);
return res;
}
void ASTSelectWithUnionQuery::formatQueryImpl(const FormatSettings & settings, FormatState & state, FormatStateStacked frame) const
{
std::string indent_str = settings.one_line ? "" : std::string(4 * frame.indent, ' ');
auto mode_to_str = [&](auto mode)
{
if (mode == SelectUnionMode::UNION_DEFAULT)
return "UNION";
else if (mode == SelectUnionMode::UNION_ALL)
return "UNION ALL";
else if (mode == SelectUnionMode::UNION_DISTINCT)
return "UNION DISTINCT";
else if (mode == SelectUnionMode::EXCEPT_DEFAULT)
return "EXCEPT";
else if (mode == SelectUnionMode::EXCEPT_ALL)
return "EXCEPT ALL";
else if (mode == SelectUnionMode::EXCEPT_DISTINCT)
return "EXCEPT DISTINCT";
else if (mode == SelectUnionMode::INTERSECT_DEFAULT)
return "INTERSECT";
else if (mode == SelectUnionMode::INTERSECT_ALL)
return "INTERSECT ALL";
else if (mode == SelectUnionMode::INTERSECT_DISTINCT)
return "INTERSECT DISTINCT";
return "";
};
for (ASTs::const_iterator it = list_of_selects->children.begin(); it != list_of_selects->children.end(); ++it)
{
if (it != list_of_selects->children.begin())
settings.ostr << settings.nl_or_ws << indent_str << (settings.hilite ? hilite_keyword : "")
<< mode_to_str((is_normalized) ? union_mode : list_of_modes[it - list_of_selects->children.begin() - 1])
<< (settings.hilite ? hilite_none : "");
if (auto * node = (*it)->as<ASTSelectWithUnionQuery>())
{
settings.ostr << settings.nl_or_ws << indent_str;
if (node->list_of_selects->children.size() == 1)
{
(node->list_of_selects->children.at(0))->formatImpl(settings, state, frame);
}
else
{
auto sub_query = std::make_shared<ASTSubquery>();
sub_query->children.push_back(*it);
sub_query->formatImpl(settings, state, frame);
}
}
else
{
if (it != list_of_selects->children.begin())
settings.ostr << settings.nl_or_ws;
(*it)->formatImpl(settings, state, frame);
}
}
}
bool ASTSelectWithUnionQuery::hasNonDefaultUnionMode() const
{
return set_of_modes.contains(SelectUnionMode::UNION_DISTINCT) || set_of_modes.contains(SelectUnionMode::INTERSECT_DISTINCT)
|| set_of_modes.contains(SelectUnionMode::EXCEPT_DISTINCT);
}
bool ASTSelectWithUnionQuery::hasQueryParameters() const
{
if (!has_query_parameters.has_value())
{
for (const auto & child : list_of_selects->children)
{
if (auto * select_node = child->as<ASTSelectQuery>())
{
if (select_node->hasQueryParameters())
{
has_query_parameters = true;
return has_query_parameters.value();
}
}
}
has_query_parameters = false;
}
return has_query_parameters.value();
}
}
|
