contrib/clickhouse/src/Backups/BackupCoordinationReplicatedSQLObjects.cpp


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

#include <Backups/BackupCoordinationReplicatedSQLObjects.h>


namespace DB
{

BackupCoordinationReplicatedSQLObjects::BackupCoordinationReplicatedSQLObjects() = default;
BackupCoordinationReplicatedSQLObjects::~BackupCoordinationReplicatedSQLObjects() = default;

void BackupCoordinationReplicatedSQLObjects::addDirectory(DirectoryPathForSQLObject && directory_path_for_sql_object)
{
    const auto & loader_zk_path = directory_path_for_sql_object.loader_zk_path;
    const auto & object_type = directory_path_for_sql_object.object_type;
    const auto & host_id = directory_path_for_sql_object.host_id;
    const auto & dir_path = directory_path_for_sql_object.dir_path;

    auto & ref = dir_paths_by_zk_path[std::make_pair(loader_zk_path, object_type)];
    ref.dir_paths.emplace(dir_path);

    /// std::max() because the calculation must give the same result being repeated on a different replica.
    ref.host_to_store = std::max(ref.host_to_store, host_id);
}

Strings BackupCoordinationReplicatedSQLObjects::getDirectories(const String & loader_zk_path, UserDefinedSQLObjectType object_type, const String & host_id) const
{
    auto it = dir_paths_by_zk_path.find(std::make_pair(loader_zk_path, object_type));
    if (it == dir_paths_by_zk_path.end())
        return {};

    const auto & dir_paths = it->second;
    if (dir_paths.host_to_store != host_id)
        return {};

    Strings res{dir_paths.dir_paths.begin(), dir_paths.dir_paths.end()};
    return res;
}

}