SyncNode.hh

 
#ifndef GRAMODS_NETWORK_SYNCNODE
#define GRAMODS_NETWORK_SYNCNODE
 
#include <gmNetwork/config.hh>
 
// Required before gmCore/OFactory.hh for some compilers
#include <gmCore/io_size.hh>
 
#include <gmNetwork/Protocol.hh>
 
#include <gmCore/Object.hh>
#include <gmCore/OFactory.hh>
#include <gmCore/PreConditionViolation.hh>
 
#include <memory>
#include <set>
 
BEGIN_NAMESPACE_GMNETWORK;
 
class SyncNode
  : public gmCore::Object {
 
public:
 
  SyncNode();
  ~SyncNode();
 
  void addPeer(std::string address);
 
  void setLocalPeerIdx(size_t idx);
 
  size_t getLocalPeerIdx();
 
  void setExitWhenAPeerIsDisconnected(bool on);
 
  void setTimeoutDelay(float t);
 
  float getTimeoutDelay();
 
  void waitForConnection();
 
  bool isConnected();
 
  std::size_t getPeersCount();
 
  std::set<std::size_t> getConnectedPeers();
 
  template<typename TYPE>
  TYPE * getProtocol() {
    static_assert(std::is_base_of<Protocol, TYPE>::value,
                  "Specified protocol is not a Protocol");
 
    std::string name = typeid(TYPE).name();
    Protocol * protocol = getProtocol(name);
 
    if (protocol) {
      TYPE * typed_protocol = dynamic_cast<TYPE*>(protocol);
      if (typed_protocol == nullptr)
        throw gmCore::PreConditionViolation
          ("Requested protocol exists but has wrong type.");
      return typed_protocol;
    }
 
    try {
 
      std::shared_ptr<TYPE> typed_protocol = std::make_shared<TYPE>();
      addProtocol(name, typed_protocol);
 
      return static_cast<TYPE*>(getProtocol(name));
 
    } catch (std::bad_weak_ptr &) {
      throw gmCore::PreConditionViolation
        ("SyncNode instance must be held by a std::shared_ptr");
    }
  }
 
  Protocol * getProtocol(std::string name);
 
  Protocol * getProtocol(char id);
 
  void addProtocol(std::string name, std::shared_ptr<Protocol> prot);
 
  void sendMessage(Protocol::Message mess);
 
  void initialize() override;
 
  virtual std::string getDefaultKey() override { return "syncNode"; }
 
  GM_OFI_DECLARE;
 
private:
 
  struct Impl;
  std::unique_ptr<Impl> _impl;
 
};
 
END_NAMESPACE_GMNETWORK;
 
#endif