f88ba0bf5a
When replication strategy class is created caller parr const reference on the config options which is, in turn, a map<string, string>. In the future r.s. classes will need to get "scylla specific" info along with legacy options and this patch prepares for that by passing more generic params argument into constructor. Currently the only inhabitant of the new params is the legacy options. Signed-off-by: Pavel Emelyanov <xemul@scylladb.com>
66 lines
2.0 KiB
C++
66 lines
2.0 KiB
C++
/*
|
|
*
|
|
* Modified by ScyllaDB
|
|
* Copyright (C) 2015-present ScyllaDB
|
|
*/
|
|
|
|
/*
|
|
* SPDX-License-Identifier: (AGPL-3.0-or-later and Apache-2.0)
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include "locator/abstract_replication_strategy.hh"
|
|
#include "locator/tablet_replication_strategy.hh"
|
|
|
|
#include <optional>
|
|
#include <unordered_set>
|
|
|
|
namespace locator {
|
|
|
|
class network_topology_strategy : public abstract_replication_strategy
|
|
, public tablet_aware_replication_strategy {
|
|
public:
|
|
network_topology_strategy(replication_strategy_params params);
|
|
|
|
virtual size_t get_replication_factor(const token_metadata&) const override {
|
|
return _rep_factor;
|
|
}
|
|
|
|
size_t get_replication_factor(const sstring& dc) const {
|
|
auto dc_factor = _dc_rep_factor.find(dc);
|
|
return (dc_factor == _dc_rep_factor.end()) ? 0 : dc_factor->second;
|
|
}
|
|
|
|
const std::vector<sstring>& get_datacenters() const {
|
|
return _datacenteres;
|
|
}
|
|
|
|
virtual bool allow_remove_node_being_replaced_from_natural_endpoints() const override {
|
|
return true;
|
|
}
|
|
|
|
public: // tablet_aware_replication_strategy
|
|
virtual effective_replication_map_ptr make_replication_map(table_id, token_metadata_ptr) const override;
|
|
virtual future<tablet_map> allocate_tablets_for_new_table(schema_ptr, token_metadata_ptr) const override;
|
|
protected:
|
|
/**
|
|
* calculate endpoints in one pass through the tokens by tracking our
|
|
* progress in each DC, rack etc.
|
|
*/
|
|
virtual future<host_id_set> calculate_natural_endpoints(
|
|
const token& search_token, const token_metadata& tm) const override;
|
|
|
|
virtual void validate_options(const gms::feature_service&) const override;
|
|
|
|
virtual std::optional<std::unordered_set<sstring>> recognized_options(const topology&) const override;
|
|
|
|
private:
|
|
// map: data centers -> replication factor
|
|
std::unordered_map<sstring, size_t> _dc_rep_factor;
|
|
|
|
std::vector<sstring> _datacenteres;
|
|
size_t _rep_factor;
|
|
};
|
|
} // namespace locator
|