2017-04-01 09:19:00 +00:00
|
|
|
#include <Parsers/ASTIdentifier.h>
|
|
|
|
#include <Parsers/ASTDropQuery.h>
|
2011-11-05 23:31:19 +00:00
|
|
|
|
2017-04-01 09:19:00 +00:00
|
|
|
#include <Parsers/CommonParsers.h>
|
|
|
|
#include <Parsers/ParserDropQuery.h>
|
2011-11-05 23:31:19 +00:00
|
|
|
|
|
|
|
|
|
|
|
namespace DB
|
|
|
|
{
|
|
|
|
|
2018-05-05 16:14:06 +00:00
|
|
|
namespace ErrorCodes
|
|
|
|
{
|
|
|
|
extern const int SYNTAX_ERROR;
|
|
|
|
extern const int LOGICAL_ERROR;
|
|
|
|
}
|
2011-11-05 23:31:19 +00:00
|
|
|
|
2017-07-10 03:28:12 +00:00
|
|
|
bool ParserDropQuery::parseImpl(Pos & pos, ASTPtr & node, Expected & expected)
|
2011-11-05 23:31:19 +00:00
|
|
|
{
|
2017-06-18 03:07:03 +00:00
|
|
|
ParserKeyword s_drop("DROP");
|
|
|
|
ParserKeyword s_detach("DETACH");
|
2018-04-21 00:35:20 +00:00
|
|
|
ParserKeyword s_truncate("TRUNCATE");
|
|
|
|
|
|
|
|
if (s_drop.ignore(pos, expected))
|
|
|
|
return parseDropQuery(pos, node, expected);
|
|
|
|
else if (s_detach.ignore(pos, expected))
|
|
|
|
return parseDetachQuery(pos, node, expected);
|
|
|
|
else if (s_truncate.ignore(pos, expected))
|
|
|
|
return parseTruncateQuery(pos, node, expected);
|
|
|
|
else
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool ParserDropQuery::parseDetachQuery(Pos & pos, ASTPtr & node, Expected & expected)
|
|
|
|
{
|
|
|
|
if (parseDropQuery(pos, node, expected))
|
|
|
|
{
|
|
|
|
ASTDropQuery * drop_query = static_cast<ASTDropQuery*>(node.get());
|
|
|
|
drop_query->kind = ASTDropQuery::Kind::Detach;
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool ParserDropQuery::parseTruncateQuery(Pos & pos, ASTPtr & node, Expected & expected)
|
|
|
|
{
|
|
|
|
if (parseDropQuery(pos, node, expected))
|
|
|
|
{
|
|
|
|
ASTDropQuery * drop_query = static_cast<ASTDropQuery*>(node.get());
|
|
|
|
drop_query->kind = ASTDropQuery::Kind::Truncate;
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool ParserDropQuery::parseDropQuery(Pos & pos, ASTPtr & node, Expected & expected)
|
|
|
|
{
|
2018-01-27 16:17:05 +00:00
|
|
|
ParserKeyword s_temporary("TEMPORARY");
|
2017-06-18 03:07:03 +00:00
|
|
|
ParserKeyword s_table("TABLE");
|
|
|
|
ParserKeyword s_database("DATABASE");
|
2017-07-12 20:58:54 +00:00
|
|
|
ParserToken s_dot(TokenType::Dot);
|
2017-06-18 03:07:03 +00:00
|
|
|
ParserKeyword s_if_exists("IF EXISTS");
|
2017-04-01 07:20:54 +00:00
|
|
|
ParserIdentifier name_p;
|
|
|
|
|
|
|
|
ASTPtr database;
|
|
|
|
ASTPtr table;
|
2017-04-21 12:39:28 +00:00
|
|
|
String cluster_str;
|
2017-04-01 07:20:54 +00:00
|
|
|
bool if_exists = false;
|
2018-01-27 16:17:05 +00:00
|
|
|
bool temporary = false;
|
2017-04-01 07:20:54 +00:00
|
|
|
|
2017-07-10 03:28:12 +00:00
|
|
|
if (s_database.ignore(pos, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
{
|
2017-07-10 03:28:12 +00:00
|
|
|
if (s_if_exists.ignore(pos, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
if_exists = true;
|
|
|
|
|
2017-07-10 03:28:12 +00:00
|
|
|
if (!name_p.parse(pos, database, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
return false;
|
2017-05-30 11:49:17 +00:00
|
|
|
|
2017-07-10 03:28:12 +00:00
|
|
|
if (ParserKeyword{"ON"}.ignore(pos, expected))
|
2017-05-30 11:49:17 +00:00
|
|
|
{
|
2017-07-10 03:28:12 +00:00
|
|
|
if (!ASTQueryWithOnCluster::parse(pos, cluster_str, expected))
|
2017-05-30 11:49:17 +00:00
|
|
|
return false;
|
|
|
|
}
|
2017-04-01 07:20:54 +00:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2018-01-27 16:17:05 +00:00
|
|
|
if (s_temporary.ignore(pos, expected))
|
|
|
|
temporary = true;
|
|
|
|
|
2017-07-10 03:28:12 +00:00
|
|
|
if (!s_table.ignore(pos, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
return false;
|
|
|
|
|
2017-07-10 03:28:12 +00:00
|
|
|
if (s_if_exists.ignore(pos, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
if_exists = true;
|
|
|
|
|
2017-07-10 03:28:12 +00:00
|
|
|
if (!name_p.parse(pos, table, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
return false;
|
|
|
|
|
2017-07-10 03:28:12 +00:00
|
|
|
if (s_dot.ignore(pos, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
{
|
|
|
|
database = table;
|
2017-07-10 03:28:12 +00:00
|
|
|
if (!name_p.parse(pos, table, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
return false;
|
|
|
|
}
|
2017-04-13 16:12:56 +00:00
|
|
|
|
2017-07-10 03:28:12 +00:00
|
|
|
if (ParserKeyword{"ON"}.ignore(pos, expected))
|
2017-04-13 16:12:56 +00:00
|
|
|
{
|
2017-07-10 03:28:12 +00:00
|
|
|
if (!ASTQueryWithOnCluster::parse(pos, cluster_str, expected))
|
2017-04-13 16:12:56 +00:00
|
|
|
return false;
|
|
|
|
}
|
2017-04-01 07:20:54 +00:00
|
|
|
}
|
|
|
|
|
2018-02-26 03:37:08 +00:00
|
|
|
auto query = std::make_shared<ASTDropQuery>();
|
2017-04-01 07:20:54 +00:00
|
|
|
node = query;
|
|
|
|
|
2018-04-21 00:35:20 +00:00
|
|
|
query->kind = ASTDropQuery::Kind::Drop;
|
2017-04-01 07:20:54 +00:00
|
|
|
query->if_exists = if_exists;
|
2018-01-27 16:17:05 +00:00
|
|
|
query->temporary = temporary;
|
2019-01-14 18:15:04 +00:00
|
|
|
|
|
|
|
getIdentifierName(database, query->database);
|
|
|
|
getIdentifierName(table, query->table);
|
|
|
|
|
2017-04-21 12:39:28 +00:00
|
|
|
query->cluster = cluster_str;
|
2017-04-01 07:20:54 +00:00
|
|
|
|
|
|
|
return true;
|
2011-11-05 23:31:19 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
}
|