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

#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); \
		} \
	};
}