blob: 12eda260712f8bcefe70fa4a52dd9e1e9e8280df (
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
78
79
80
81
82
83
84
85
86
87
88
89
90
|
#include <Parsers/Access/ASTShowCreateAccessEntityQuery.h>
#include <Parsers/Access/ASTRowPolicyName.h>
#include <Common/quoteString.h>
#include <IO/Operators.h>
namespace DB
{
namespace
{
void formatNames(const Strings & names, const IAST::FormatSettings & settings)
{
bool need_comma = false;
for (const auto & name : names)
{
if (std::exchange(need_comma, true))
settings.ostr << ',';
settings.ostr << ' ' << backQuoteIfNeed(name);
}
}
}
String ASTShowCreateAccessEntityQuery::getKeyword() const
{
size_t total_count = (names.size()) + (row_policy_names ? row_policy_names->size() : 0) + current_user + current_quota;
bool multiple = (total_count != 1) || all || !short_name.empty() || database_and_table_name;
const auto & type_info = AccessEntityTypeInfo::get(type);
return multiple ? type_info.plural_name : type_info.name;
}
String ASTShowCreateAccessEntityQuery::getID(char) const
{
return String("SHOW CREATE ") + getKeyword() + " query";
}
ASTPtr ASTShowCreateAccessEntityQuery::clone() const
{
auto res = std::make_shared<ASTShowCreateAccessEntityQuery>(*this);
if (row_policy_names)
res->row_policy_names = std::static_pointer_cast<ASTRowPolicyNames>(row_policy_names->clone());
return res;
}
void ASTShowCreateAccessEntityQuery::formatQueryImpl(const FormatSettings & settings, FormatState &, FormatStateStacked) const
{
settings.ostr << (settings.hilite ? hilite_keyword : "") << "SHOW CREATE " << getKeyword() << (settings.hilite ? hilite_none : "");
if (!names.empty())
formatNames(names, settings);
if (row_policy_names)
{
settings.ostr << " ";
row_policy_names->format(settings);
}
if (!short_name.empty())
settings.ostr << " " << backQuoteIfNeed(short_name);
if (database_and_table_name)
{
const String & database = database_and_table_name->first;
const String & table_name = database_and_table_name->second;
settings.ostr << (settings.hilite ? hilite_keyword : "") << " ON " << (settings.hilite ? hilite_none : "");
settings.ostr << (database.empty() ? "" : backQuoteIfNeed(database) + ".");
settings.ostr << (table_name.empty() ? "*" : backQuoteIfNeed(table_name));
}
}
void ASTShowCreateAccessEntityQuery::replaceEmptyDatabase(const String & current_database)
{
if (row_policy_names)
row_policy_names->replaceEmptyDatabase(current_database);
if (database_and_table_name)
{
String & database = database_and_table_name->first;
if (database.empty())
database = current_database;
}
}
}
|
