Cluster.cpp

#include <xrpld/overlay/Cluster.h>
 
#include <xrpld/overlay/ClusterNode.h>
 
#include <xrpl/basics/Log.h>
#include <xrpl/basics/StringUtilities.h>
#include <xrpl/basics/chrono.h>
#include <xrpl/beast/utility/Journal.h>
#include <xrpl/config/BasicConfig.h>
#include <xrpl/protocol/PublicKey.h>
#include <xrpl/protocol/tokens.h>
 
#include <boost/regex/v5/regex.hpp>
#include <boost/regex/v5/regex_match.hpp>
 
#include <cstddef>
#include <cstdint>
#include <functional>
#include <mutex>
#include <optional>
#include <string>
#include <utility>
 
namespace xrpl {
 
Cluster::Cluster(beast::Journal j) : j_(j)
{
}
 
std::optional<std::string>
Cluster::member(PublicKey const& identity) const
{
    std::scoped_lock const lock(mutex_);
 
    auto iter = nodes_.find(identity);
    if (iter == nodes_.end())
        return std::nullopt;
    return iter->name();
}
 
bool
Cluster::isMember(PublicKey const& identity) const
{
    std::scoped_lock const lock(mutex_);
 
    return nodes_.contains(identity);
}
 
std::size_t
Cluster::size() const
{
    std::scoped_lock const lock(mutex_);
 
    return nodes_.size();
}
 
bool
Cluster::update(
    PublicKey const& identity,
    std::string name,
    std::uint32_t loadFee,
    NetClock::time_point reportTime)
{
    std::scoped_lock const lock(mutex_);
 
    auto iter = nodes_.find(identity);
 
    if (iter != nodes_.end())
    {
        if (reportTime <= iter->getReportTime())
            return false;
 
        if (name.empty())
            name = iter->name();
 
        iter = nodes_.erase(iter);
    }
 
    nodes_.emplace_hint(iter, identity, std::move(name), loadFee, reportTime);
    return true;
}
 
void
Cluster::forEach(std::function<void(ClusterNode const&)> func) const
{
    std::scoped_lock const lock(mutex_);
    for (auto const& ni : nodes_)
        func(ni);
}
 
bool
Cluster::load(Section const& nodes)
{
    static boost::regex const kRE(
        "[[:space:]]*"       // skip leading whitespace
        "([[:alnum:]]+)"     // node identity
        "(?:"                // begin optional comment block
        "[[:space:]]+"       // (skip all leading whitespace)
        "(?:"                // begin optional comment
        "(.*[^[:space:]]+)"  // the comment
        "[[:space:]]*"       // (skip all trailing whitespace)
        ")?"                 // end optional comment
        ")?"                 // end optional comment block
    );
 
    for (auto const& n : nodes.values())
    {
        boost::smatch match;
 
        if (!boost::regex_match(n, match, kRE))
        {
            JLOG(j_.error()) << "Malformed entry: '" << n << "'";
            return false;
        }
 
        auto const id = parseBase58<PublicKey>(TokenType::NodePublic, match[1].str());
 
        if (!id)
        {
            JLOG(j_.error()) << "Invalid node identity: " << match[1];
            return false;
        }
 
        if (member(*id))
        {
            JLOG(j_.warn()) << "Duplicate node identity: " << match[1];
            continue;
        }
 
        update(*id, trimWhitespace(match[2]));
    }
 
    return true;
}
 
}  // namespace xrpl