#include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include "Common/Exception.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; ParserShowAccessQuery show_access_p; ParserShowAccessEntitiesQuery show_access_entities_p; ParserShowCreateAccessEntityQuery show_create_access_entity_p; ParserShowGrantsQuery show_grants_p; ParserShowPrivilegesQuery show_privileges_p; ParserExplainQuery explain_p(end); ASTPtr query; bool parsed = explain_p.parse(pos, query, expected) || select_p.parse(pos, query, expected) || show_create_access_entity_p.parse(pos, query, expected) /// should be before `show_tables_p` || 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) || show_access_p.parse(pos, query, expected) || show_access_entities_p.parse(pos, query, expected) || show_grants_p.parse(pos, query, expected) || show_privileges_p.parse(pos, query, expected); if (!parsed) return false; /// FIXME: try to prettify this cast using `as<>()` auto & query_with_output = dynamic_cast(*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; ParserKeyword s_compression_method("COMPRESSION"); if (s_compression_method.ignore(pos, expected)) { ParserStringLiteral compression; if (!compression.parse(pos, query_with_output.compression, 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); // SETTINGS after FORMAT is not parsed by the SELECT parser (ParserSelectQuery) // Pass them manually, to apply in InterpreterSelectQuery::initSettings() if (query->as()) { QueryWithOutputSettingsPushDownVisitor::Data data{query_with_output.settings_ast}; QueryWithOutputSettingsPushDownVisitor(data).visit(query); } } node = std::move(query); return true; } }