PeerSet.h

#pragma once
 
#include <xrpld/app/main/Application.h>
#include <xrpld/overlay/Peer.h>
#include <xrpld/overlay/detail/ProtocolMessage.h>
 
#include <google/protobuf/message.h>
 
#include <xrpl.pb.h>
 
#include <cstddef>
#include <functional>
#include <memory>
#include <set>
 
namespace xrpl {

class PeerSet
{
public:
    virtual ~PeerSet() = default;

    virtual void
    addPeers(
        std::size_t limit,
        std::function<bool(std::shared_ptr<Peer> const&)> hasItem,
        std::function<void(std::shared_ptr<Peer> const&)> onPeerAdded) = 0;

    template <typename MessageType>
    void
    sendRequest(MessageType const& message, std::shared_ptr<Peer> const& peer)
    {
        this->sendRequest(message, protocolMessageType(message), peer);
    }
 
    virtual void
    sendRequest(
        ::google::protobuf::Message const& message,
        protocol::MessageType type,
        std::shared_ptr<Peer> const& peer) = 0;

    [[nodiscard]] virtual std::set<Peer::id_t> const&
    getPeerIds() const = 0;
};
 
class PeerSetBuilder
{
public:
    virtual ~PeerSetBuilder() = default;
 
    virtual std::unique_ptr<PeerSet>
    build() = 0;
};
 
std::unique_ptr<PeerSetBuilder>
makePeerSetBuilder(Application& app);

std::unique_ptr<PeerSet>
makeDummyPeerSet(Application& app);
 
}  // namespace xrpl