mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-04 21:42:39 +00:00
348 lines
14 KiB
C++
348 lines
14 KiB
C++
#include <Parsers/ASTAlterQuery.h>
|
|
#include <iomanip>
|
|
#include <IO/WriteHelpers.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
namespace ErrorCodes
|
|
{
|
|
extern const int UNEXPECTED_AST_STRUCTURE;
|
|
}
|
|
|
|
ASTPtr ASTAlterCommand::clone() const
|
|
{
|
|
auto res = std::make_shared<ASTAlterCommand>(*this);
|
|
res->children.clear();
|
|
|
|
if (col_decl)
|
|
{
|
|
res->col_decl = col_decl->clone();
|
|
res->children.push_back(res->col_decl);
|
|
}
|
|
if (column)
|
|
{
|
|
res->column = column->clone();
|
|
res->children.push_back(res->column);
|
|
}
|
|
if (order_by)
|
|
{
|
|
res->order_by = order_by->clone();
|
|
res->children.push_back(res->order_by);
|
|
}
|
|
if (partition)
|
|
{
|
|
res->partition = partition->clone();
|
|
res->children.push_back(res->partition);
|
|
}
|
|
if (predicate)
|
|
{
|
|
res->predicate = predicate->clone();
|
|
res->children.push_back(res->predicate);
|
|
}
|
|
if (ttl)
|
|
{
|
|
res->ttl = ttl->clone();
|
|
res->children.push_back(res->ttl);
|
|
}
|
|
if (values)
|
|
{
|
|
res->values = values->clone();
|
|
res->children.push_back(res->values);
|
|
}
|
|
|
|
return res;
|
|
}
|
|
|
|
void ASTAlterCommand::formatImpl(
|
|
const FormatSettings & settings, FormatState & state, FormatStateStacked frame) const
|
|
{
|
|
std::string indent_str = settings.one_line ? "" : std::string(4 * frame.indent, ' ');
|
|
|
|
if (type == ASTAlterCommand::ADD_COLUMN)
|
|
{
|
|
settings.ostr << (settings.hilite ? hilite_keyword : "") << indent_str << "ADD COLUMN " << (if_not_exists ? "IF NOT EXISTS " : "") << (settings.hilite ? hilite_none : "");
|
|
col_decl->formatImpl(settings, state, frame);
|
|
|
|
/// AFTER
|
|
if (column)
|
|
{
|
|
settings.ostr << (settings.hilite ? hilite_keyword : "") << indent_str << " AFTER " << (settings.hilite ? hilite_none : "");
|
|
column->formatImpl(settings, state, frame);
|
|
}
|
|
}
|
|
else if (type == ASTAlterCommand::DROP_COLUMN)
|
|
{
|
|
settings.ostr << (settings.hilite ? hilite_keyword : "") << indent_str
|
|
<< (clear_column ? "CLEAR " : "DROP ") << "COLUMN " << (if_exists ? "IF EXISTS " : "") << (settings.hilite ? hilite_none : "");
|
|
column->formatImpl(settings, state, frame);
|
|
if (partition)
|
|
{
|
|
settings.ostr << (settings.hilite ? hilite_keyword : "") << indent_str<< " IN PARTITION " << (settings.hilite ? hilite_none : "");
|
|
partition->formatImpl(settings, state, frame);
|
|
}
|
|
}
|
|
else if (type == ASTAlterCommand::MODIFY_COLUMN)
|
|
{
|
|
settings.ostr << (settings.hilite ? hilite_keyword : "") << indent_str << "MODIFY COLUMN " << (if_exists ? "IF EXISTS " : "") << (settings.hilite ? hilite_none : "");
|
|
col_decl->formatImpl(settings, state, frame);
|
|
}
|
|
else if (type == ASTAlterCommand::COMMENT_COLUMN)
|
|
{
|
|
settings.ostr << (settings.hilite ? hilite_keyword : "") << indent_str << "COMMENT COLUMN " << (if_exists ? "IF EXISTS " : "") << (settings.hilite ? hilite_none : "");
|
|
column->formatImpl(settings, state, frame);
|
|
settings.ostr << " " << (settings.hilite ? hilite_none : "");
|
|
comment->formatImpl(settings, state, frame);
|
|
}
|
|
else if (type == ASTAlterCommand::MODIFY_ORDER_BY)
|
|
{
|
|
settings.ostr << (settings.hilite ? hilite_keyword : "") << indent_str << "MODIFY ORDER BY " << (settings.hilite ? hilite_none : "");
|
|
order_by->formatImpl(settings, state, frame);
|
|
}
|
|
else if (type == ASTAlterCommand::ADD_INDEX)
|
|
{
|
|
settings.ostr << (settings.hilite ? hilite_keyword : "") << indent_str << "ADD INDEX " << (if_not_exists ? "IF NOT EXISTS " : "") << (settings.hilite ? hilite_none : "");
|
|
index_decl->formatImpl(settings, state, frame);
|
|
|
|
/// AFTER
|
|
if (index)
|
|
{
|
|
settings.ostr << (settings.hilite ? hilite_keyword : "") << indent_str << " AFTER " << (settings.hilite ? hilite_none : "");
|
|
index->formatImpl(settings, state, frame);
|
|
}
|
|
}
|
|
else if (type == ASTAlterCommand::DROP_INDEX)
|
|
{
|
|
settings.ostr << (settings.hilite ? hilite_keyword : "") << indent_str
|
|
<< (clear_index ? "CLEAR " : "DROP ") << "INDEX " << (if_exists ? "IF EXISTS " : "") << (settings.hilite ? hilite_none : "");
|
|
index->formatImpl(settings, state, frame);
|
|
if (partition)
|
|
{
|
|
settings.ostr << (settings.hilite ? hilite_keyword : "") << indent_str<< " IN PARTITION " << (settings.hilite ? hilite_none : "");
|
|
partition->formatImpl(settings, state, frame);
|
|
}
|
|
}
|
|
else if (type == ASTAlterCommand::MATERIALIZE_INDEX)
|
|
{
|
|
settings.ostr << (settings.hilite ? hilite_keyword : "") << indent_str
|
|
<< "MATERIALIZE INDEX " << (settings.hilite ? hilite_none : "");
|
|
index->formatImpl(settings, state, frame);
|
|
if (partition)
|
|
{
|
|
settings.ostr << (settings.hilite ? hilite_keyword : "") << indent_str<< " IN PARTITION " << (settings.hilite ? hilite_none : "");
|
|
partition->formatImpl(settings, state, frame);
|
|
}
|
|
}
|
|
else if (type == ASTAlterCommand::DROP_PARTITION)
|
|
{
|
|
settings.ostr << (settings.hilite ? hilite_keyword : "") << indent_str << (detach ? "DETACH" : "DROP") << " PARTITION "
|
|
<< (settings.hilite ? hilite_none : "");
|
|
partition->formatImpl(settings, state, frame);
|
|
}
|
|
else if (type == ASTAlterCommand::ATTACH_PARTITION)
|
|
{
|
|
settings.ostr << (settings.hilite ? hilite_keyword : "") << indent_str << "ATTACH "
|
|
<< (part ? "PART " : "PARTITION ") << (settings.hilite ? hilite_none : "");
|
|
partition->formatImpl(settings, state, frame);
|
|
}
|
|
else if (type == ASTAlterCommand::MOVE_PARTITION)
|
|
{
|
|
settings.ostr << (settings.hilite ? hilite_keyword : "") << indent_str << "MOVE "
|
|
<< (part ? "PART " : "PARTITION ") << (settings.hilite ? hilite_none : "");
|
|
partition->formatImpl(settings, state, frame);
|
|
settings.ostr << " TO ";
|
|
switch (move_destination_type)
|
|
{
|
|
case MoveDestinationType::DISK:
|
|
settings.ostr << "DISK ";
|
|
break;
|
|
case MoveDestinationType::VOLUME:
|
|
settings.ostr << "VOLUME ";
|
|
break;
|
|
case MoveDestinationType::NONE:
|
|
throw Exception("Unexpected MOVE destination (NONE)", ErrorCodes::UNEXPECTED_AST_STRUCTURE);
|
|
}
|
|
WriteBufferFromOwnString move_destination_name_buf;
|
|
writeQuoted(move_destination_name, move_destination_name_buf);
|
|
settings.ostr << move_destination_name_buf.str();
|
|
}
|
|
else if (type == ASTAlterCommand::REPLACE_PARTITION)
|
|
{
|
|
settings.ostr << (settings.hilite ? hilite_keyword : "") << indent_str << (replace ? "REPLACE" : "ATTACH") << " PARTITION "
|
|
<< (settings.hilite ? hilite_none : "");
|
|
partition->formatImpl(settings, state, frame);
|
|
settings.ostr << (settings.hilite ? hilite_keyword : "") << " FROM " << (settings.hilite ? hilite_none : "");
|
|
if (!from_database.empty())
|
|
{
|
|
settings.ostr << (settings.hilite ? hilite_identifier : "") << backQuoteIfNeed(from_database)
|
|
<< (settings.hilite ? hilite_none : "") << ".";
|
|
}
|
|
settings.ostr << (settings.hilite ? hilite_identifier : "") << backQuoteIfNeed(from_table) << (settings.hilite ? hilite_none : "");
|
|
}
|
|
else if (type == ASTAlterCommand::FETCH_PARTITION)
|
|
{
|
|
settings.ostr << (settings.hilite ? hilite_keyword : "") << indent_str << "FETCH "
|
|
<< "PARTITION " << (settings.hilite ? hilite_none : "");
|
|
partition->formatImpl(settings, state, frame);
|
|
settings.ostr << (settings.hilite ? hilite_keyword : "")
|
|
<< " FROM " << (settings.hilite ? hilite_none : "") << std::quoted(from, '\'');
|
|
}
|
|
else if (type == ASTAlterCommand::FREEZE_PARTITION)
|
|
{
|
|
settings.ostr << (settings.hilite ? hilite_keyword : "") << indent_str << "FREEZE PARTITION " << (settings.hilite ? hilite_none : "");
|
|
partition->formatImpl(settings, state, frame);
|
|
|
|
if (!with_name.empty())
|
|
{
|
|
settings.ostr << " " << (settings.hilite ? hilite_keyword : "") << "WITH NAME" << (settings.hilite ? hilite_none : "")
|
|
<< " " << std::quoted(with_name, '\'');
|
|
}
|
|
}
|
|
else if (type == ASTAlterCommand::FREEZE_ALL)
|
|
{
|
|
settings.ostr << (settings.hilite ? hilite_keyword : "") << indent_str << "FREEZE";
|
|
|
|
if (!with_name.empty())
|
|
{
|
|
settings.ostr << " " << (settings.hilite ? hilite_keyword : "") << "WITH NAME" << (settings.hilite ? hilite_none : "")
|
|
<< " " << std::quoted(with_name, '\'');
|
|
}
|
|
}
|
|
else if (type == ASTAlterCommand::DELETE)
|
|
{
|
|
settings.ostr << (settings.hilite ? hilite_keyword : "") << indent_str << "DELETE WHERE " << (settings.hilite ? hilite_none : "");
|
|
predicate->formatImpl(settings, state, frame);
|
|
}
|
|
else if (type == ASTAlterCommand::UPDATE)
|
|
{
|
|
settings.ostr << (settings.hilite ? hilite_keyword : "") << indent_str << "UPDATE " << (settings.hilite ? hilite_none : "");
|
|
update_assignments->formatImpl(settings, state, frame);
|
|
|
|
settings.ostr << (settings.hilite ? hilite_keyword : "") << " WHERE " << (settings.hilite ? hilite_none : "");
|
|
predicate->formatImpl(settings, state, frame);
|
|
}
|
|
else if (type == ASTAlterCommand::MODIFY_TTL)
|
|
{
|
|
settings.ostr << (settings.hilite ? hilite_keyword : "") << indent_str << "MODIFY TTL " << (settings.hilite ? hilite_none : "");
|
|
ttl->formatImpl(settings, state, frame);
|
|
}
|
|
else if (type == ASTAlterCommand::LIVE_VIEW_REFRESH)
|
|
{
|
|
settings.ostr << (settings.hilite ? hilite_keyword : "") << indent_str << "REFRESH " << (settings.hilite ? hilite_none : "");
|
|
}
|
|
else if (type == ASTAlterCommand::LIVE_CHANNEL_ADD)
|
|
{
|
|
settings.ostr << (settings.hilite ? hilite_keyword : "") << indent_str << "ADD " << (settings.hilite ? hilite_none : "");
|
|
|
|
values->formatImpl(settings, state, frame);
|
|
}
|
|
else if (type == ASTAlterCommand::LIVE_CHANNEL_DROP)
|
|
{
|
|
settings.ostr << (settings.hilite ? hilite_keyword : "") << indent_str << "DROP " << (settings.hilite ? hilite_none : "");
|
|
|
|
values->formatImpl(settings, state, frame);
|
|
}
|
|
else if (type == ASTAlterCommand::LIVE_CHANNEL_MODIFY)
|
|
{
|
|
settings.ostr << (settings.hilite ? hilite_keyword : "") << indent_str << "MODIFY " << (settings.hilite ? hilite_none : "");
|
|
|
|
values->formatImpl(settings, state, frame);
|
|
}
|
|
else if (type == ASTAlterCommand::LIVE_CHANNEL_SUSPEND)
|
|
{
|
|
settings.ostr << (settings.hilite ? hilite_keyword : "") << indent_str << "SUSPEND " << (settings.hilite ? hilite_none : "");
|
|
|
|
values->formatImpl(settings, state, frame);
|
|
}
|
|
else if (type == ASTAlterCommand::LIVE_CHANNEL_RESUME)
|
|
{
|
|
settings.ostr << (settings.hilite ? hilite_keyword : "") << indent_str << "RESUME " << (settings.hilite ? hilite_none : "");
|
|
|
|
values->formatImpl(settings, state, frame);
|
|
}
|
|
else if (type == ASTAlterCommand::LIVE_CHANNEL_REFRESH)
|
|
{
|
|
settings.ostr << (settings.hilite ? hilite_keyword : "") << indent_str << "REFRESH " << (settings.hilite ? hilite_none : "");
|
|
|
|
values->formatImpl(settings, state, frame);
|
|
}
|
|
else
|
|
throw Exception("Unexpected type of ALTER", ErrorCodes::UNEXPECTED_AST_STRUCTURE);
|
|
}
|
|
|
|
|
|
ASTPtr ASTAlterCommandList::clone() const
|
|
{
|
|
auto res = std::make_shared<ASTAlterCommandList>();
|
|
for (ASTAlterCommand * command : commands)
|
|
res->add(command->clone());
|
|
return res;
|
|
}
|
|
|
|
void ASTAlterCommandList::formatImpl(const FormatSettings & settings, FormatState & state, FormatStateStacked frame) const
|
|
{
|
|
std::string indent_str = settings.one_line ? "" : std::string(4 * frame.indent, ' ');
|
|
|
|
for (size_t i = 0; i < commands.size(); ++i)
|
|
{
|
|
static_cast<IAST *>(commands[i])->formatImpl(settings, state, frame);
|
|
|
|
std::string comma = (i < (commands.size() - 1)) ? "," : "";
|
|
settings.ostr << (settings.hilite ? hilite_keyword : "") << comma << (settings.hilite ? hilite_none : "");
|
|
|
|
settings.ostr << settings.nl_or_ws;
|
|
}
|
|
}
|
|
|
|
|
|
/** Get the text that identifies this element. */
|
|
String ASTAlterQuery::getID(char delim) const
|
|
{
|
|
return "AlterQuery" + (delim + database) + delim + table;
|
|
}
|
|
|
|
ASTPtr ASTAlterQuery::clone() const
|
|
{
|
|
auto res = std::make_shared<ASTAlterQuery>(*this);
|
|
res->children.clear();
|
|
|
|
if (command_list)
|
|
res->set(res->command_list, command_list->clone());
|
|
|
|
return res;
|
|
}
|
|
|
|
void ASTAlterQuery::formatQueryImpl(const FormatSettings & settings, FormatState & state, FormatStateStacked frame) const
|
|
{
|
|
frame.need_parens = false;
|
|
|
|
std::string indent_str = settings.one_line ? "" : std::string(4u * frame.indent, ' ');
|
|
|
|
if (is_live_view)
|
|
settings.ostr << (settings.hilite ? hilite_keyword : "") << indent_str << "ALTER LIVE VIEW " << (settings.hilite ? hilite_none : "");
|
|
else if (is_live_channel)
|
|
settings.ostr << (settings.hilite ? hilite_keyword : "") << indent_str << "ALTER LIVE CHANNEL " << (settings.hilite ? hilite_none : "");
|
|
else
|
|
settings.ostr << (settings.hilite ? hilite_keyword : "") << indent_str << "ALTER TABLE " << (settings.hilite ? hilite_none : "");
|
|
|
|
if (!table.empty())
|
|
{
|
|
if (!database.empty())
|
|
{
|
|
settings.ostr << indent_str << backQuoteIfNeed(database);
|
|
settings.ostr << ".";
|
|
}
|
|
settings.ostr << indent_str << backQuoteIfNeed(table);
|
|
}
|
|
formatOnCluster(settings);
|
|
settings.ostr << settings.nl_or_ws;
|
|
|
|
FormatStateStacked frame_nested = frame;
|
|
frame_nested.need_parens = false;
|
|
++frame_nested.indent;
|
|
static_cast<IAST *>(command_list)->formatImpl(settings, state, frame_nested);
|
|
}
|
|
|
|
}
|