2017-04-01 09:19:00 +00:00
|
|
|
#include <Parsers/ASTQueryWithOutput.h>
|
2023-03-24 02:44:52 +00:00
|
|
|
|
|
|
|
#include <Common/assert_cast.h>
|
2022-07-06 11:39:27 +00:00
|
|
|
#include <Parsers/ASTSetQuery.h>
|
2017-01-11 19:05:46 +00:00
|
|
|
|
|
|
|
namespace DB
|
|
|
|
{
|
|
|
|
|
2017-01-19 12:02:30 +00:00
|
|
|
void ASTQueryWithOutput::cloneOutputOptions(ASTQueryWithOutput & cloned) const
|
2017-01-11 19:05:46 +00:00
|
|
|
{
|
|
|
|
if (out_file)
|
|
|
|
{
|
|
|
|
cloned.out_file = out_file->clone();
|
|
|
|
cloned.children.push_back(cloned.out_file);
|
|
|
|
}
|
|
|
|
if (format)
|
|
|
|
{
|
|
|
|
cloned.format = format->clone();
|
|
|
|
cloned.children.push_back(cloned.format);
|
|
|
|
}
|
2019-01-24 19:22:26 +00:00
|
|
|
if (settings_ast)
|
2019-01-23 19:23:37 +00:00
|
|
|
{
|
2019-01-24 19:22:26 +00:00
|
|
|
cloned.settings_ast = settings_ast->clone();
|
|
|
|
cloned.children.push_back(cloned.settings_ast);
|
2019-01-23 19:23:37 +00:00
|
|
|
}
|
2017-01-11 19:05:46 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
void ASTQueryWithOutput::formatImpl(const FormatSettings & s, FormatState & state, FormatStateStacked frame) const
|
|
|
|
{
|
|
|
|
formatQueryImpl(s, state, frame);
|
|
|
|
|
2017-08-03 17:00:41 +00:00
|
|
|
std::string indent_str = s.one_line ? "" : std::string(4u * frame.indent, ' ');
|
2017-01-11 19:05:46 +00:00
|
|
|
|
|
|
|
if (out_file)
|
|
|
|
{
|
|
|
|
s.ostr << (s.hilite ? hilite_keyword : "") << s.nl_or_ws << indent_str << "INTO OUTFILE " << (s.hilite ? hilite_none : "");
|
|
|
|
out_file->formatImpl(s, state, frame);
|
2023-05-25 08:09:33 +00:00
|
|
|
|
|
|
|
s.ostr << (s.hilite ? hilite_keyword : "");
|
|
|
|
if (is_outfile_append)
|
|
|
|
s.ostr << " APPEND";
|
|
|
|
if (is_into_outfile_with_stdout)
|
|
|
|
s.ostr << " AND STDOUT";
|
|
|
|
s.ostr << (s.hilite ? hilite_none : "");
|
2017-01-11 19:05:46 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
if (format)
|
|
|
|
{
|
|
|
|
s.ostr << (s.hilite ? hilite_keyword : "") << s.nl_or_ws << indent_str << "FORMAT " << (s.hilite ? hilite_none : "");
|
|
|
|
format->formatImpl(s, state, frame);
|
|
|
|
}
|
2019-01-23 19:23:37 +00:00
|
|
|
|
2022-08-22 15:39:33 +00:00
|
|
|
if (settings_ast && assert_cast<ASTSetQuery *>(settings_ast.get())->print_in_format)
|
2019-01-23 19:23:37 +00:00
|
|
|
{
|
|
|
|
s.ostr << (s.hilite ? hilite_keyword : "") << s.nl_or_ws << indent_str << "SETTINGS " << (s.hilite ? hilite_none : "");
|
2019-01-24 19:22:26 +00:00
|
|
|
settings_ast->formatImpl(s, state, frame);
|
2019-01-23 19:23:37 +00:00
|
|
|
}
|
2017-01-11 19:05:46 +00:00
|
|
|
}
|
|
|
|
|
2018-02-12 18:41:53 +00:00
|
|
|
bool ASTQueryWithOutput::resetOutputASTIfExist(IAST & ast)
|
|
|
|
{
|
2019-03-12 12:41:57 +00:00
|
|
|
/// FIXME: try to prettify this cast using `as<>()`
|
|
|
|
if (auto * ast_with_output = dynamic_cast<ASTQueryWithOutput *>(&ast))
|
2018-02-12 18:41:53 +00:00
|
|
|
{
|
|
|
|
ast_with_output->format.reset();
|
|
|
|
ast_with_output->out_file.reset();
|
2019-01-24 19:22:26 +00:00
|
|
|
ast_with_output->settings_ast.reset();
|
2018-02-12 18:41:53 +00:00
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2017-01-11 19:05:46 +00:00
|
|
|
}
|