GWSResender.h

#pragma once

#include <map>
#include <set>

#include <Poco/Clock.h>
#include <Poco/Event.h>
#include <Poco/Mutex.h>
#include <Poco/Timespan.h>

#include "loop/StoppableRunnable.h"
#include "loop/StopControl.h"
#include "server/GWSConnector.h"
#include "server/GWSListener.h"
#include "util/Loggable.h"

namespace BeeeOn {

class GWSResender :
        public StoppableRunnable,
        public GWSListener,
        Loggable {
public:
        GWSResender();

        void setConnector(GWSConnector::Ptr connector);

        void setResendTimeout(const Poco::Timespan &timeout);

        void run() override;
        void stop() override;

        void onTrySend(const GWMessage::Ptr message) override;

        void onSent(const GWMessage::Ptr message) override;

        void onResponse(const GWResponse::Ptr response) override;
        void onAck(const GWAck::Ptr ack) override;

        void onOther(const GWMessage::Ptr message) override;

protected:
        typedef std::multimap<Poco::Clock, GWMessage::Ptr> WaitingList;

        WaitingList::const_iterator resendOrGet(const Poco::Clock &now);

        WaitingList &waiting();

        bool resendable(const GWMessage::Ptr message) const;

        void findAndDrop(const GWMessage::Ptr message);

private:
        GWSConnector::Ptr m_connector;
        Poco::Timespan m_resendTimeout;
        WaitingList m_waiting;
        std::map<GlobalID, WaitingList::iterator> m_refs;
        std::set<GlobalID> m_pending;
        StopControl m_stopControl;
        Poco::Event m_event;
        Poco::FastMutex m_lock;
};

}