ClickHouse/dbms/src/Parsers/ParserCreateQuery.cpp

418 lines
9.7 KiB
C++
Raw Normal View History

2011-08-18 18:48:00 +00:00
#include <DB/Parsers/ASTFunction.h>
#include <DB/Parsers/ASTIdentifier.h>
#include <DB/Parsers/ASTExpressionList.h>
#include <DB/Parsers/ASTCreateQuery.h>
#include <DB/Parsers/ExpressionListParsers.h>
#include <DB/Parsers/ParserCreateQuery.h>
2011-11-01 15:16:04 +00:00
#include <DB/Parsers/ParserSelectQuery.h>
2011-08-18 18:48:00 +00:00
namespace DB
{
bool ParserNestedTable::parseImpl(Pos & pos, Pos end, ASTPtr & node, Expected & expected)
{
ParserWhiteSpaceOrComments ws;
ParserString open("(");
ParserString close(")");
ParserIdentifier name_p;
ParserNameTypePairList columns_p;
ASTPtr name;
ASTPtr columns;
Pos begin = pos;
/// Пока name == 'Nested', возможно потом появятся альтернативные вложенные структуры данных
if (!name_p.parse(pos, end, name, expected))
return false;
ws.ignore(pos, end);
if (!open.ignore(pos, end))
return false;
ws.ignore(pos, end);
if (!columns_p.parse(pos, end, columns, expected))
return false;
ws.ignore(pos, end);
if (!close.ignore(pos, end))
return false;
ASTFunction * func = new ASTFunction(StringRange(begin, pos));
node = func;
func->name = typeid_cast<ASTIdentifier &>(*name).name;
func->arguments = columns;
func->children.push_back(columns);
return true;
}
bool ParserIdentifierWithParameters::parseImpl(Pos & pos, Pos end, ASTPtr & node, Expected & expected)
{
Pos begin = pos;
ParserFunction function_or_array;
if (function_or_array.parse(pos, end, node, expected))
return true;
pos = begin;
ParserNestedTable nested;
if (nested.parse(pos, end, node, expected))
return true;
pos = begin;
return false;
}
2011-08-18 18:48:00 +00:00
bool ParserIdentifierWithOptionalParameters::parseImpl(Pos & pos, Pos end, ASTPtr & node, Expected & expected)
2011-08-18 18:48:00 +00:00
{
ParserIdentifier non_parametric;
ParserIdentifierWithParameters parametric;
2011-08-18 18:48:00 +00:00
Pos begin = pos;
if (parametric.parse(pos, end, node, expected))
{
return true;
}
pos = begin;
ASTPtr ident;
if (non_parametric.parse(pos, end, ident, expected))
{
ASTFunction * func = new ASTFunction(StringRange(begin, pos));
node = func;
func->name = typeid_cast<ASTIdentifier &>(*ident).name;
2011-08-18 18:48:00 +00:00
return true;
}
pos = begin;
return false;
}
bool ParserNameTypePairList::parseImpl(Pos & pos, Pos end, ASTPtr & node, Expected & expected)
{
return ParserList(ParserPtr(new ParserNameTypePair), ParserPtr(new ParserString(",")), false).parse(pos, end, node, expected);
}
bool ParserEngine::parseImpl(Pos & pos, Pos end, ASTPtr & storage, Expected & expected)
2011-11-01 15:16:04 +00:00
{
ParserWhiteSpaceOrComments ws;
2011-11-05 23:31:19 +00:00
ParserString s_engine("ENGINE", true, true);
2011-11-01 15:16:04 +00:00
ParserString s_eq("=");
ParserIdentifierWithOptionalParameters storage_p;
ws.ignore(pos, end);
2011-11-01 15:16:04 +00:00
if (s_engine.ignore(pos, end, expected))
{
ws.ignore(pos, end);
if (!s_eq.ignore(pos, end, expected))
return false;
ws.ignore(pos, end);
if (!storage_p.parse(pos, end, storage, expected))
return false;
ws.ignore(pos, end);
}
return true;
}
bool ParserCreateQuery::parseImpl(Pos & pos, Pos end, ASTPtr & node, Expected & expected)
2011-08-18 18:48:00 +00:00
{
Pos begin = pos;
ParserWhiteSpaceOrComments ws;
ParserString s_create("CREATE", true, true);
ParserString s_temporary("TEMPORARY", true, true);
2011-08-18 18:48:00 +00:00
ParserString s_attach("ATTACH", true, true);
ParserString s_table("TABLE", true, true);
2011-11-05 23:31:19 +00:00
ParserString s_database("DATABASE", true, true);
2011-10-30 05:19:41 +00:00
ParserString s_dot(".");
2011-08-18 18:48:00 +00:00
ParserString s_lparen("(");
ParserString s_rparen(")");
2011-11-05 23:31:19 +00:00
ParserString s_if("IF", true, true);
ParserString s_not("NOT", true, true);
ParserString s_exists("EXISTS", true, true);
ParserString s_as("AS", true, true);
ParserString s_select("SELECT", true, true);
ParserString s_view("VIEW", true, true);
ParserString s_materialized("MATERIALIZED", true, true);
ParserString s_populate("POPULATE", true, true);
2011-11-01 15:16:04 +00:00
ParserEngine engine_p;
2011-08-18 18:48:00 +00:00
ParserIdentifier name_p;
ParserNameTypePairList columns_p;
2011-08-18 18:48:00 +00:00
2011-10-30 05:19:41 +00:00
ASTPtr database;
ASTPtr table;
2011-08-18 18:48:00 +00:00
ASTPtr columns;
ASTPtr storage;
ASTPtr inner_storage;
2011-10-31 17:30:44 +00:00
ASTPtr as_database;
ASTPtr as_table;
2011-11-01 15:16:04 +00:00
ASTPtr select;
2011-08-18 18:48:00 +00:00
bool attach = false;
2011-08-19 18:31:14 +00:00
bool if_not_exists = false;
bool is_view = false;
bool is_materialized_view = false;
bool is_populate = false;
bool is_temporary = false;
2011-08-18 18:48:00 +00:00
ws.ignore(pos, end);
if (!s_create.ignore(pos, end, expected))
{
if (s_attach.ignore(pos, end, expected))
attach = true;
else
return false;
}
ws.ignore(pos, end);
if (s_temporary.ignore(pos, end, expected))
{
is_temporary = true;
ws.ignore(pos, end);
}
2011-11-05 23:31:19 +00:00
if (s_database.ignore(pos, end, expected))
2011-10-30 05:19:41 +00:00
{
ws.ignore(pos, end);
2011-11-05 23:31:19 +00:00
if (s_if.ignore(pos, end, expected)
&& ws.ignore(pos, end)
&& s_not.ignore(pos, end, expected)
&& ws.ignore(pos, end)
&& s_exists.ignore(pos, end, expected)
&& ws.ignore(pos, end))
if_not_exists = true;
2011-08-18 18:48:00 +00:00
2011-11-05 23:31:19 +00:00
if (!name_p.parse(pos, end, database, expected))
return false;
}
else if (s_table.ignore(pos, end, expected))
2011-11-05 23:31:19 +00:00
{
2011-10-31 17:30:44 +00:00
ws.ignore(pos, end);
2011-08-18 18:48:00 +00:00
2011-11-05 23:31:19 +00:00
if (s_if.ignore(pos, end, expected)
&& ws.ignore(pos, end)
&& s_not.ignore(pos, end, expected)
&& ws.ignore(pos, end)
&& s_exists.ignore(pos, end, expected)
&& ws.ignore(pos, end))
if_not_exists = true;
if (!name_p.parse(pos, end, table, expected))
2011-10-31 17:30:44 +00:00
return false;
2011-08-18 18:48:00 +00:00
2011-10-31 17:30:44 +00:00
ws.ignore(pos, end);
2011-11-01 15:16:04 +00:00
2011-11-05 23:31:19 +00:00
if (s_dot.ignore(pos, end, expected))
{
database = table;
if (!name_p.parse(pos, end, table, expected))
return false;
2011-11-01 15:38:01 +00:00
2011-11-05 23:31:19 +00:00
ws.ignore(pos, end);
}
2011-08-18 18:48:00 +00:00
2011-11-05 23:31:19 +00:00
/// Список столбцов
if (s_lparen.ignore(pos, end, expected))
2011-10-31 17:30:44 +00:00
{
2011-11-05 23:31:19 +00:00
ws.ignore(pos, end);
if (!columns_p.parse(pos, end, columns, expected))
return false;
ws.ignore(pos, end);
if (!s_rparen.ignore(pos, end, expected))
return false;
2011-11-05 23:31:19 +00:00
ws.ignore(pos, end);
if (!engine_p.parse(pos, end, storage, expected))
return false;
/// Для engine VIEW необходимо так же считать запрос AS SELECT
if (storage && (typeid_cast<ASTFunction &>(*storage).name == "View"
|| typeid_cast<ASTFunction &>(*storage).name == "MaterializedView"))
{
if (!s_as.ignore(pos, end, expected))
return false;
ws.ignore(pos, end);
Pos before_select = pos;
if (!s_select.ignore(pos, end, expected))
return false;
pos = before_select;
ParserSelectQuery select_p;
select_p.parse(pos, end, select, expected);
}
2011-11-01 15:16:04 +00:00
}
else
{
2011-11-05 23:31:19 +00:00
engine_p.parse(pos, end, storage, expected);
if (!s_as.ignore(pos, end, expected))
2011-10-31 17:30:44 +00:00
return false;
2011-08-18 18:48:00 +00:00
2011-10-31 17:30:44 +00:00
ws.ignore(pos, end);
2011-11-01 15:16:04 +00:00
2011-11-05 23:31:19 +00:00
/// AS SELECT ...
Pos before_select = pos;
if (s_select.ignore(pos, end, expected))
{
pos = before_select;
ParserSelectQuery select_p;
select_p.parse(pos, end, select, expected);
}
else
2011-11-01 15:16:04 +00:00
{
2011-11-05 23:31:19 +00:00
/// AS [db.]table
2011-11-01 15:16:04 +00:00
if (!name_p.parse(pos, end, as_table, expected))
return false;
ws.ignore(pos, end);
2011-11-05 23:31:19 +00:00
if (s_dot.ignore(pos, end, expected))
{
as_database = as_table;
if (!name_p.parse(pos, end, as_table, expected))
return false;
ws.ignore(pos, end);
}
2012-07-12 20:06:45 +00:00
ws.ignore(pos, end);
/// Опционально - может быть указана ENGINE.
engine_p.parse(pos, end, storage, expected);
2011-11-01 15:16:04 +00:00
}
2011-10-31 17:30:44 +00:00
}
2014-04-24 18:49:07 +00:00
}
else
{
/// VIEW or MATERIALIZED VIEW
if (s_materialized.ignore(pos, end, expected) && ws.ignore(pos, end, expected))
is_materialized_view = true;
else
is_view = true;
if (!s_view.ignore(pos, end, expected) || !ws.ignore(pos, end, expected))
return false;
if (s_if.ignore(pos, end, expected)
&& ws.ignore(pos, end)
&& s_not.ignore(pos, end, expected)
&& ws.ignore(pos, end)
&& s_exists.ignore(pos, end, expected)
&& ws.ignore(pos, end))
if_not_exists = true;
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);
}
2014-05-12 17:27:48 +00:00
/// Опционально - может быть указан список столбцов. Он должен полностью соответствовать SELECT-у.
if (s_lparen.ignore(pos, end, expected))
{
ws.ignore(pos, end);
if (!columns_p.parse(pos, end, columns, expected))
return false;
ws.ignore(pos, end);
if (!s_rparen.ignore(pos, end, expected))
return false;
}
/// Опционально - может быть указана внутренняя ENGINE для MATERIALIZED VIEW
engine_p.parse(pos, end, inner_storage, expected);
ws.ignore(pos, end);
if (s_populate.ignore(pos, end, expected))
is_populate = true;
ws.ignore(pos, end);
/// AS SELECT ...
if (!s_as.ignore(pos, end, expected))
return false;
ws.ignore(pos, end);
Pos before_select = pos;
if (!s_select.ignore(pos, end, expected))
return false;
pos = before_select;
ParserSelectQuery select_p;
select_p.parse(pos, end, select, expected);
2011-10-31 17:30:44 +00:00
}
2011-08-18 18:48:00 +00:00
2011-11-05 23:31:19 +00:00
ws.ignore(pos, end);
2011-08-18 18:48:00 +00:00
ASTCreateQuery * query = new ASTCreateQuery(StringRange(begin, pos));
node = query;
query->attach = attach;
2011-08-19 18:31:14 +00:00
query->if_not_exists = if_not_exists;
query->is_view = is_view;
query->is_materialized_view = is_materialized_view;
query->is_populate = is_populate;
query->is_temporary = is_temporary;
2011-10-31 06:37:12 +00:00
if (database)
query->database = typeid_cast<ASTIdentifier &>(*database).name;
2011-11-05 23:31:19 +00:00
if (table)
query->table = typeid_cast<ASTIdentifier &>(*table).name;
if (inner_storage)
query->inner_storage = inner_storage;
2011-08-18 18:48:00 +00:00
query->columns = columns;
query->storage = storage;
2011-10-31 17:30:44 +00:00
if (as_database)
query->as_database = typeid_cast<ASTIdentifier &>(*as_database).name;
2011-10-31 17:30:44 +00:00
if (as_table)
query->as_table = typeid_cast<ASTIdentifier &>(*as_table).name;
2011-11-01 15:16:04 +00:00
query->select = select;
2011-08-18 18:48:00 +00:00
2012-12-27 17:40:54 +00:00
if (columns)
query->children.push_back(columns);
2011-11-01 15:16:04 +00:00
if (storage)
query->children.push_back(storage);
if (select)
query->children.push_back(select);
if (inner_storage)
query->children.push_back(inner_storage);
2011-08-18 18:48:00 +00:00
return true;
}
}