mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-05 05:52:05 +00:00
96 lines
3.2 KiB
C++
96 lines
3.2 KiB
C++
#include <Parsers/ASTSelectWithUnionQuery.h>
|
|
#include <Parsers/ASTSubquery.h>
|
|
#include <Common/typeid_cast.h>
|
|
#include <Parsers/SelectUnionMode.h>
|
|
#include <IO/Operators.h>
|
|
|
|
#include <iostream>
|
|
|
|
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);
|
|
}
|
|
|
|
}
|