2013-08-07 13:07:42 +00:00
|
|
|
#include <DB/Parsers/ParserAlterQuery.h>
|
|
|
|
#include <DB/Parsers/CommonParsers.h>
|
|
|
|
#include <DB/Parsers/ExpressionElementParsers.h>
|
|
|
|
#include <DB/Parsers/ParserCreateQuery.h>
|
|
|
|
#include <DB/Parsers/ASTIdentifier.h>
|
2014-05-20 16:46:33 +00:00
|
|
|
#include <DB/Parsers/ExpressionElementParsers.h>
|
2013-08-07 13:07:42 +00:00
|
|
|
#include <DB/Parsers/ASTAlterQuery.h>
|
|
|
|
|
|
|
|
namespace DB
|
|
|
|
{
|
2014-06-12 00:48:56 +00:00
|
|
|
bool ParserAlterQuery::parseImpl(Pos & pos, Pos end, ASTPtr & node, Expected & expected)
|
2013-08-07 13:07:42 +00:00
|
|
|
{
|
|
|
|
Pos begin = pos;
|
|
|
|
|
|
|
|
ParserWhiteSpaceOrComments ws;
|
|
|
|
ParserString s_alter("ALTER", true, true);
|
|
|
|
ParserString s_table("TABLE", true, true);
|
|
|
|
ParserString s_dot(".");
|
|
|
|
|
|
|
|
ParserString s_add("ADD", true, true);
|
|
|
|
ParserString s_column("COLUMN", true, true);
|
|
|
|
ParserString s_after("AFTER", true, true);
|
2014-02-28 08:22:15 +00:00
|
|
|
ParserString s_modify("MODIFY", true, true);
|
2013-08-07 13:07:42 +00:00
|
|
|
|
|
|
|
ParserString s_drop("DROP", true, true);
|
|
|
|
ParserString s_comma(",");
|
|
|
|
|
2014-05-20 18:01:36 +00:00
|
|
|
ParserIdentifier table_parser;
|
2014-05-20 16:46:33 +00:00
|
|
|
ParserCompoundIdentifier parser_name;
|
|
|
|
ParserCompoundNameTypePair parser_name_type;
|
2013-08-07 13:07:42 +00:00
|
|
|
|
|
|
|
ASTPtr table;
|
|
|
|
ASTPtr database;
|
|
|
|
ASTPtr col_type;
|
|
|
|
ASTPtr col_after;
|
|
|
|
ASTPtr col_drop;
|
|
|
|
|
2013-08-08 09:50:15 +00:00
|
|
|
ASTAlterQuery * query = new ASTAlterQuery();
|
2013-09-14 00:39:34 +00:00
|
|
|
ASTPtr query_ptr = query;
|
2013-08-07 13:07:42 +00:00
|
|
|
|
|
|
|
ws.ignore(pos, end);
|
|
|
|
if (!s_alter.ignore(pos, end, expected))
|
|
|
|
return false;
|
|
|
|
|
|
|
|
ws.ignore(pos, end);
|
|
|
|
if (!s_table.ignore(pos, end, expected))
|
|
|
|
return false;
|
|
|
|
|
|
|
|
ws.ignore(pos, end);
|
|
|
|
|
2014-05-20 18:01:36 +00:00
|
|
|
if (!table_parser.parse(pos, end, database, expected))
|
2013-08-07 13:07:42 +00:00
|
|
|
return false;
|
|
|
|
|
2013-08-09 00:18:03 +00:00
|
|
|
/// Parse [db].name
|
2013-08-07 13:07:42 +00:00
|
|
|
if (s_dot.ignore(pos, end))
|
|
|
|
{
|
2014-05-20 18:01:36 +00:00
|
|
|
if (!table_parser.parse(pos, end, table, expected))
|
2013-08-07 13:07:42 +00:00
|
|
|
return false;
|
|
|
|
|
|
|
|
query->table = dynamic_cast<ASTIdentifier &>(*table).name;
|
|
|
|
query->database = dynamic_cast<ASTIdentifier &>(*database).name;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
table = database;
|
|
|
|
query->table = dynamic_cast<ASTIdentifier &>(*table).name;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool parsing_finished = false;
|
|
|
|
do
|
|
|
|
{
|
|
|
|
ASTAlterQuery::Parameters params;
|
|
|
|
ws.ignore(pos, end);
|
|
|
|
|
|
|
|
if (s_add.ignore(pos, end, expected))
|
|
|
|
{
|
|
|
|
ws.ignore(pos, end);
|
|
|
|
s_column.ignore(pos, end, expected);
|
|
|
|
ws.ignore(pos, end);
|
|
|
|
|
|
|
|
parser_name_type.parse(pos, end, params.name_type, expected);
|
|
|
|
|
|
|
|
ws.ignore(pos, end);
|
|
|
|
if (s_after.ignore(pos, end, expected))
|
|
|
|
{
|
|
|
|
ws.ignore(pos, end);
|
|
|
|
|
|
|
|
if(!parser_name.parse(pos, end, params.column, expected))
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
params.type = ASTAlterQuery::ADD;
|
|
|
|
}
|
2014-02-28 08:22:15 +00:00
|
|
|
else if (s_drop.ignore(pos, end, expected))
|
2013-08-07 13:07:42 +00:00
|
|
|
{
|
2014-02-28 08:22:15 +00:00
|
|
|
ws.ignore(pos, end);
|
|
|
|
s_column.ignore(pos, end, expected);
|
|
|
|
ws.ignore(pos, end);
|
2013-08-07 13:07:42 +00:00
|
|
|
|
2014-02-28 08:22:15 +00:00
|
|
|
parser_name.parse(pos, end, params.column, expected);
|
2013-08-07 13:07:42 +00:00
|
|
|
|
2014-02-28 08:22:15 +00:00
|
|
|
params.type = ASTAlterQuery::DROP;
|
|
|
|
}
|
|
|
|
else if (s_modify.ignore(pos, end, expected))
|
|
|
|
{
|
|
|
|
ws.ignore(pos, end);
|
|
|
|
s_column.ignore(pos, end, expected);
|
|
|
|
ws.ignore(pos, end);
|
|
|
|
|
|
|
|
parser_name_type.parse(pos, end, params.name_type, expected);
|
|
|
|
|
|
|
|
ws.ignore(pos, end);
|
|
|
|
|
|
|
|
params.type = ASTAlterQuery::MODIFY;
|
2013-08-07 13:07:42 +00:00
|
|
|
}
|
2014-02-28 08:22:15 +00:00
|
|
|
else
|
|
|
|
return false;
|
2013-08-07 13:07:42 +00:00
|
|
|
|
|
|
|
ws.ignore(pos, end);
|
|
|
|
|
|
|
|
if (!s_comma.ignore(pos, end, expected))
|
|
|
|
{
|
|
|
|
ws.ignore(pos, end);
|
|
|
|
parsing_finished = true;
|
|
|
|
}
|
|
|
|
|
|
|
|
query->parameters.push_back(params);
|
|
|
|
}
|
|
|
|
while (!parsing_finished);
|
|
|
|
|
2013-08-08 09:50:15 +00:00
|
|
|
query->range = StringRange(begin, end);
|
2013-09-14 00:39:34 +00:00
|
|
|
node = query_ptr;
|
2013-08-07 13:07:42 +00:00
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
}
|