2017-04-01 09:19:00 +00:00
|
|
|
#include <Parsers/ParserAlterQuery.h>
|
|
|
|
#include <Parsers/CommonParsers.h>
|
|
|
|
#include <Parsers/ExpressionElementParsers.h>
|
|
|
|
#include <Parsers/ParserCreateQuery.h>
|
|
|
|
#include <Parsers/ASTIdentifier.h>
|
|
|
|
#include <Parsers/ExpressionElementParsers.h>
|
|
|
|
#include <Parsers/ASTAlterQuery.h>
|
|
|
|
#include <Parsers/ASTLiteral.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-04-01 07:20:54 +00:00
|
|
|
Pos begin = pos;
|
|
|
|
|
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_reshard("RESHARD");
|
|
|
|
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_copy("COPY");
|
|
|
|
ParserKeyword s_to("TO");
|
|
|
|
ParserKeyword s_using("USING");
|
|
|
|
ParserKeyword s_coordinate("COORDINATE");
|
|
|
|
ParserKeyword s_with("WITH");
|
|
|
|
ParserKeyword s_name("NAME");
|
2017-04-01 07:20:54 +00:00
|
|
|
|
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;
|
|
|
|
ParserLiteral parser_literal;
|
|
|
|
ParserUnsignedInteger parser_uint;
|
|
|
|
ParserStringLiteral parser_string_literal;
|
|
|
|
|
|
|
|
ASTPtr table;
|
|
|
|
ASTPtr database;
|
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;
|
|
|
|
|
2017-07-10 03:28:12 +00:00
|
|
|
if (!table_parser.parse(pos, database, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
return false;
|
|
|
|
|
|
|
|
/// Parse [db].name
|
2017-07-10 03:28:12 +00:00
|
|
|
if (s_dot.ignore(pos))
|
2017-04-01 07:20:54 +00:00
|
|
|
{
|
2017-07-10 03:28:12 +00:00
|
|
|
if (!table_parser.parse(pos, table, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
return false;
|
|
|
|
|
|
|
|
query->table = typeid_cast<ASTIdentifier &>(*table).name;
|
|
|
|
query->database = typeid_cast<ASTIdentifier &>(*database).name;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
table = database;
|
|
|
|
query->table = typeid_cast<ASTIdentifier &>(*table).name;
|
|
|
|
}
|
|
|
|
|
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-07-10 03:28:12 +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-07-10 03:28:12 +00:00
|
|
|
if (!parser_literal.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-07-10 03:28:12 +00:00
|
|
|
if (!parser_literal.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-07-10 03:28:12 +00:00
|
|
|
if (!parser_literal.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-07-10 03:28:12 +00:00
|
|
|
if (!parser_literal.parse(pos, params.partition, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
return false;
|
2014-08-07 11:46:01 +00:00
|
|
|
|
2017-04-01 07:20:54 +00:00
|
|
|
params.type = ASTAlterQuery::ATTACH_PARTITION;
|
|
|
|
}
|
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-07-10 03:28:12 +00:00
|
|
|
if (!parser_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-07-10 03:28:12 +00:00
|
|
|
if (!parser_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-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-07-10 03:28:12 +00:00
|
|
|
if (!parser_literal.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
|
|
|
}
|
2017-07-10 03:28:12 +00:00
|
|
|
else if (s_reshard.ignore(pos, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
{
|
2017-07-10 03:28:12 +00:00
|
|
|
ParserList weighted_zookeeper_paths_p(std::make_unique<ParserWeightedZooKeeperPath>(), std::make_unique<ParserToken>(TokenType::Comma), false);
|
2017-07-12 02:41:41 +00:00
|
|
|
ParserExpression parser_sharding_key_expr;
|
2017-04-01 07:20:54 +00:00
|
|
|
ParserStringLiteral parser_coordinator;
|
2016-01-28 01:00:27 +00:00
|
|
|
|
2017-07-10 03:28:12 +00:00
|
|
|
if (s_copy.ignore(pos, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
params.do_copy = true;
|
2016-03-25 11:48:45 +00:00
|
|
|
|
2017-07-10 03:28:12 +00:00
|
|
|
if (s_partition.ignore(pos, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
{
|
2017-07-10 03:28:12 +00:00
|
|
|
if (!parser_uint.parse(pos, params.partition, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
return false;
|
|
|
|
}
|
2016-01-28 01:00:27 +00:00
|
|
|
|
2017-07-10 03:28:12 +00:00
|
|
|
if (!s_to.ignore(pos, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
return false;
|
2016-01-28 01:00:27 +00:00
|
|
|
|
2017-07-10 03:28:12 +00:00
|
|
|
if (!weighted_zookeeper_paths_p.parse(pos, params.weighted_zookeeper_paths, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
return false;
|
2016-01-28 01:00:27 +00:00
|
|
|
|
2017-07-10 03:28:12 +00:00
|
|
|
if (!s_using.ignore(pos, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
return false;
|
2016-01-28 01:00:27 +00:00
|
|
|
|
2017-07-10 03:28:12 +00:00
|
|
|
if (!parser_sharding_key_expr.parse(pos, params.sharding_key_expr, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
return false;
|
2016-01-28 01:00:27 +00:00
|
|
|
|
2017-07-10 03:28:12 +00:00
|
|
|
if (s_coordinate.ignore(pos, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
{
|
2017-07-10 03:28:12 +00:00
|
|
|
if (!s_with.ignore(pos, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
return false;
|
2016-03-01 17:47:53 +00:00
|
|
|
|
2017-07-10 03:28:12 +00:00
|
|
|
if (!parser_coordinator.parse(pos, params.coordinator, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
return false;
|
|
|
|
}
|
2016-03-01 17:47:53 +00:00
|
|
|
|
2017-04-01 07:20:54 +00:00
|
|
|
params.type = ASTAlterQuery::RESHARD_PARTITION;
|
|
|
|
}
|
|
|
|
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-07-10 03:28:12 +00:00
|
|
|
query->range = StringRange(begin, pos);
|
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
|
|
|
}
|
|
|
|
}
|