ClickHouse/dbms/src/Parsers/ParserQueryWithOutput.cpp

#include <DB/Parsers/ParserQueryWithOutput.h>
#include <DB/Parsers/ParserShowTablesQuery.h>
#include <DB/Parsers/ParserSelectQuery.h>
#include <DB/Parsers/ParserTablePropertiesQuery.h>
#include <DB/Parsers/ParserShowProcesslistQuery.h>
#include <DB/Parsers/ParserCheckQuery.h>
#include <DB/Parsers/ParserKillQueryQuery.h>
#include <DB/Parsers/ASTIdentifier.h>
#include <DB/Parsers/ExpressionElementParsers.h>
#include <DB/Common/typeid_cast.h>


namespace DB
{

bool ParserQueryWithOutput::parseImpl(Pos & pos, Pos end, ASTPtr & node, Pos & max_parsed_pos, Expected & expected)
{
	ParserShowTablesQuery show_tables_p;
	ParserSelectQuery select_p;
	ParserTablePropertiesQuery table_p;
	ParserShowProcesslistQuery show_processlist_p;
	ParserCheckQuery check_p;
	ParserKillQueryQuery kill_query_p;

	ASTPtr query;

	bool parsed = select_p.parse(pos, end, query, max_parsed_pos, expected)
		|| show_tables_p.parse(pos, end, query, max_parsed_pos, expected)
		|| table_p.parse(pos, end, query, max_parsed_pos, expected)
		|| show_processlist_p.parse(pos, end, query, max_parsed_pos, expected)
		|| check_p.parse(pos, end, query, max_parsed_pos, expected)
		|| kill_query_p.parse(pos, end, query, max_parsed_pos, expected);

	if (!parsed)
		return false;

	auto & query_with_output = dynamic_cast<ASTQueryWithOutput &>(*query);

	ParserString s_into("INTO", /* word_boundary_ = */ true, /* case_insensitive_ = */ true);
	if (s_into.ignore(pos, end, max_parsed_pos, expected))
	{
		ws.ignore(pos, end);

		ParserString s_outfile("OUTFILE", true, true);
		if (!s_outfile.ignore(pos, end, max_parsed_pos, expected))
		{
			expected = "OUTFILE";
			return false;
		}

		ws.ignore(pos, end);

		ParserStringLiteral out_file_p;
		if (!out_file_p.parse(pos, end, query_with_output.out_file, max_parsed_pos, expected))
			return false;

		query_with_output.children.push_back(query_with_output.out_file);

		ws.ignore(pos, end);
	}

	ParserString s_format("FORMAT", true, true);

	if (s_format.ignore(pos, end, max_parsed_pos, expected))
	{
		ws.ignore(pos, end);

		ParserIdentifier format_p;

		if (!format_p.parse(pos, end, query_with_output.format, max_parsed_pos, expected))
			return false;
		typeid_cast<ASTIdentifier &>(*(query_with_output.format)).kind = ASTIdentifier::Format;

		query_with_output.children.push_back(query_with_output.format);

		ws.ignore(pos, end);
	}

	node = query;
	return true;
}

}