ClickHouse/dbms/include/DB/Parsers/ASTQueryWithOutput.h

55 lines
1.5 KiB
C++
Raw Normal View History

#pragma once
#include <DB/Parsers/IAST.h>
namespace DB
{
2015-06-25 17:38:54 +00:00
/** Запрос с секцией FORMAT.
2015-06-25 17:38:54 +00:00
*/
class ASTQueryWithOutput : public IAST
{
public:
ASTPtr format;
2014-12-17 15:26:24 +00:00
ASTQueryWithOutput() = default;
ASTQueryWithOutput(const StringRange range_) : IAST(range_) {}
2015-06-25 17:38:54 +00:00
/** Возвращает указатель на формат. Если типом объекта является ASTSelectQuery,
* то эта функция возвращает указатель на формат из последнего SELECT'а цепочки UNION ALL.
*/
virtual const IAST * getFormat() const { return format.get(); }
};
/// Объявляет класс-наследник ASTQueryWithOutput с реализованными методами getID и clone.
#define DEFINE_AST_QUERY_WITH_OUTPUT(Name, ID, Query) \
class Name : public ASTQueryWithOutput \
{ \
public: \
Name() {} \
Name(StringRange range_) : ASTQueryWithOutput(range_) {} \
2014-12-17 15:26:24 +00:00
String getID() const override { return ID; }; \
\
2014-12-17 15:26:24 +00:00
ASTPtr clone() const override \
{ \
2016-06-09 04:37:21 +00:00
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); \
} \
2016-06-09 04:37:21 +00:00
return res; \
} \
\
protected: \
void formatImpl(const FormatSettings & settings, FormatState & state, FormatStateStacked frame) const override \
{ \
settings.ostr << (settings.hilite ? hilite_keyword : "") << Query << (settings.hilite ? hilite_none : ""); \
} \
};
}