ClickHouse/dbms/src/Parsers/ParserShowTablesQuery.cpp

104 lines
2.1 KiB
C++

#include <DB/Parsers/ASTLiteral.h>
#include <DB/Parsers/ASTIdentifier.h>
#include <DB/Parsers/ASTShowTablesQuery.h>
#include <DB/Parsers/CommonParsers.h>
#include <DB/Parsers/ParserShowTablesQuery.h>
#include <DB/Parsers/ExpressionElementParsers.h>
namespace DB
{
bool ParserShowTablesQuery::parseImpl(Pos & pos, Pos end, ASTPtr & node, String & expected)
{
Pos begin = pos;
ParserWhiteSpaceOrComments ws;
ParserString s_show("SHOW", true, true);
ParserString s_tables("TABLES", true, true);
ParserString s_databases("DATABASES", true, true);
ParserString s_from("FROM", true, true);
ParserString s_like("LIKE", true, true);
ParserString s_format("FORMAT", true, true);
ParserStringLiteral like_p;
ParserIdentifier name_p;
ASTPtr like;
ASTPtr database;
ASTPtr format;
ws.ignore(pos, end);
if (!s_show.ignore(pos, end, expected))
return false;
ws.ignore(pos, end);
if (s_databases.ignore(pos, end))
{
ASTShowTablesQuery * query = new ASTShowTablesQuery(StringRange(begin, pos));
query->databases = true;
node = query;
return true;
}
if (!s_tables.ignore(pos, end, expected))
return false;
ws.ignore(pos, end);
if (s_from.ignore(pos, end, expected))
{
ws.ignore(pos, end);
if (!name_p.parse(pos, end, database, expected))
return false;
}
ws.ignore(pos, end);
if (s_like.ignore(pos, end, expected))
{
ws.ignore(pos, end);
if (!like_p.parse(pos, end, like, expected))
return false;
}
ws.ignore(pos, end);
if (s_format.ignore(pos, end, expected))
{
ws.ignore(pos, end);
ParserIdentifier format_p;
if (!format_p.parse(pos, end, format, expected))
return false;
dynamic_cast<ASTIdentifier &>(*format).kind = ASTIdentifier::Format;
ws.ignore(pos, end);
}
ASTShowTablesQuery * query = new ASTShowTablesQuery(StringRange(begin, pos));
node = query;
if (database)
query->from = dynamic_cast<ASTIdentifier &>(*database).name;
if (like)
query->like = safeGet<const String &>(dynamic_cast<ASTLiteral &>(*like).value);
if (format)
{
query->format = format;
query->children.push_back(format);
}
return true;
}
}