IProtocol.hpp

Go to the documentation of this file.

#ifndef IPROTOCOL_HPP
#define IPROTOCOL_HPP

#include "CGlobalConfiguration.hpp"
#include <memory>
#include <set>

#include <boost/asio.hpp>
#include <boost/enable_shared_from_this.hpp>
#include <boost/noncopyable.hpp>

namespace freedm {
    namespace broker {

class ModuleMessage;
class ProtocolMessage;
class ProtocolMessageWindow;

class IProtocol
    : private boost::noncopyable,
      public boost::enable_shared_from_this<IProtocol>
{
    public:
        virtual ~IProtocol() { };
        virtual void Send(const ModuleMessage& msg) = 0;
        virtual void ReceiveACK(const ProtocolMessage& msg) = 0;
        virtual bool Receive(const ProtocolMessage& msg) = 0;
        virtual void OnReceive() = 0;
        virtual void SendACK(const ProtocolMessage& msg) = 0;
        virtual void Stop() = 0;
        virtual void ChangePhase(bool) { };
        bool GetStopped() { return m_stopped; };
        void SetStopped(bool v) { m_stopped = v; };
        boost::asio::ip::udp::socket& GetSocket();
        void SetReliability(int r);
        int GetReliability() const;
        std::string GetUUID() const;
    protected:
        IProtocol(std::string uuid, boost::asio::ip::udp::endpoint endpoint);
        virtual void WriteCallback(const boost::system::error_code&) { }
        virtual void Write(ProtocolMessageWindow& msg);
    private:
        boost::asio::ip::udp::endpoint m_endpoint;
 
        std::string m_uuid;

        bool m_stopped;

        int m_reliability;
};

    }
}

#endif