ClickHouse/dbms/src/Parsers/ParserSetQuery.cpp

100 lines
1.9 KiB
C++
Raw Normal View History

2012-08-02 19:03:32 +00:00
#include <DB/Parsers/ASTIdentifier.h>
#include <DB/Parsers/ASTLiteral.h>
#include <DB/Parsers/ASTSetQuery.h>
#include <DB/Parsers/CommonParsers.h>
#include <DB/Parsers/ParserSetQuery.h>
namespace DB
{
/// Парсит name = value.
static bool parseNameValuePair(ASTSetQuery::Change & change, IParser::Pos & pos, IParser::Pos end, IParser::Pos & max_parsed_pos, Expected & expected)
2012-08-02 19:03:32 +00:00
{
ParserIdentifier name_p;
ParserLiteral value_p;
ParserWhiteSpaceOrComments ws;
ParserString s_eq("=");
2012-08-02 19:03:32 +00:00
ASTPtr name;
ASTPtr value;
ws.ignore(pos, end);
if (!name_p.parse(pos, end, name, max_parsed_pos, expected))
2012-08-02 19:03:32 +00:00
return false;
ws.ignore(pos, end);
if (!s_eq.ignore(pos, end, max_parsed_pos, expected))
2012-08-02 19:03:32 +00:00
return false;
ws.ignore(pos, end);
if (!value_p.parse(pos, end, value, max_parsed_pos, expected))
2012-08-02 19:03:32 +00:00
return false;
ws.ignore(pos, end);
change.name = typeid_cast<const ASTIdentifier &>(*name).name;
change.value = typeid_cast<const ASTLiteral &>(*value).value;
2012-08-02 19:03:32 +00:00
return true;
}
bool ParserSetQuery::parseImpl(Pos & pos, Pos end, ASTPtr & node, Pos & max_parsed_pos, Expected & expected)
2012-08-02 19:03:32 +00:00
{
Pos begin = pos;
ParserWhiteSpaceOrComments ws;
ParserString s_comma(",");
bool global = false;
2012-08-02 19:03:32 +00:00
if (!parse_only_internals)
{
ParserString s_set("SET", true, true);
ParserString s_global("GLOBAL", true, true);
2012-08-02 19:03:32 +00:00
ws.ignore(pos, end);
2012-08-02 19:03:32 +00:00
if (!s_set.ignore(pos, end, max_parsed_pos, expected))
return false;
ws.ignore(pos, end);
global = s_global.ignore(pos, end, max_parsed_pos, expected);
}
2012-08-02 19:03:32 +00:00
ASTSetQuery::Changes changes;
while (true)
{
ws.ignore(pos, end);
2012-08-02 19:03:32 +00:00
if (!changes.empty() && !s_comma.ignore(pos, end))
break;
ws.ignore(pos, end);
changes.push_back(ASTSetQuery::Change());
if (!parseNameValuePair(changes.back(), pos, end, max_parsed_pos, expected))
2012-08-02 19:03:32 +00:00
return false;
}
auto query = std::make_shared<ASTSetQuery>(StringRange(begin, pos));
2012-08-02 19:03:32 +00:00
node = query;
query->changes = changes;
query->global = global;
2012-08-02 19:03:32 +00:00
return true;
}
}