2017-04-01 09:19:00 +00:00
|
|
|
#include <Parsers/ParserAlterQuery.h>
|
|
|
|
#include <Parsers/CommonParsers.h>
|
|
|
|
#include <Parsers/ExpressionElementParsers.h>
|
|
|
|
#include <Parsers/ParserCreateQuery.h>
|
2017-09-06 20:34:26 +00:00
|
|
|
#include <Parsers/ParserPartition.h>
|
2017-04-01 09:19:00 +00:00
|
|
|
#include <Parsers/ASTIdentifier.h>
|
|
|
|
#include <Parsers/ASTAlterQuery.h>
|
|
|
|
#include <Parsers/ASTLiteral.h>
|
2018-05-21 13:49:54 +00:00
|
|
|
#include <Parsers/parseDatabaseAndTableName.h>
|
|
|
|
|
2013-08-07 13:07:42 +00:00
|
|
|
|
|
|
|
namespace DB
|
|
|
|
{
|
2016-01-28 01:00:27 +00:00
|
|
|
|
2017-07-10 03:28:12 +00:00
|
|
|
bool ParserAlterQuery::parseImpl(Pos & pos, ASTPtr & node, Expected & expected)
|
2013-08-07 13:07:42 +00:00
|
|
|
{
|
2017-06-18 03:07:03 +00:00
|
|
|
ParserKeyword s_alter_table("ALTER TABLE");
|
|
|
|
ParserKeyword s_add_column("ADD COLUMN");
|
|
|
|
ParserKeyword s_drop_column("DROP COLUMN");
|
2017-06-22 11:01:30 +00:00
|
|
|
ParserKeyword s_clear_column("CLEAR COLUMN");
|
2017-06-18 03:07:03 +00:00
|
|
|
ParserKeyword s_modify_column("MODIFY COLUMN");
|
|
|
|
ParserKeyword s_modify_primary_key("MODIFY PRIMARY KEY");
|
|
|
|
|
|
|
|
ParserKeyword s_attach_partition("ATTACH PARTITION");
|
|
|
|
ParserKeyword s_detach_partition("DETACH PARTITION");
|
|
|
|
ParserKeyword s_drop_partition("DROP PARTITION");
|
|
|
|
ParserKeyword s_attach_part("ATTACH PART");
|
|
|
|
ParserKeyword s_fetch_partition("FETCH PARTITION");
|
|
|
|
ParserKeyword s_freeze_partition("FREEZE PARTITION");
|
|
|
|
ParserKeyword s_partition("PARTITION");
|
|
|
|
|
|
|
|
ParserKeyword s_after("AFTER");
|
|
|
|
ParserKeyword s_from("FROM");
|
2017-06-22 11:01:30 +00:00
|
|
|
ParserKeyword s_in_partition("IN PARTITION");
|
2017-06-18 03:07:03 +00:00
|
|
|
ParserKeyword s_with("WITH");
|
|
|
|
ParserKeyword s_name("NAME");
|
2017-04-01 07:20:54 +00:00
|
|
|
|
2018-02-02 16:02:43 +00:00
|
|
|
ParserKeyword s_delete_where("DELETE WHERE");
|
|
|
|
|
2017-07-10 03:28:12 +00:00
|
|
|
ParserToken s_dot(TokenType::Dot);
|
|
|
|
ParserToken s_comma(TokenType::Comma);
|
2017-04-01 07:20:54 +00:00
|
|
|
|
|
|
|
ParserIdentifier table_parser;
|
|
|
|
ParserCompoundIdentifier parser_name;
|
|
|
|
ParserCompoundColumnDeclaration parser_col_decl;
|
2017-09-06 20:34:26 +00:00
|
|
|
ParserPartition parser_partition;
|
2017-04-01 07:20:54 +00:00
|
|
|
ParserStringLiteral parser_string_literal;
|
2018-02-02 16:02:43 +00:00
|
|
|
ParserExpression exp_elem;
|
2017-04-01 07:20:54 +00:00
|
|
|
|
2017-04-21 12:39:28 +00:00
|
|
|
String cluster_str;
|
2017-04-01 07:20:54 +00:00
|
|
|
ASTPtr col_type;
|
|
|
|
ASTPtr col_after;
|
|
|
|
ASTPtr col_drop;
|
|
|
|
|
|
|
|
auto query = std::make_shared<ASTAlterQuery>();
|
|
|
|
|
2017-07-10 03:28:12 +00:00
|
|
|
if (!s_alter_table.ignore(pos, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
return false;
|
|
|
|
|
2018-05-21 13:49:54 +00:00
|
|
|
if (!parseDatabaseAndTableName(pos, expected, query->database, query->table))
|
2017-04-01 07:20:54 +00:00
|
|
|
return false;
|
|
|
|
|
2017-07-10 03:28:12 +00:00
|
|
|
if (ParserKeyword{"ON"}.ignore(pos, expected))
|
2017-04-21 12:39:28 +00:00
|
|
|
{
|
2017-07-10 03:28:12 +00:00
|
|
|
if (!ASTQueryWithOnCluster::parse(pos, cluster_str, expected))
|
2017-04-21 12:39:28 +00:00
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
2017-04-01 07:20:54 +00:00
|
|
|
bool parsing_finished = false;
|
|
|
|
do
|
|
|
|
{
|
|
|
|
ASTAlterQuery::Parameters params;
|
|
|
|
|
2017-07-10 03:28:12 +00:00
|
|
|
if (s_add_column.ignore(pos, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
{
|
2017-07-10 03:28:12 +00:00
|
|
|
if (!parser_col_decl.parse(pos, params.col_decl, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
return false;
|
|
|
|
|
2017-07-10 03:28:12 +00:00
|
|
|
if (s_after.ignore(pos, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
{
|
2017-11-17 20:42:03 +00:00
|
|
|
if (!parser_name.parse(pos, params.column, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
params.type = ASTAlterQuery::ADD_COLUMN;
|
|
|
|
}
|
2017-07-10 03:28:12 +00:00
|
|
|
else if (s_drop_partition.ignore(pos, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
{
|
2017-09-06 20:34:26 +00:00
|
|
|
if (!parser_partition.parse(pos, params.partition, expected))
|
2017-06-18 03:07:03 +00:00
|
|
|
return false;
|
2017-04-01 07:20:54 +00:00
|
|
|
|
2017-06-18 03:07:03 +00:00
|
|
|
params.type = ASTAlterQuery::DROP_PARTITION;
|
|
|
|
}
|
2017-07-10 03:28:12 +00:00
|
|
|
else if (s_drop_column.ignore(pos, expected))
|
2017-06-18 03:07:03 +00:00
|
|
|
{
|
2017-07-10 03:28:12 +00:00
|
|
|
if (!parser_name.parse(pos, params.column, expected))
|
2017-06-18 03:07:03 +00:00
|
|
|
return false;
|
|
|
|
|
|
|
|
params.type = ASTAlterQuery::DROP_COLUMN;
|
|
|
|
params.detach = false;
|
2017-06-22 11:01:30 +00:00
|
|
|
}
|
2017-07-10 03:28:12 +00:00
|
|
|
else if (s_clear_column.ignore(pos, expected))
|
2017-06-22 11:01:30 +00:00
|
|
|
{
|
2017-07-10 03:28:12 +00:00
|
|
|
if (!parser_name.parse(pos, params.column, expected))
|
2017-06-22 11:01:30 +00:00
|
|
|
return false;
|
|
|
|
|
|
|
|
params.type = ASTAlterQuery::DROP_COLUMN;
|
|
|
|
params.clear_column = true;
|
|
|
|
params.detach = false;
|
|
|
|
|
2017-07-10 03:28:12 +00:00
|
|
|
if (s_in_partition.ignore(pos, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
{
|
2017-09-06 20:34:26 +00:00
|
|
|
if (!parser_partition.parse(pos, params.partition, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
return false;
|
|
|
|
}
|
|
|
|
}
|
2017-07-10 03:28:12 +00:00
|
|
|
else if (s_detach_partition.ignore(pos, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
{
|
2017-09-06 20:34:26 +00:00
|
|
|
if (!parser_partition.parse(pos, params.partition, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
return false;
|
|
|
|
|
|
|
|
params.type = ASTAlterQuery::DROP_PARTITION;
|
|
|
|
params.detach = true;
|
|
|
|
}
|
2017-07-10 03:28:12 +00:00
|
|
|
else if (s_attach_partition.ignore(pos, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
{
|
2017-09-06 20:34:26 +00:00
|
|
|
if (!parser_partition.parse(pos, params.partition, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
return false;
|
2014-08-07 11:46:01 +00:00
|
|
|
|
2018-05-21 13:49:54 +00:00
|
|
|
if (s_from.ignore(pos))
|
|
|
|
{
|
|
|
|
if (!parseDatabaseAndTableName(pos, expected, params.from_database, params.from_table))
|
|
|
|
return false;
|
|
|
|
|
|
|
|
params.replace = false;
|
|
|
|
params.type = ASTAlterQuery::REPLACE_PARTITION;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
params.type = ASTAlterQuery::ATTACH_PARTITION;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else if (ParserKeyword{"REPLACE PARTITION"}.ignore(pos, expected))
|
|
|
|
{
|
|
|
|
if (!parser_partition.parse(pos, params.partition, expected))
|
|
|
|
return false;
|
|
|
|
|
|
|
|
if (!s_from.ignore(pos, expected))
|
|
|
|
return false;
|
|
|
|
|
|
|
|
if (!parseDatabaseAndTableName(pos, expected, params.from_database, params.from_table))
|
|
|
|
return false;
|
|
|
|
|
|
|
|
params.replace = true;
|
|
|
|
params.type = ASTAlterQuery::REPLACE_PARTITION;
|
2017-04-01 07:20:54 +00:00
|
|
|
}
|
2017-07-10 03:28:12 +00:00
|
|
|
else if (s_attach_part.ignore(pos, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
{
|
2017-09-06 20:34:26 +00:00
|
|
|
if (!parser_string_literal.parse(pos, params.partition, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
return false;
|
2014-10-09 20:28:33 +00:00
|
|
|
|
2017-06-18 03:07:03 +00:00
|
|
|
params.part = true;
|
|
|
|
params.type = ASTAlterQuery::ATTACH_PARTITION;
|
|
|
|
}
|
2017-07-10 03:28:12 +00:00
|
|
|
else if (s_fetch_partition.ignore(pos, expected))
|
2017-06-18 03:07:03 +00:00
|
|
|
{
|
2017-09-06 20:34:26 +00:00
|
|
|
if (!parser_partition.parse(pos, params.partition, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
return false;
|
2014-10-09 20:28:33 +00:00
|
|
|
|
2017-07-10 03:28:12 +00:00
|
|
|
if (!s_from.ignore(pos, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
return false;
|
2014-10-09 20:28:33 +00:00
|
|
|
|
2017-04-01 07:20:54 +00:00
|
|
|
ASTPtr ast_from;
|
2017-07-10 03:28:12 +00:00
|
|
|
if (!parser_string_literal.parse(pos, ast_from, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
return false;
|
2014-10-09 20:28:33 +00:00
|
|
|
|
2017-04-01 07:20:54 +00:00
|
|
|
params.from = typeid_cast<const ASTLiteral &>(*ast_from).value.get<const String &>();
|
|
|
|
params.type = ASTAlterQuery::FETCH_PARTITION;
|
|
|
|
}
|
2017-07-10 03:28:12 +00:00
|
|
|
else if (s_freeze_partition.ignore(pos, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
{
|
2017-09-06 20:34:26 +00:00
|
|
|
if (!parser_partition.parse(pos, params.partition, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
return false;
|
2014-11-11 04:11:07 +00:00
|
|
|
|
2017-04-01 07:20:54 +00:00
|
|
|
/// WITH NAME 'name' - place local backup to directory with specified name
|
2017-07-10 03:28:12 +00:00
|
|
|
if (s_with.ignore(pos, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
{
|
2017-07-10 03:28:12 +00:00
|
|
|
if (!s_name.ignore(pos, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
return false;
|
2016-06-28 20:50:37 +00:00
|
|
|
|
2017-04-01 07:20:54 +00:00
|
|
|
ASTPtr ast_with_name;
|
2017-07-10 03:28:12 +00:00
|
|
|
if (!parser_string_literal.parse(pos, ast_with_name, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
return false;
|
2016-06-28 20:50:37 +00:00
|
|
|
|
2017-04-01 07:20:54 +00:00
|
|
|
params.with_name = typeid_cast<const ASTLiteral &>(*ast_with_name).value.get<const String &>();
|
|
|
|
}
|
2016-06-28 20:50:37 +00:00
|
|
|
|
2017-04-01 07:20:54 +00:00
|
|
|
params.type = ASTAlterQuery::FREEZE_PARTITION;
|
|
|
|
}
|
2017-07-10 03:28:12 +00:00
|
|
|
else if (s_modify_column.ignore(pos, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
{
|
2017-07-10 03:28:12 +00:00
|
|
|
if (!parser_col_decl.parse(pos, params.col_decl, expected))
|
2017-06-18 03:07:03 +00:00
|
|
|
return false;
|
2016-05-05 18:28:46 +00:00
|
|
|
|
2017-06-18 03:07:03 +00:00
|
|
|
params.type = ASTAlterQuery::MODIFY_COLUMN;
|
|
|
|
}
|
2017-07-10 03:28:12 +00:00
|
|
|
else if (s_modify_primary_key.ignore(pos, expected))
|
2017-06-18 03:07:03 +00:00
|
|
|
{
|
2017-07-10 03:28:12 +00:00
|
|
|
if (pos->type != TokenType::OpeningRoundBracket)
|
2017-06-18 03:07:03 +00:00
|
|
|
return false;
|
2017-07-10 03:28:12 +00:00
|
|
|
++pos;
|
2016-05-17 20:11:43 +00:00
|
|
|
|
2017-07-10 03:28:12 +00:00
|
|
|
if (!ParserNotEmptyExpressionList(false).parse(pos, params.primary_key, expected))
|
2017-06-18 03:07:03 +00:00
|
|
|
return false;
|
2016-05-17 20:11:43 +00:00
|
|
|
|
2017-07-10 03:28:12 +00:00
|
|
|
if (pos->type != TokenType::ClosingRoundBracket)
|
2017-06-18 03:07:03 +00:00
|
|
|
return false;
|
2017-07-10 03:28:12 +00:00
|
|
|
++pos;
|
2016-05-05 18:28:46 +00:00
|
|
|
|
2017-06-18 03:07:03 +00:00
|
|
|
params.type = ASTAlterQuery::MODIFY_PRIMARY_KEY;
|
2017-04-01 07:20:54 +00:00
|
|
|
}
|
2018-02-02 16:02:43 +00:00
|
|
|
else if (s_delete_where.ignore(pos, expected))
|
|
|
|
{
|
|
|
|
if (!exp_elem.parse(pos, params.predicate, expected))
|
|
|
|
return false;
|
|
|
|
|
|
|
|
params.type = ASTAlterQuery::DELETE;
|
|
|
|
}
|
2017-04-01 07:20:54 +00:00
|
|
|
else
|
|
|
|
return false;
|
2013-08-07 13:07:42 +00:00
|
|
|
|
2017-07-10 03:28:12 +00:00
|
|
|
if (!s_comma.ignore(pos, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
parsing_finished = true;
|
2013-08-07 13:07:42 +00:00
|
|
|
|
2017-04-01 07:20:54 +00:00
|
|
|
query->addParameters(params);
|
|
|
|
}
|
|
|
|
while (!parsing_finished);
|
2013-08-07 13:07:42 +00:00
|
|
|
|
2017-04-21 12:39:28 +00:00
|
|
|
query->cluster = cluster_str;
|
2017-04-01 07:20:54 +00:00
|
|
|
node = query;
|
2013-08-07 13:07:42 +00:00
|
|
|
|
2017-04-01 07:20:54 +00:00
|
|
|
return true;
|
2013-08-07 13:07:42 +00:00
|
|
|
}
|
|
|
|
}
|