2017-04-01 09:19:00 +00:00
|
|
|
#include <Parsers/ASTIdentifier.h>
|
|
|
|
#include <Parsers/ASTLiteral.h>
|
|
|
|
#include <Parsers/ASTSetQuery.h>
|
2012-08-02 19:03:32 +00:00
|
|
|
|
2017-04-01 09:19:00 +00:00
|
|
|
#include <Parsers/CommonParsers.h>
|
|
|
|
#include <Parsers/ParserSetQuery.h>
|
2012-08-02 19:03:32 +00:00
|
|
|
|
2017-04-01 09:19:00 +00:00
|
|
|
#include <Common/typeid_cast.h>
|
2016-11-20 12:43:20 +00:00
|
|
|
|
2012-08-02 19:03:32 +00:00
|
|
|
|
|
|
|
namespace DB
|
|
|
|
{
|
|
|
|
|
|
|
|
|
2017-04-02 17:37:49 +00:00
|
|
|
/// Parse `name = value`.
|
2017-07-10 03:28:12 +00:00
|
|
|
static bool parseNameValuePair(ASTSetQuery::Change & change, IParser::Pos & pos, Expected & expected)
|
2012-08-02 19:03:32 +00:00
|
|
|
{
|
2017-04-01 07:20:54 +00:00
|
|
|
ParserIdentifier name_p;
|
|
|
|
ParserLiteral value_p;
|
2017-07-10 03:28:12 +00:00
|
|
|
ParserToken s_eq(TokenType::Equals);
|
2014-06-26 00:58:14 +00:00
|
|
|
|
2017-04-01 07:20:54 +00:00
|
|
|
ASTPtr name;
|
|
|
|
ASTPtr value;
|
2012-08-02 19:03:32 +00:00
|
|
|
|
2017-07-10 03:28:12 +00:00
|
|
|
if (!name_p.parse(pos, name, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
return false;
|
2012-08-02 19:03:32 +00:00
|
|
|
|
2017-07-10 03:28:12 +00:00
|
|
|
if (!s_eq.ignore(pos, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
return false;
|
2012-08-02 19:03:32 +00:00
|
|
|
|
2017-07-10 03:28:12 +00:00
|
|
|
if (!value_p.parse(pos, value, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
return false;
|
2012-08-02 19:03:32 +00:00
|
|
|
|
2017-04-01 07:20:54 +00:00
|
|
|
change.name = typeid_cast<const ASTIdentifier &>(*name).name;
|
|
|
|
change.value = typeid_cast<const ASTLiteral &>(*value).value;
|
2012-08-02 19:03:32 +00:00
|
|
|
|
2017-04-01 07:20:54 +00:00
|
|
|
return true;
|
2012-08-02 19:03:32 +00:00
|
|
|
}
|
|
|
|
|
2014-06-26 00:58:14 +00:00
|
|
|
|
2017-07-10 03:28:12 +00:00
|
|
|
bool ParserSetQuery::parseImpl(Pos & pos, ASTPtr & node, Expected & expected)
|
2012-08-02 19:03:32 +00:00
|
|
|
{
|
2017-04-01 07:20:54 +00:00
|
|
|
Pos begin = pos;
|
2012-08-02 19:03:32 +00:00
|
|
|
|
2017-07-10 03:28:12 +00:00
|
|
|
ParserToken s_comma(TokenType::Comma);
|
2014-06-26 00:58:14 +00:00
|
|
|
|
2017-04-01 07:20:54 +00:00
|
|
|
if (!parse_only_internals)
|
|
|
|
{
|
2017-06-18 03:07:03 +00:00
|
|
|
ParserKeyword s_set("SET");
|
2012-08-02 19:03:32 +00:00
|
|
|
|
2017-07-10 03:28:12 +00:00
|
|
|
if (!s_set.ignore(pos, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
return false;
|
|
|
|
}
|
2012-08-02 19:03:32 +00:00
|
|
|
|
2017-04-01 07:20:54 +00:00
|
|
|
ASTSetQuery::Changes changes;
|
2012-08-02 19:03:32 +00:00
|
|
|
|
2017-04-01 07:20:54 +00:00
|
|
|
while (true)
|
|
|
|
{
|
2017-07-10 03:28:12 +00:00
|
|
|
if (!changes.empty() && !s_comma.ignore(pos))
|
2017-04-01 07:20:54 +00:00
|
|
|
break;
|
2012-08-02 19:03:32 +00:00
|
|
|
|
2017-04-01 07:20:54 +00:00
|
|
|
changes.push_back(ASTSetQuery::Change());
|
2012-08-02 19:03:32 +00:00
|
|
|
|
2017-07-10 03:28:12 +00:00
|
|
|
if (!parseNameValuePair(changes.back(), pos, expected))
|
2017-04-01 07:20:54 +00:00
|
|
|
return false;
|
|
|
|
}
|
2012-08-02 19:03:32 +00:00
|
|
|
|
2017-04-01 07:20:54 +00:00
|
|
|
auto query = std::make_shared<ASTSetQuery>(StringRange(begin, pos));
|
|
|
|
node = query;
|
2012-08-02 19:03:32 +00:00
|
|
|
|
2017-09-17 18:49:43 +00:00
|
|
|
query->is_standalone = !parse_only_internals;
|
2017-04-01 07:20:54 +00:00
|
|
|
query->changes = changes;
|
2014-06-26 00:58:14 +00:00
|
|
|
|
2017-04-01 07:20:54 +00:00
|
|
|
return true;
|
2012-08-02 19:03:32 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
}
|