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

122 lines
3.6 KiB
C
Raw Normal View History

2011-08-18 18:48:00 +00:00
#pragma once
#include <DB/Parsers/ASTExpressionList.h>
#include <DB/Parsers/ASTFunction.h>
namespace DB
{
/** CREATE TABLE или ATTACH TABLE запрос
*/
class ASTCreateQuery : public IAST
{
public:
2014-12-17 15:26:24 +00:00
bool attach{false}; /// Запрос ATTACH TABLE, а не CREATE TABLE.
bool if_not_exists{false};
bool is_view{false};
bool is_materialized_view{false};
bool is_populate{false};
bool is_temporary{false};
2011-10-30 05:19:41 +00:00
String database;
String table;
2011-08-18 18:48:00 +00:00
ASTPtr columns;
ASTPtr storage;
ASTPtr inner_storage; /// Внутренний engine для запроса CREATE MATERIALIZED VIEW
2011-10-31 17:30:44 +00:00
String as_database;
String as_table;
2011-11-01 15:16:04 +00:00
ASTPtr select;
2011-08-18 18:48:00 +00:00
2014-12-17 15:26:24 +00:00
ASTCreateQuery() = default;
ASTCreateQuery(const StringRange range_) : IAST(range_) {}
2015-08-09 05:10:43 +00:00
2011-08-18 18:48:00 +00:00
/** Получить текст, который идентифицирует этот элемент. */
2014-12-17 15:26:24 +00:00
String getID() const override { return (attach ? "AttachQuery_" : "CreateQuery_") + database + "_" + table; };
2011-12-12 06:15:34 +00:00
2014-12-17 15:26:24 +00:00
ASTPtr clone() const override
2011-12-12 06:15:34 +00:00
{
ASTCreateQuery * res = new ASTCreateQuery(*this);
2014-12-17 15:26:24 +00:00
ASTPtr ptr{res};
2011-12-12 06:15:34 +00:00
res->children.clear();
if (columns) { res->columns = columns->clone(); res->children.push_back(res->columns); }
if (storage) { res->storage = storage->clone(); res->children.push_back(res->storage); }
if (select) { res->select = select->clone(); res->children.push_back(res->select); }
if (inner_storage) { res->inner_storage = inner_storage->clone(); res->children.push_back(res->inner_storage); }
2011-12-12 06:15:34 +00:00
2014-12-17 15:26:24 +00:00
return ptr;
2011-12-12 06:15:34 +00:00
}
protected:
void formatImpl(const FormatSettings & settings, FormatState & state, FormatStateStacked frame) const override
{
frame.need_parens = false;
if (!database.empty() && table.empty())
{
settings.ostr << (settings.hilite ? hilite_keyword : "") << (attach ? "ATTACH DATABASE " : "CREATE DATABASE ") << (if_not_exists ? "IF NOT EXISTS " : "") << (settings.hilite ? hilite_none : "")
<< backQuoteIfNeed(database);
return;
}
{
std::string what = "TABLE";
if (is_view)
what = "VIEW";
if (is_materialized_view)
what = "MATERIALIZED VIEW";
settings.ostr
<< (settings.hilite ? hilite_keyword : "")
<< (attach ? "ATTACH " : "CREATE ")
<< (is_temporary ? "TEMPORARY " : "")
<< what
<< " " << (if_not_exists ? "IF NOT EXISTS " : "")
<< (settings.hilite ? hilite_none : "")
<< (!database.empty() ? backQuoteIfNeed(database) + "." : "") << backQuoteIfNeed(table);
}
if (!as_table.empty())
{
settings.ostr << (settings.hilite ? hilite_keyword : "") << " AS " << (settings.hilite ? hilite_none : "")
<< (!as_database.empty() ? backQuoteIfNeed(as_database) + "." : "") << backQuoteIfNeed(as_table);
}
if (columns)
{
settings.ostr << (settings.one_line ? " (" : "\n(");
2015-08-09 05:10:43 +00:00
FormatStateStacked frame_nested = frame;
++frame_nested.indent;
columns->formatImpl(settings, state, frame_nested);
settings.ostr << (settings.one_line ? ")" : "\n)");
}
if (storage && !is_materialized_view && !is_view)
{
settings.ostr << (settings.hilite ? hilite_keyword : "") << " ENGINE" << (settings.hilite ? hilite_none : "") << " = ";
storage->formatImpl(settings, state, frame);
}
if (inner_storage)
{
settings.ostr << (settings.hilite ? hilite_keyword : "") << " ENGINE" << (settings.hilite ? hilite_none : "") << " = ";
inner_storage->formatImpl(settings, state, frame);
}
if (is_populate)
{
settings.ostr << (settings.hilite ? hilite_keyword : "") << " POPULATE" << (settings.hilite ? hilite_none : "");
}
if (select)
{
settings.ostr << (settings.hilite ? hilite_keyword : "") << " AS" << settings.nl_or_ws << (settings.hilite ? hilite_none : "");
select->formatImpl(settings, state, frame);
}
}
2011-08-18 18:48:00 +00:00
};
}