blob: 4bf1e6cb231127635cc54070844ddfa66c053037 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
#include <Parsers/ASTQueryWithOutput.h>
#include <Common/assert_cast.h>
#include <Parsers/ASTSetQuery.h>
namespace DB
{
void ASTQueryWithOutput::cloneOutputOptions(ASTQueryWithOutput & cloned) const
{
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);
}
if (settings_ast)
{
cloned.settings_ast = settings_ast->clone();
cloned.children.push_back(cloned.settings_ast);
}
}
void ASTQueryWithOutput::formatImpl(const FormatSettings & s, FormatState & state, FormatStateStacked frame) const
{
formatQueryImpl(s, state, frame);
std::string indent_str = s.one_line ? "" : std::string(4u * frame.indent, ' ');
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);
s.ostr << (s.hilite ? hilite_keyword : "");
if (is_outfile_append)
s.ostr << " APPEND";
if (is_outfile_truncate)
s.ostr << " TRUNCATE";
if (is_into_outfile_with_stdout)
s.ostr << " AND STDOUT";
s.ostr << (s.hilite ? hilite_none : "");
}
if (format)
{
s.ostr << (s.hilite ? hilite_keyword : "") << s.nl_or_ws << indent_str << "FORMAT " << (s.hilite ? hilite_none : "");
format->formatImpl(s, state, frame);
}
if (settings_ast && assert_cast<ASTSetQuery *>(settings_ast.get())->print_in_format)
{
s.ostr << (s.hilite ? hilite_keyword : "") << s.nl_or_ws << indent_str << "SETTINGS " << (s.hilite ? hilite_none : "");
settings_ast->formatImpl(s, state, frame);
}
}
bool ASTQueryWithOutput::resetOutputASTIfExist(IAST & ast)
{
/// FIXME: try to prettify this cast using `as<>()`
if (auto * ast_with_output = dynamic_cast<ASTQueryWithOutput *>(&ast))
{
ast_with_output->format.reset();
ast_with_output->out_file.reset();
ast_with_output->settings_ast.reset();
return true;
}
return false;
}
}
|
