mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-14 10:22:10 +00:00
96 lines
2.3 KiB
C++
96 lines
2.3 KiB
C++
#include <Parsers/ASTIdentifier.h>
|
|
#include <Parsers/ASTRenameQuery.h>
|
|
|
|
#include <Parsers/CommonParsers.h>
|
|
#include <Parsers/ParserRenameQuery.h>
|
|
|
|
#include <Common/typeid_cast.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
|
|
/// Parse database.table or table.
|
|
static bool parseDatabaseAndTable(
|
|
ASTRenameQuery::Table & db_and_table, IParser::Pos & pos, IParser::Pos end, IParser::Pos & max_parsed_pos, Expected & expected)
|
|
{
|
|
ParserIdentifier name_p;
|
|
ParserWhiteSpaceOrComments ws;
|
|
ParserString s_dot(".");
|
|
|
|
ASTPtr database;
|
|
ASTPtr table;
|
|
|
|
ws.ignore(pos, end);
|
|
|
|
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);
|
|
}
|
|
|
|
db_and_table.database = database ? typeid_cast<const ASTIdentifier &>(*database).name : "";
|
|
db_and_table.table = typeid_cast<const ASTIdentifier &>(*table).name;
|
|
|
|
return true;
|
|
}
|
|
|
|
|
|
bool ParserRenameQuery::parseImpl(Pos & pos, Pos end, ASTPtr & node, Pos & max_parsed_pos, Expected & expected)
|
|
{
|
|
Pos begin = pos;
|
|
|
|
ParserWhiteSpaceOrComments ws;
|
|
ParserString s_rename("RENAME", true, true);
|
|
ParserString s_table("TABLE", true, true);
|
|
ParserString s_to("TO", true, true);
|
|
ParserString s_comma(",");
|
|
|
|
ws.ignore(pos, end);
|
|
|
|
if (!s_rename.ignore(pos, end, max_parsed_pos, expected))
|
|
return false;
|
|
|
|
ws.ignore(pos, end);
|
|
|
|
if (!s_table.ignore(pos, end, max_parsed_pos, expected))
|
|
return false;
|
|
|
|
ASTRenameQuery::Elements elements;
|
|
|
|
while (true)
|
|
{
|
|
ws.ignore(pos, end);
|
|
|
|
if (!elements.empty() && !s_comma.ignore(pos, end))
|
|
break;
|
|
|
|
ws.ignore(pos, end);
|
|
|
|
elements.push_back(ASTRenameQuery::Element());
|
|
|
|
if (!parseDatabaseAndTable(elements.back().from, pos, end, max_parsed_pos, expected)
|
|
|| !s_to.ignore(pos, end)
|
|
|| !parseDatabaseAndTable(elements.back().to, pos, end, max_parsed_pos, expected))
|
|
return false;
|
|
}
|
|
|
|
auto query = std::make_shared<ASTRenameQuery>(StringRange(begin, pos));
|
|
node = query;
|
|
|
|
query->elements = elements;
|
|
return true;
|
|
}
|
|
|
|
|
|
}
|