mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-09-30 13:40:50 +00:00
43 lines
1.2 KiB
C++
43 lines
1.2 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, ' ');
|
||
|
|
||
|
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 << hilite_keyword << "UNION ALL" << hilite_none << settings.nl_or_ws;
|
||
|
|
||
|
(*it)->formatImpl(settings, state, frame);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
|
||
|
void ASTSelectWithUnionQuery::setDatabaseIfNeeded(const String & database_name)
|
||
|
{
|
||
|
for (auto & child : list_of_selects->children)
|
||
|
typeid_cast<ASTSelectQuery &>(*child).setDatabaseIfNeeded(database_name);
|
||
|
}
|
||
|
|
||
|
}
|