blob: 89b75a103c2e57e1b2dbbe618200e1a00b1b9923 (
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
91
92
93
94
95
96
97
98
|
#include <Backups/BackupUtils.h>
#include <Access/Common/AccessRightsElement.h>
#include <Databases/DDLRenamingVisitor.h>
#include <Interpreters/DatabaseCatalog.h>
#include <Common/setThreadName.h>
namespace DB
{
DDLRenamingMap makeRenamingMapFromBackupQuery(const ASTBackupQuery::Elements & elements)
{
DDLRenamingMap map;
for (const auto & element : elements)
{
switch (element.type)
{
case ASTBackupQuery::TABLE:
{
const String & table_name = element.table_name;
const String & database_name = element.database_name;
const String & new_table_name = element.new_table_name;
const String & new_database_name = element.new_database_name;
assert(!table_name.empty());
assert(!new_table_name.empty());
assert(!database_name.empty());
assert(!new_database_name.empty());
map.setNewTableName({database_name, table_name}, {new_database_name, new_table_name});
break;
}
case ASTBackupQuery::TEMPORARY_TABLE:
{
const String & table_name = element.table_name;
const String & new_table_name = element.new_table_name;
assert(!table_name.empty());
assert(!new_table_name.empty());
map.setNewTableName({DatabaseCatalog::TEMPORARY_DATABASE, table_name}, {DatabaseCatalog::TEMPORARY_DATABASE, new_table_name});
break;
}
case ASTBackupQuery::DATABASE:
{
const String & database_name = element.database_name;
const String & new_database_name = element.new_database_name;
assert(!database_name.empty());
assert(!new_database_name.empty());
map.setNewDatabaseName(database_name, new_database_name);
break;
}
case ASTBackupQuery::ALL: break;
}
}
return map;
}
/// Returns access required to execute BACKUP query.
AccessRightsElements getRequiredAccessToBackup(const ASTBackupQuery::Elements & elements)
{
AccessRightsElements required_access;
for (const auto & element : elements)
{
switch (element.type)
{
case ASTBackupQuery::TABLE:
{
required_access.emplace_back(AccessType::BACKUP, element.database_name, element.table_name);
break;
}
case ASTBackupQuery::TEMPORARY_TABLE:
{
/// It's always allowed to backup temporary tables.
break;
}
case ASTBackupQuery::DATABASE:
{
/// TODO: It's better to process `element.except_tables` somehow.
required_access.emplace_back(AccessType::BACKUP, element.database_name);
break;
}
case ASTBackupQuery::ALL:
{
/// TODO: It's better to process `element.except_databases` & `element.except_tables` somehow.
required_access.emplace_back(AccessType::BACKUP);
break;
}
}
}
return required_access;
}
}
|
