2011-08-18 18:48:00 +00:00
|
|
|
#pragma once
|
|
|
|
|
2017-04-01 09:19:00 +00:00
|
|
|
#include <Parsers/ASTExpressionList.h>
|
|
|
|
#include <Parsers/ASTFunction.h>
|
2017-08-03 17:00:41 +00:00
|
|
|
#include <Parsers/ASTQueryWithOutput.h>
|
2017-04-21 12:39:28 +00:00
|
|
|
#include <Parsers/ASTQueryWithOnCluster.h>
|
2011-08-18 18:48:00 +00:00
|
|
|
|
|
|
|
|
|
|
|
namespace DB
|
|
|
|
{
|
|
|
|
|
|
|
|
|
2017-04-13 16:12:56 +00:00
|
|
|
/// CREATE TABLE or ATTACH TABLE query
|
2017-08-03 17:00:41 +00:00
|
|
|
class ASTCreateQuery : public ASTQueryWithOutput, public ASTQueryWithOnCluster
|
2011-08-18 18:48:00 +00:00
|
|
|
{
|
|
|
|
public:
|
2017-05-27 17:27:16 +00:00
|
|
|
bool attach{false}; /// Query ATTACH TABLE, not CREATE TABLE.
|
2017-04-01 07:20:54 +00:00
|
|
|
bool if_not_exists{false};
|
|
|
|
bool is_view{false};
|
|
|
|
bool is_materialized_view{false};
|
|
|
|
bool is_populate{false};
|
|
|
|
bool is_temporary{false};
|
|
|
|
String database;
|
|
|
|
String table;
|
|
|
|
ASTPtr columns;
|
|
|
|
ASTPtr storage;
|
2017-05-27 17:27:16 +00:00
|
|
|
ASTPtr inner_storage; /// Internal engine for the CREATE MATERIALIZED VIEW query
|
2017-04-01 07:20:54 +00:00
|
|
|
String as_database;
|
|
|
|
String as_table;
|
|
|
|
ASTPtr select;
|
|
|
|
|
|
|
|
ASTCreateQuery() = default;
|
2017-08-03 17:00:41 +00:00
|
|
|
ASTCreateQuery(const StringRange range_) : ASTQueryWithOutput(range_) {}
|
2017-04-01 07:20:54 +00:00
|
|
|
|
2017-05-27 17:27:16 +00:00
|
|
|
/** Get the text that identifies this element. */
|
2017-04-01 07:20:54 +00:00
|
|
|
String getID() const override { return (attach ? "AttachQuery_" : "CreateQuery_") + database + "_" + table; };
|
|
|
|
|
|
|
|
ASTPtr clone() const override
|
|
|
|
{
|
|
|
|
auto res = std::make_shared<ASTCreateQuery>(*this);
|
|
|
|
res->children.clear();
|
|
|
|
|
2017-04-25 15:21:03 +00:00
|
|
|
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); }
|
2017-04-01 07:20:54 +00:00
|
|
|
|
2017-08-03 17:00:41 +00:00
|
|
|
cloneOutputOptions(*res);
|
|
|
|
|
2017-04-01 07:20:54 +00:00
|
|
|
return res;
|
|
|
|
}
|
2015-08-05 21:38:31 +00:00
|
|
|
|
2017-04-13 16:12:56 +00:00
|
|
|
ASTPtr getRewrittenASTWithoutOnCluster(const std::string & new_database) const override
|
|
|
|
{
|
|
|
|
auto query_ptr = clone();
|
|
|
|
ASTCreateQuery & query = static_cast<ASTCreateQuery &>(*query_ptr);
|
|
|
|
|
|
|
|
query.cluster.clear();
|
|
|
|
if (query.database.empty())
|
|
|
|
query.database = new_database;
|
|
|
|
|
|
|
|
return query_ptr;
|
|
|
|
}
|
|
|
|
|
2015-08-05 21:38:31 +00:00
|
|
|
protected:
|
2017-08-03 17:00:41 +00:00
|
|
|
void formatQueryImpl(const FormatSettings & settings, FormatState & state, FormatStateStacked frame) const override
|
2017-04-01 07:20:54 +00:00
|
|
|
{
|
|
|
|
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);
|
2017-04-25 15:21:03 +00:00
|
|
|
formatOnCluster(settings);
|
2017-04-01 07:20:54 +00:00
|
|
|
|
|
|
|
if (storage)
|
|
|
|
{
|
|
|
|
settings.ostr << (settings.hilite ? hilite_keyword : "") << " ENGINE" << (settings.hilite ? hilite_none : "") << " = ";
|
|
|
|
storage->formatImpl(settings, state, frame);
|
|
|
|
}
|
|
|
|
|
|
|
|
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 " : "")
|
2017-04-13 16:12:56 +00:00
|
|
|
<< what << " "
|
|
|
|
<< (if_not_exists ? "IF NOT EXISTS " : "")
|
2017-04-01 07:20:54 +00:00
|
|
|
<< (settings.hilite ? hilite_none : "")
|
2017-04-21 12:39:28 +00:00
|
|
|
<< (!database.empty() ? backQuoteIfNeed(database) + "." : "") << backQuoteIfNeed(table);
|
|
|
|
formatOnCluster(settings);
|
2017-04-01 07:20:54 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
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(");
|
|
|
|
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
|
|
|
};
|
|
|
|
|
|
|
|
}
|