mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-10-02 14:40:50 +00:00
124 lines
2.7 KiB
C++
124 lines
2.7 KiB
C++
|
#include <DB/Parsers/ASTIdentifier.h>
|
||
|
#include <DB/Parsers/ASTSelectQuery.h>
|
||
|
#include <DB/Parsers/ASTInsertQuery.h>
|
||
|
|
||
|
#include <DB/Parsers/CommonParsers.h>
|
||
|
#include <DB/Parsers/ExpressionElementParsers.h>
|
||
|
#include <DB/Parsers/ExpressionListParsers.h>
|
||
|
#include <DB/Parsers/ParserSelectQuery.h>
|
||
|
#include <DB/Parsers/ParserInsertQuery.h>
|
||
|
|
||
|
|
||
|
namespace DB
|
||
|
{
|
||
|
|
||
|
|
||
|
bool ParserInsertQuery::parseImpl(Pos & pos, Pos end, ASTPtr & node, String & expected)
|
||
|
{
|
||
|
Pos begin = pos;
|
||
|
|
||
|
ParserWhiteSpaceOrComments ws;
|
||
|
ParserString s_insert("INSERT", true, true);
|
||
|
ParserString s_into("INTO", true, true);
|
||
|
ParserString s_dot(".");
|
||
|
ParserString s_values("VALUES", true, true);
|
||
|
ParserString s_format("FORMAT", true, true);
|
||
|
ParserString s_select("SELECT", true, true);
|
||
|
ParserString s_lparen("(");
|
||
|
ParserString s_rparen(")");
|
||
|
ParserIdentifier name_p;
|
||
|
ParserList columns_p(new ParserIdentifier, new ParserString(","), false);
|
||
|
|
||
|
ASTPtr database;
|
||
|
ASTPtr table;
|
||
|
ASTPtr columns;
|
||
|
ASTPtr format;
|
||
|
ASTPtr select;
|
||
|
/// Данные для вставки
|
||
|
const char * data = NULL;
|
||
|
|
||
|
ws.ignore(pos, end);
|
||
|
|
||
|
/// INSERT INTO
|
||
|
if (!s_insert.ignore(pos, end, expected)
|
||
|
|| !ws.ignore(pos, end)
|
||
|
|| !s_into.ignore(pos, end, expected))
|
||
|
return false;
|
||
|
|
||
|
ws.ignore(pos, end);
|
||
|
|
||
|
if (!name_p.parse(pos, end, table, expected))
|
||
|
return false;
|
||
|
|
||
|
ws.ignore(pos, end);
|
||
|
|
||
|
if (s_dot.ignore(pos, end, expected))
|
||
|
{
|
||
|
database = table;
|
||
|
if (!name_p.parse(pos, end, table, expected))
|
||
|
return false;
|
||
|
|
||
|
ws.ignore(pos, end);
|
||
|
}
|
||
|
|
||
|
ws.ignore(pos, end);
|
||
|
|
||
|
/// Есть ли список столбцов
|
||
|
if (s_lparen.ignore(pos, end, expected)
|
||
|
&& (!columns_p.parse(pos, end, columns, expected)
|
||
|
|| !ws.ignore(pos, end)
|
||
|
|| !s_rparen.ignore(pos, end, expected)))
|
||
|
return false;
|
||
|
|
||
|
ws.ignore(pos, end);
|
||
|
|
||
|
Pos before_select = pos;
|
||
|
|
||
|
/// VALUES или FORMAT или SELECT
|
||
|
if (s_values.ignore(pos, end, expected))
|
||
|
{
|
||
|
ws.ignore(pos, end);
|
||
|
data = pos;
|
||
|
}
|
||
|
else if (s_format.ignore(pos, end, expected))
|
||
|
{
|
||
|
if (!name_p.parse(pos, end, format, expected))
|
||
|
return false;
|
||
|
|
||
|
ws.ignore(pos, end);
|
||
|
data = pos;
|
||
|
}
|
||
|
else if (s_select.ignore(pos, end, expected))
|
||
|
{
|
||
|
pos = before_select;
|
||
|
ParserSelectQuery select_p;
|
||
|
select_p.parse(pos, end, select, expected);
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
expected = "VALUES or FORMAT or SELECT";
|
||
|
return false;
|
||
|
}
|
||
|
|
||
|
ASTInsertQuery * query = new ASTInsertQuery(StringRange(begin, pos));
|
||
|
node = query;
|
||
|
|
||
|
query->database = dynamic_cast<ASTIdentifier &>(*database).name;
|
||
|
query->table = dynamic_cast<ASTIdentifier &>(*table).name;
|
||
|
query->format = dynamic_cast<ASTIdentifier &>(*format).name;
|
||
|
query->columns = columns;
|
||
|
query->select = select;
|
||
|
query->data = data;
|
||
|
query->end = end;
|
||
|
|
||
|
if (columns)
|
||
|
query->children.push_back(columns);
|
||
|
if (select)
|
||
|
query->children.push_back(select);
|
||
|
|
||
|
return true;
|
||
|
}
|
||
|
|
||
|
|
||
|
}
|