ClickHouse/dbms/src/Parsers/ParserQueryWithOutput.cpp

118 lines
3.6 KiB
C++
Raw Normal View History

#include <Parsers/ParserQueryWithOutput.h>
#include <Parsers/ParserShowTablesQuery.h>
2018-02-25 00:50:53 +00:00
#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/ParserSetQuery.h>
2018-09-21 15:20:23 +00:00
#include <Parsers/ASTExplainQuery.h>
2016-11-20 12:43:20 +00:00
2015-10-21 14:57:10 +00:00
namespace DB
{
bool ParserQueryWithOutput::parseImpl(Pos & pos, ASTPtr & node, Expected & expected)
2015-10-21 14:57:10 +00:00
{
ParserShowTablesQuery show_tables_p;
2018-02-25 00:50:53 +00:00
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;
ASTPtr query;
2018-09-21 15:20:23 +00:00
ParserKeyword s_ast("AST");
ParserKeyword s_analyze("ANALYZE");
2018-09-21 15:20:23 +00:00
bool explain_ast = false;
bool analyze_syntax = false;
if (enable_explain && s_ast.ignore(pos, expected))
2018-09-21 15:20:23 +00:00
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);
if (!parsed)
return false;
/// FIXME: try to prettify this cast using `as<>()`
auto * query_with_output = dynamic_cast<ASTQueryWithOutput *>(query.get());
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");
2015-10-21 14:57:10 +00:00
if (s_format.ignore(pos, expected))
{
ParserIdentifier format_p;
2015-10-21 14:57:10 +00:00
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);
}
2018-09-21 15:20:23 +00:00
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);
2018-09-21 15:20:23 +00:00
node->children.push_back(query);
}
else
node = query;
return true;
2015-10-21 14:57:10 +00:00
}
}