mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-12 09:22:05 +00:00
124 lines
3.2 KiB
C++
124 lines
3.2 KiB
C++
#include <Parsers/ParserCreateIndexQuery.h>
|
|
|
|
#include <Parsers/ASTCreateIndexQuery.h>
|
|
#include <Parsers/ASTFunction.h>
|
|
#include <Parsers/ASTIdentifier.h>
|
|
#include <Parsers/ASTIndexDeclaration.h>
|
|
#include <Parsers/ASTLiteral.h>
|
|
#include <Parsers/CommonParsers.h>
|
|
#include <Parsers/ExpressionListParsers.h>
|
|
#include <Parsers/ParserDataType.h>
|
|
#include <Parsers/parseDatabaseAndTableName.h>
|
|
|
|
namespace DB
|
|
{
|
|
|
|
bool ParserCreateIndexDeclaration::parseImpl(Pos & pos, ASTPtr & node, Expected & expected)
|
|
{
|
|
ParserKeyword s_type("TYPE");
|
|
ParserKeyword s_granularity("GRANULARITY");
|
|
|
|
ParserDataType data_type_p;
|
|
ParserExpression expression_p;
|
|
ParserUnsignedInteger granularity_p;
|
|
|
|
ASTPtr expr;
|
|
ASTPtr type;
|
|
ASTPtr granularity;
|
|
|
|
/// Skip name parser for SQL-standard CREATE INDEX
|
|
if (!expression_p.parse(pos, expr, expected))
|
|
return false;
|
|
|
|
if (!s_type.ignore(pos, expected))
|
|
return false;
|
|
|
|
if (!data_type_p.parse(pos, type, expected))
|
|
return false;
|
|
|
|
if (!s_granularity.ignore(pos, expected))
|
|
return false;
|
|
|
|
if (!granularity_p.parse(pos, granularity, expected))
|
|
return false;
|
|
|
|
auto index = std::make_shared<ASTIndexDeclaration>();
|
|
index->part_of_create_index_query = true;
|
|
index->granularity = granularity->as<ASTLiteral &>().value.safeGet<UInt64>();
|
|
index->set(index->expr, expr);
|
|
index->set(index->type, type);
|
|
node = index;
|
|
|
|
return true;
|
|
}
|
|
|
|
bool ParserCreateIndexQuery::parseImpl(IParser::Pos & pos, ASTPtr & node, Expected & expected)
|
|
{
|
|
auto query = std::make_shared<ASTCreateIndexQuery>();
|
|
node = query;
|
|
|
|
ParserKeyword s_create("CREATE");
|
|
ParserKeyword s_index("INDEX");
|
|
ParserKeyword s_if_not_exists("IF NOT EXISTS");
|
|
ParserKeyword s_on("ON");
|
|
ParserIdentifier index_name_p;
|
|
ParserCreateIndexDeclaration parser_create_idx_decl;
|
|
|
|
ASTPtr index_name;
|
|
ASTPtr index_decl;
|
|
|
|
String cluster_str;
|
|
bool if_not_exists = false;
|
|
|
|
if (!s_create.ignore(pos, expected))
|
|
return false;
|
|
|
|
if (!s_index.ignore(pos, expected))
|
|
return false;
|
|
|
|
if (s_if_not_exists.ignore(pos, expected))
|
|
if_not_exists = true;
|
|
|
|
if (!index_name_p.parse(pos, index_name, expected))
|
|
return false;
|
|
|
|
/// ON [db.] table_name
|
|
if (!s_on.ignore(pos, expected))
|
|
return false;
|
|
|
|
if (!parseDatabaseAndTableAsAST(pos, expected, query->database, query->table))
|
|
return false;
|
|
|
|
/// [ON cluster_name]
|
|
if (s_on.ignore(pos, expected))
|
|
{
|
|
if (!ASTQueryWithOnCluster::parse(pos, cluster_str, expected))
|
|
return false;
|
|
}
|
|
|
|
if (!parser_create_idx_decl.parse(pos, index_decl, expected))
|
|
return false;
|
|
|
|
auto & ast_index_decl = index_decl->as<ASTIndexDeclaration &>();
|
|
ast_index_decl.name = index_name->as<ASTIdentifier &>().name();
|
|
|
|
query->index_name = index_name;
|
|
query->children.push_back(index_name);
|
|
|
|
query->index_decl = index_decl;
|
|
query->children.push_back(index_decl);
|
|
|
|
query->if_not_exists = if_not_exists;
|
|
query->cluster = cluster_str;
|
|
|
|
if (query->database)
|
|
query->children.push_back(query->database);
|
|
|
|
if (query->table)
|
|
query->children.push_back(query->table);
|
|
|
|
return true;
|
|
}
|
|
|
|
}
|