mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-14 18:32:29 +00:00
121 lines
3.7 KiB
C++
121 lines
3.7 KiB
C++
#include <Parsers/ParserQueryWithOutput.h>
|
|
#include <Parsers/ParserShowTablesQuery.h>
|
|
#include <Parsers/ParserSelectWithUnionQuery.h>
|
|
#include <Parsers/ParserTablePropertiesQuery.h>
|
|
#include <Parsers/ParserDescribeTableQuery.h>
|
|
#include <Parsers/ParserShowProcesslistQuery.h>
|
|
#include <Parsers/ParserCheckQuery.h>
|
|
#include <Parsers/ParserCreateQuery.h>
|
|
#include <Parsers/ParserRenameQuery.h>
|
|
#include <Parsers/ParserAlterQuery.h>
|
|
#include <Parsers/ParserDropQuery.h>
|
|
#include <Parsers/ParserKillQueryQuery.h>
|
|
#include <Parsers/ParserOptimizeQuery.h>
|
|
#include <Parsers/ParserWatchQuery.h>
|
|
#include <Parsers/ParserSetQuery.h>
|
|
#include <Parsers/ASTExplainQuery.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
bool ParserQueryWithOutput::parseImpl(Pos & pos, ASTPtr & node, Expected & expected)
|
|
{
|
|
ParserShowTablesQuery show_tables_p;
|
|
ParserSelectWithUnionQuery select_p;
|
|
ParserTablePropertiesQuery table_p;
|
|
ParserDescribeTableQuery describe_table_p;
|
|
ParserShowProcesslistQuery show_processlist_p;
|
|
ParserCreateQuery create_p;
|
|
ParserAlterQuery alter_p;
|
|
ParserRenameQuery rename_p;
|
|
ParserDropQuery drop_p;
|
|
ParserCheckQuery check_p;
|
|
ParserOptimizeQuery optimize_p;
|
|
ParserKillQueryQuery kill_query_p;
|
|
ParserWatchQuery watch_p;
|
|
|
|
ASTPtr query;
|
|
|
|
ParserKeyword s_ast("AST");
|
|
ParserKeyword s_analyze("ANALYZE");
|
|
bool explain_ast = false;
|
|
bool analyze_syntax = false;
|
|
|
|
if (enable_explain && s_ast.ignore(pos, expected))
|
|
explain_ast = true;
|
|
|
|
if (enable_explain && s_analyze.ignore(pos, expected))
|
|
analyze_syntax = true;
|
|
|
|
bool parsed = select_p.parse(pos, query, expected)
|
|
|| show_tables_p.parse(pos, query, expected)
|
|
|| table_p.parse(pos, query, expected)
|
|
|| describe_table_p.parse(pos, query, expected)
|
|
|| show_processlist_p.parse(pos, query, expected)
|
|
|| create_p.parse(pos, query, expected)
|
|
|| alter_p.parse(pos, query, expected)
|
|
|| rename_p.parse(pos, query, expected)
|
|
|| drop_p.parse(pos, query, expected)
|
|
|| check_p.parse(pos, query, expected)
|
|
|| kill_query_p.parse(pos, query, expected)
|
|
|| optimize_p.parse(pos, query, expected)
|
|
|| watch_p.parse(pos, query, expected);
|
|
|
|
if (!parsed)
|
|
return false;
|
|
|
|
/// FIXME: try to prettify this cast using `as<>()`
|
|
auto & query_with_output = dynamic_cast<ASTQueryWithOutput &>(*query);
|
|
|
|
ParserKeyword s_into_outfile("INTO OUTFILE");
|
|
if (s_into_outfile.ignore(pos, expected))
|
|
{
|
|
ParserStringLiteral out_file_p;
|
|
if (!out_file_p.parse(pos, query_with_output.out_file, expected))
|
|
return false;
|
|
|
|
query_with_output.children.push_back(query_with_output.out_file);
|
|
}
|
|
|
|
ParserKeyword s_format("FORMAT");
|
|
|
|
if (s_format.ignore(pos, expected))
|
|
{
|
|
ParserIdentifier format_p;
|
|
|
|
if (!format_p.parse(pos, query_with_output.format, expected))
|
|
return false;
|
|
setIdentifierSpecial(query_with_output.format);
|
|
|
|
query_with_output.children.push_back(query_with_output.format);
|
|
}
|
|
|
|
// SETTINGS key1 = value1, key2 = value2, ...
|
|
ParserKeyword s_settings("SETTINGS");
|
|
if (s_settings.ignore(pos, expected))
|
|
{
|
|
ParserSetQuery parser_settings(true);
|
|
if (!parser_settings.parse(pos, query_with_output.settings_ast, expected))
|
|
return false;
|
|
query_with_output.children.push_back(query_with_output.settings_ast);
|
|
}
|
|
|
|
if (explain_ast)
|
|
{
|
|
node = std::make_shared<ASTExplainQuery>(ASTExplainQuery::ParsedAST);
|
|
node->children.push_back(query);
|
|
}
|
|
else if (analyze_syntax)
|
|
{
|
|
node = std::make_shared<ASTExplainQuery>(ASTExplainQuery::AnalyzedSyntax);
|
|
node->children.push_back(query);
|
|
}
|
|
else
|
|
node = query;
|
|
|
|
return true;
|
|
}
|
|
|
|
}
|