mirror of
https://github.com/scylladb/scylladb.git
synced 2026-04-26 11:30:36 +00:00
e7ef6149d7
Add a abstract_replication_strategy::get_primary_ranges() method, which is very similar to the existing get_ranges(), except that only the "primary" owner of each range will return it in its list. This is needed for the "primary range" repair option, which asks to repair only the primary range. This option is useful when the user plans to start a repair on *all* nodes, we shouldn't repair the same token range multiple times, so each range should be repaired by only one of the nodes. abstract_replication_strategy::get_primary_ranges() is similar to Origin's StorageService.getPrimaryRangesForEndpoint(). Signed-off-by: Nadav Har'El <nyh@cloudius-systems.com>
95 lines
3.2 KiB
C++
95 lines
3.2 KiB
C++
/*
|
|
* Copyright (C) 2015 Cloudius Systems, Ltd.
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <memory>
|
|
#include <functional>
|
|
#include <unordered_map>
|
|
#include "gms/inet_address.hh"
|
|
#include "dht/i_partitioner.hh"
|
|
#include "token_metadata.hh"
|
|
#include "snitch_base.hh"
|
|
|
|
// forward declaration since database.hh includes this file
|
|
class keyspace;
|
|
|
|
namespace locator {
|
|
|
|
using inet_address = gms::inet_address;
|
|
using token = dht::token;
|
|
|
|
enum class replication_strategy_type {
|
|
simple,
|
|
local,
|
|
network_topology,
|
|
};
|
|
|
|
class abstract_replication_strategy {
|
|
private:
|
|
long _last_invalidated_ring_version = 0;
|
|
std::unordered_map<token, std::vector<inet_address>> _cached_endpoints;
|
|
uint64_t _cache_hits_count = 0;
|
|
|
|
static logging::logger logger;
|
|
|
|
std::unordered_map<token, std::vector<inet_address>>&
|
|
get_cached_endpoints();
|
|
protected:
|
|
sstring _ks_name;
|
|
// TODO: Do we need this member at all?
|
|
//keyspace* _keyspace = nullptr;
|
|
std::map<sstring, sstring> _config_options;
|
|
token_metadata& _token_metadata;
|
|
snitch_ptr& _snitch;
|
|
replication_strategy_type _my_type;
|
|
|
|
template <typename... Args>
|
|
void err(const char* fmt, Args&&... args) const {
|
|
logger.error(fmt, std::forward<Args>(args)...);
|
|
}
|
|
|
|
template <typename... Args>
|
|
void warn(const char* fmt, Args&&... args) const {
|
|
logger.warn(fmt, std::forward<Args>(args)...);
|
|
}
|
|
|
|
template <typename... Args>
|
|
void debug(const char* fmt, Args&&... args) const {
|
|
logger.debug(fmt, std::forward<Args>(args)...);
|
|
}
|
|
|
|
void validate_replication_factor(sstring rf);
|
|
|
|
virtual std::vector<inet_address> calculate_natural_endpoints(const token& search_token) = 0;
|
|
|
|
public:
|
|
abstract_replication_strategy(
|
|
const sstring& keyspace_name,
|
|
token_metadata& token_metadata,
|
|
snitch_ptr& snitch,
|
|
const std::map<sstring, sstring>& config_options,
|
|
replication_strategy_type my_type);
|
|
virtual ~abstract_replication_strategy() {}
|
|
static std::unique_ptr<abstract_replication_strategy> create_replication_strategy(const sstring& ks_name, const sstring& strategy_name, token_metadata& token_metadata, const std::map<sstring, sstring>& config_options);
|
|
virtual std::vector<inet_address> get_natural_endpoints(const token& search_token);
|
|
virtual size_t get_replication_factor() const = 0;
|
|
uint64_t get_cache_hits_count() const { return _cache_hits_count; }
|
|
replication_strategy_type get_type() const { return _my_type; }
|
|
|
|
// get_ranges() returns the list of ranges held by the given endpoint.
|
|
// It the analogue of Origin's getAddressRanges().get(endpoint).
|
|
// This function is not efficient, and not meant for the fast path.
|
|
std::vector<range<token>> get_ranges(inet_address ep);
|
|
// get_primary_ranges() returns the list of "primary ranges" for the given
|
|
// endpoint. "Primary ranges" are the ranges that the node is responsible
|
|
// for storing replica primarily, which means this is the first node
|
|
// returned calculate_natural_endpoints().
|
|
// This function is the analogue of Origin's
|
|
// StorageService.getPrimaryRangesForEndpoint().
|
|
std::vector<range<token>> get_primary_ranges(inet_address ep);
|
|
};
|
|
|
|
}
|