mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-16 03:12:43 +00:00
131 lines
3.5 KiB
C++
131 lines
3.5 KiB
C++
#include <Parsers/ASTDropQuery.h>
|
|
|
|
#include <Parsers/CommonParsers.h>
|
|
#include <Parsers/ParserDropQuery.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
namespace
|
|
{
|
|
|
|
bool parseDropQuery(IParser::Pos & pos, ASTPtr & node, Expected & expected, const ASTDropQuery::Kind kind)
|
|
{
|
|
ParserKeyword s_temporary("TEMPORARY");
|
|
ParserKeyword s_table("TABLE");
|
|
ParserKeyword s_dictionary("DICTIONARY");
|
|
ParserKeyword s_view("VIEW");
|
|
ParserKeyword s_database("DATABASE");
|
|
ParserToken s_dot(TokenType::Dot);
|
|
ParserKeyword s_if_exists("IF EXISTS");
|
|
ParserIdentifier name_p(true);
|
|
ParserKeyword s_permanently("PERMANENTLY");
|
|
ParserKeyword s_no_delay("NO DELAY");
|
|
ParserKeyword s_sync("SYNC");
|
|
|
|
ASTPtr database;
|
|
ASTPtr table;
|
|
String cluster_str;
|
|
bool if_exists = false;
|
|
bool temporary = false;
|
|
bool is_dictionary = false;
|
|
bool is_view = false;
|
|
bool no_delay = false;
|
|
bool permanently = false;
|
|
|
|
if (s_database.ignore(pos, expected))
|
|
{
|
|
if (s_if_exists.ignore(pos, expected))
|
|
if_exists = true;
|
|
|
|
if (!name_p.parse(pos, database, expected))
|
|
return false;
|
|
}
|
|
else
|
|
{
|
|
if (s_view.ignore(pos, expected))
|
|
is_view = true;
|
|
else if (s_dictionary.ignore(pos, expected))
|
|
is_dictionary = true;
|
|
else if (s_temporary.ignore(pos, expected))
|
|
temporary = true;
|
|
|
|
/// for TRUNCATE queries TABLE keyword is assumed as default and can be skipped
|
|
if (!is_view && !is_dictionary && (!s_table.ignore(pos, expected) && kind != ASTDropQuery::Kind::Truncate))
|
|
{
|
|
return false;
|
|
}
|
|
|
|
if (s_if_exists.ignore(pos, expected))
|
|
if_exists = true;
|
|
|
|
if (!name_p.parse(pos, table, expected))
|
|
return false;
|
|
|
|
if (s_dot.ignore(pos, expected))
|
|
{
|
|
database = table;
|
|
if (!name_p.parse(pos, table, expected))
|
|
return false;
|
|
}
|
|
}
|
|
|
|
/// common for tables / dictionaries / databases
|
|
if (ParserKeyword{"ON"}.ignore(pos, expected))
|
|
{
|
|
if (!ASTQueryWithOnCluster::parse(pos, cluster_str, expected))
|
|
return false;
|
|
}
|
|
|
|
if (kind == ASTDropQuery::Kind::Detach && s_permanently.ignore(pos, expected))
|
|
permanently = true;
|
|
|
|
/// actually for TRUNCATE NO DELAY / SYNC means nothing
|
|
if (s_no_delay.ignore(pos, expected) || s_sync.ignore(pos, expected))
|
|
no_delay = true;
|
|
|
|
auto query = std::make_shared<ASTDropQuery>();
|
|
node = query;
|
|
|
|
query->kind = kind;
|
|
query->if_exists = if_exists;
|
|
query->temporary = temporary;
|
|
query->is_dictionary = is_dictionary;
|
|
query->is_view = is_view;
|
|
query->no_delay = no_delay;
|
|
query->permanently = permanently;
|
|
query->database = database;
|
|
query->table = table;
|
|
|
|
if (database)
|
|
query->children.push_back(database);
|
|
|
|
if (table)
|
|
query->children.push_back(table);
|
|
|
|
query->cluster = cluster_str;
|
|
|
|
return true;
|
|
}
|
|
|
|
}
|
|
|
|
bool ParserDropQuery::parseImpl(Pos & pos, ASTPtr & node, Expected & expected)
|
|
{
|
|
ParserKeyword s_drop("DROP");
|
|
ParserKeyword s_detach("DETACH");
|
|
ParserKeyword s_truncate("TRUNCATE");
|
|
|
|
if (s_drop.ignore(pos, expected))
|
|
return parseDropQuery(pos, node, expected, ASTDropQuery::Kind::Drop);
|
|
else if (s_detach.ignore(pos, expected))
|
|
return parseDropQuery(pos, node, expected, ASTDropQuery::Kind::Detach);
|
|
else if (s_truncate.ignore(pos, expected))
|
|
return parseDropQuery(pos, node, expected, ASTDropQuery::Kind::Truncate);
|
|
else
|
|
return false;
|
|
}
|
|
|
|
}
|