Martin Slachta
59 lines
1.6 KiB
C++
59 lines
1.6 KiB
C++
#pragma once
|
|
|
|
#include "ProtobufMessages.hpp"
|
|
#include "message_protocol/MessageConnection.hpp"
|
|
#include "message_protocol/MessageEndpoint.hpp"
|
|
|
|
#include <memory>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include <spdlog/spdlog.h>
|
|
|
|
namespace tw::net {
|
|
|
|
// Listens on a dedicated port for incoming zone-server peer connections and
|
|
// opens outgoing connections to known peers.
|
|
//
|
|
// Incoming and outgoing peers are held together so that broadcast messages
|
|
// (ZoneHello, ZoneBye) reach every peer uniformly.
|
|
class ZoneClusterLink {
|
|
std::unique_ptr<msg::MessageEndpoint> m_endpoint;
|
|
ProtobufMessages m_messages;
|
|
|
|
public:
|
|
explicit ZoneClusterLink(int port);
|
|
|
|
// Poll for messages and accept any pending peer connections.
|
|
void update();
|
|
|
|
// Connect to a remote zone peer and register it in the peer list.
|
|
msg::MessageConnection* connect_to_peer(const std::string& host, int port);
|
|
|
|
// Send a protobuf message to a single peer.
|
|
template<typename T>
|
|
void send_mesg(msg::MessageConnection* peer, const T& msg) {
|
|
auto send_r = m_messages.send(peer, msg, false);
|
|
if(!send_r) {
|
|
spdlog::error("ZoneClusterLink: failed to send to peer {}: {}",
|
|
peer->peer_id(), send_r.error().message());
|
|
}
|
|
}
|
|
|
|
// Send a protobuf message to all connected peers.
|
|
template<typename T>
|
|
void broadcast(const T& msg) {
|
|
m_messages.broadcast(msg);
|
|
}
|
|
|
|
std::vector<msg::MessageConnection*> peers() const {
|
|
return m_endpoint->peers();
|
|
}
|
|
|
|
msg::MessageEndpoint& endpoint() {
|
|
return *m_endpoint;
|
|
}
|
|
};
|
|
|
|
} // namespace tw::net
|