mirror of
https://github.com/scylladb/scylladb.git
synced 2026-04-24 10:30:38 +00:00
f61f6c27e3
before this change, we rely on the default-generated fmt::formatter created from operator<<, but fmt v10 dropped the default-generated formatter. in this change, we define a formatter for gms::endpoint_state, and change update the callers of `operator<<` to use `fmt::print()`. but we cannot drop `operator<<` yet, as we are still using the templated operator<< and templated fmt::formatter to print containers in scylla and in seastar -- they are still using `operator<<` under the hood. Refs #13245 Signed-off-by: Kefu Chai <kefu.chai@scylladb.com> Closes scylladb/scylladb#16705
75 lines
2.4 KiB
C++
75 lines
2.4 KiB
C++
/*
|
|
*
|
|
* Modified by ScyllaDB
|
|
* Copyright (C) 2015-present ScyllaDB
|
|
*/
|
|
|
|
/*
|
|
* SPDX-License-Identifier: (AGPL-3.0-or-later and Apache-2.0)
|
|
*/
|
|
|
|
#include "gms/endpoint_state.hh"
|
|
#include "gms/i_endpoint_state_change_subscriber.hh"
|
|
#include <optional>
|
|
#include <ostream>
|
|
#include <boost/lexical_cast.hpp>
|
|
|
|
namespace gms {
|
|
|
|
static_assert(std::is_default_constructible_v<heart_beat_state>);
|
|
static_assert(std::is_nothrow_copy_constructible_v<heart_beat_state>);
|
|
static_assert(std::is_nothrow_move_constructible_v<heart_beat_state>);
|
|
|
|
static_assert(std::is_nothrow_default_constructible_v<application_state_map>);
|
|
|
|
// Note: although std::map::find is not guaranteed to be noexcept
|
|
// it depends on the comparator used and in this case comparing application_state
|
|
// is noexcept. Therefore, we can safely mark this method noexcept.
|
|
const versioned_value* endpoint_state::get_application_state_ptr(application_state key) const noexcept {
|
|
auto it = _application_state.find(key);
|
|
if (it == _application_state.end()) {
|
|
return nullptr;
|
|
} else {
|
|
return &it->second;
|
|
}
|
|
}
|
|
|
|
std::ostream& operator<<(std::ostream& os, const endpoint_state& x) {
|
|
fmt::print(os, "{}", x);
|
|
return os;
|
|
}
|
|
|
|
bool endpoint_state::is_cql_ready() const noexcept {
|
|
auto* app_state = get_application_state_ptr(application_state::RPC_READY);
|
|
if (!app_state) {
|
|
return false;
|
|
}
|
|
try {
|
|
return boost::lexical_cast<int>(app_state->value());
|
|
} catch (...) {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
future<> i_endpoint_state_change_subscriber::on_application_state_change(inet_address endpoint,
|
|
const gms::application_state_map& states, application_state app_state, permit_id pid,
|
|
std::function<future<>(inet_address, const gms::versioned_value&, permit_id)> func) {
|
|
auto it = states.find(app_state);
|
|
if (it != states.end()) {
|
|
return func(endpoint, it->second, pid);
|
|
}
|
|
return make_ready_future<>();
|
|
}
|
|
|
|
}
|
|
|
|
auto fmt::formatter<gms::endpoint_state>::format(const gms::endpoint_state& x,
|
|
fmt::format_context& ctx) const -> decltype(ctx.out()) {
|
|
auto out = ctx.out();
|
|
out = fmt::format_to(out, "HeartBeatState = {}, AppStateMap =", x._heart_beat_state);
|
|
for (auto& [state, value] : x._application_state) {
|
|
out = fmt::format_to(out, " {{ {} : {} }} ", state, value);
|
|
}
|
|
return out;
|
|
}
|