mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-06 14:32:22 +00:00
48 lines
1.4 KiB
C++
48 lines
1.4 KiB
C++
#include <Parsers/ASTSelectWithUnionQuery.h>
|
|
#include <Parsers/ASTSelectQuery.h>
|
|
#include <Common/typeid_cast.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);
|
|
|
|
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 == Mode::Unspecified)
|
|
return "";
|
|
else if (mode == Mode::ALL)
|
|
return "ALL";
|
|
else
|
|
return "DISTINCT";
|
|
};
|
|
|
|
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 : "") << "UNION "
|
|
<< mode_to_str(union_modes[it - list_of_selects->children.begin() - 1]) << (settings.hilite ? hilite_none : "")
|
|
<< settings.nl_or_ws;
|
|
|
|
(*it)->formatImpl(settings, state, frame);
|
|
}
|
|
}
|
|
|
|
}
|