Protocol.hh

 
#ifndef GRAMODS_NETWORK_PROTOCOL
#define GRAMODS_NETWORK_PROTOCOL
 
#include <gmNetwork/config.hh>
 
#include <memory>
#include <mutex>
#include <vector>
#include <limits>
#include <set>
 
BEGIN_NAMESPACE_GMNETWORK;
 
class SyncNode;
 
class Protocol
  : public std::enable_shared_from_this<Protocol> {
 
public:
 
  static const size_t HEADER_LENGTH;
 
  Protocol() {}
 
  virtual ~Protocol() {}
 
  virtual char getProtocolFlag() { return 0; }
 
  struct Message {
 
    Message() {}
 
    Message(std::vector<char> hdr);
 
    Message(char protocol, std::vector<char> data);
 
    std::vector<char> getHeader();
 
    size_t from_peer_idx = std::numeric_limits<size_t>::max();
 
    size_t to_peer_idx = std::numeric_limits<size_t>::max();
 
    char protocol = 0;
 
    size_t length = 0;
 
    std::vector<char> data;
  };
 
  virtual void processMessage(Message m);
 
  virtual void lostPeer(size_t idx);
 
  void setSyncNode(SyncNode *sync_node);
 
protected:
 
  void sendMessage(std::vector<char> data);
 
  size_t getLocalPeerIdx();
 
  std::set<size_t> getConnectedPeers();
 
  SyncNode *sync_node;
 
  std::mutex sync_node_lock;
};
 
END_NAMESPACE_GMNETWORK;
 
#endif