mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-30 19:42:00 +00:00
66 lines
2.0 KiB
C++
66 lines
2.0 KiB
C++
#pragma once
|
||
|
||
#include <DB/Parsers/IAST.h>
|
||
#include <DB/Parsers/ASTQueryWithOutput.h>
|
||
|
||
|
||
namespace DB
|
||
{
|
||
|
||
|
||
/** Запрос с указанием названия таблицы и, возможно, БД и секцией FORMAT.
|
||
*/
|
||
class ASTQueryWithTableAndOutput : public ASTQueryWithOutput
|
||
{
|
||
public:
|
||
String database;
|
||
String table;
|
||
|
||
ASTQueryWithTableAndOutput() = default;
|
||
ASTQueryWithTableAndOutput(const StringRange range_) : ASTQueryWithOutput(range_) {}
|
||
|
||
protected:
|
||
void formatHelper(const FormatSettings & settings, FormatState & state, FormatStateStacked frame, const char * name) const
|
||
{
|
||
settings.ostr << (settings.hilite ? hilite_keyword : "") << name << " " << (settings.hilite ? hilite_none : "")
|
||
<< (!database.empty() ? backQuoteIfNeed(database) + "." : "") << backQuoteIfNeed(table);
|
||
|
||
if (format)
|
||
{
|
||
std::string indent_str = settings.one_line ? "" : std::string(4 * frame.indent, ' ');
|
||
settings.ostr << (settings.hilite ? hilite_keyword : "") << settings.nl_or_ws << indent_str << "FORMAT " << (settings.hilite ? hilite_none : "");
|
||
format->formatImpl(settings, state, frame);
|
||
}
|
||
}
|
||
};
|
||
|
||
|
||
/// Объявляет класс-наследник ASTQueryWithTableAndOutput с реализованными методами getID и clone.
|
||
#define DEFINE_AST_QUERY_WITH_TABLE_AND_OUTPUT(Name, ID, Query) \
|
||
class Name : public ASTQueryWithTableAndOutput \
|
||
{ \
|
||
public: \
|
||
Name() = default; \
|
||
Name(const StringRange range_) : ASTQueryWithTableAndOutput(range_) {} \
|
||
String getID() const override { return ID"_" + database + "_" + table; }; \
|
||
\
|
||
ASTPtr clone() const override \
|
||
{ \
|
||
std::shared_ptr<Name> res = std::make_shared<Name>(*this); \
|
||
res->children.clear(); \
|
||
if (format) \
|
||
{ \
|
||
res->format = format->clone(); \
|
||
res->children.push_back(res->format); \
|
||
} \
|
||
return res; \
|
||
} \
|
||
\
|
||
protected: \
|
||
void formatImpl(const FormatSettings & settings, FormatState & state, FormatStateStacked frame) const override \
|
||
{ \
|
||
formatHelper(settings, state, frame, Query); \
|
||
} \
|
||
};
|
||
}
|