contrib/clickhouse/src/Backups/BackupCoordinationReplicatedAccess.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/BackupCoordinationReplicatedAccess.h>


namespace DB
{

BackupCoordinationReplicatedAccess::BackupCoordinationReplicatedAccess() = default;
BackupCoordinationReplicatedAccess::~BackupCoordinationReplicatedAccess() = default;

void BackupCoordinationReplicatedAccess::addFilePath(FilePathForAccessEntitry && file_path_for_access_entity)
{
    const auto & access_zk_path = file_path_for_access_entity.access_zk_path;
    const auto & access_entity_type = file_path_for_access_entity.access_entity_type;
    const auto & host_id = file_path_for_access_entity.host_id;
    const auto & file_path = file_path_for_access_entity.file_path;

    auto & ref = file_paths_by_zk_path[std::make_pair(access_zk_path, access_entity_type)];
    ref.file_paths.emplace(file_path);

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

Strings BackupCoordinationReplicatedAccess::getFilePaths(const String & access_zk_path, AccessEntityType access_entity_type, const String & host_id) const
{
    auto it = file_paths_by_zk_path.find(std::make_pair(access_zk_path, access_entity_type));
    if (it == file_paths_by_zk_path.end())
        return {};

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

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

}