633746b87d
Currently snitch drivers register themselves in class-registry with all sorts of construction options possible. All those different constuctors are in fact "config options". When later snitch will declare its dependencies (gossiper and system keyspace), it will require patching all this registrations, which's very inconvenient. This patch introduces the snitch_config struct and replaces all the snitch constructors with the snitch_driver(snitch_config cfg) one. Signed-off-by: Pavel Emelyanov <xemul@scylladb.com>
49 lines
1.2 KiB
C++
49 lines
1.2 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 <seastar/core/sstring.hh>
|
|
#include "gms/inet_address.hh"
|
|
#include "snitch_base.hh"
|
|
#include "utils/fb_utilities.hh"
|
|
|
|
namespace locator {
|
|
|
|
using inet_address = gms::inet_address;
|
|
|
|
/**
|
|
* A simple endpoint snitch implementation that assumes datacenter and rack information is encoded
|
|
* in the 2nd and 3rd octets of the ip address, respectively.
|
|
*/
|
|
struct rack_inferring_snitch : public snitch_base {
|
|
rack_inferring_snitch(const snitch_config& cfg) {
|
|
_my_dc = get_datacenter(utils::fb_utilities::get_broadcast_address());
|
|
_my_rack = get_rack(utils::fb_utilities::get_broadcast_address());
|
|
|
|
// This snitch is ready on creation
|
|
set_snitch_ready();
|
|
}
|
|
|
|
virtual sstring get_rack(inet_address endpoint) override {
|
|
return std::to_string(uint8_t(endpoint.bytes()[2]));
|
|
}
|
|
|
|
virtual sstring get_datacenter(inet_address endpoint) override {
|
|
return std::to_string(uint8_t(endpoint.bytes()[1]));
|
|
}
|
|
|
|
virtual sstring get_name() const override {
|
|
return "org.apache.cassandra.locator.RackInferringSnitch";
|
|
}
|
|
};
|
|
|
|
} // namespace locator
|