mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-03 21:12:28 +00:00
103 lines
2.2 KiB
C++
103 lines
2.2 KiB
C++
#include <DB/Parsers/ASTIdentifier.h>
|
|
#include <DB/Parsers/ASTDropQuery.h>
|
|
|
|
#include <DB/Parsers/CommonParsers.h>
|
|
#include <DB/Parsers/ParserDropQuery.h>
|
|
|
|
#include <DB/Common/typeid_cast.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
|
|
bool ParserDropQuery::parseImpl(Pos & pos, Pos end, ASTPtr & node, Pos & max_parsed_pos, Expected & expected)
|
|
{
|
|
Pos begin = pos;
|
|
|
|
ParserWhiteSpaceOrComments ws;
|
|
ParserString s_drop("DROP", true, true);
|
|
ParserString s_detach("DETACH", true, true);
|
|
ParserString s_table("TABLE", true, true);
|
|
ParserString s_database("DATABASE", true, true);
|
|
ParserString s_dot(".");
|
|
ParserString s_if("IF", true, true);
|
|
ParserString s_exists("EXISTS", true, true);
|
|
ParserIdentifier name_p;
|
|
|
|
ASTPtr database;
|
|
ASTPtr table;
|
|
bool detach = false;
|
|
bool if_exists = false;
|
|
|
|
ws.ignore(pos, end);
|
|
|
|
if (!s_drop.ignore(pos, end, max_parsed_pos, expected))
|
|
{
|
|
if (s_detach.ignore(pos, end, max_parsed_pos, expected))
|
|
detach = true;
|
|
else
|
|
return false;
|
|
}
|
|
|
|
ws.ignore(pos, end);
|
|
|
|
if (s_database.ignore(pos, end, max_parsed_pos, expected))
|
|
{
|
|
ws.ignore(pos, end);
|
|
|
|
if (s_if.ignore(pos, end, max_parsed_pos, expected)
|
|
&& ws.ignore(pos, end)
|
|
&& s_exists.ignore(pos, end, max_parsed_pos, expected)
|
|
&& ws.ignore(pos, end))
|
|
if_exists = true;
|
|
|
|
if (!name_p.parse(pos, end, database, max_parsed_pos, expected))
|
|
return false;
|
|
}
|
|
else
|
|
{
|
|
if (!s_table.ignore(pos, end, max_parsed_pos, expected))
|
|
return false;
|
|
|
|
ws.ignore(pos, end);
|
|
|
|
if (s_if.ignore(pos, end, max_parsed_pos, expected)
|
|
&& ws.ignore(pos, end)
|
|
&& s_exists.ignore(pos, end, max_parsed_pos, expected)
|
|
&& ws.ignore(pos, end))
|
|
if_exists = true;
|
|
|
|
if (!name_p.parse(pos, end, table, max_parsed_pos, expected))
|
|
return false;
|
|
|
|
ws.ignore(pos, end);
|
|
|
|
if (s_dot.ignore(pos, end, max_parsed_pos, expected))
|
|
{
|
|
database = table;
|
|
if (!name_p.parse(pos, end, table, max_parsed_pos, expected))
|
|
return false;
|
|
|
|
ws.ignore(pos, end);
|
|
}
|
|
}
|
|
|
|
ws.ignore(pos, end);
|
|
|
|
auto query = std::make_shared<ASTDropQuery>(StringRange(begin, pos));
|
|
node = query;
|
|
|
|
query->detach = detach;
|
|
query->if_exists = if_exists;
|
|
if (database)
|
|
query->database = typeid_cast<ASTIdentifier &>(*database).name;
|
|
if (table)
|
|
query->table = typeid_cast<ASTIdentifier &>(*table).name;
|
|
|
|
return true;
|
|
}
|
|
|
|
|
|
}
|