reactor.hpp

Go to the documentation of this file.

#ifndef CORAL_NET_REACTOR_HPP
#define CORAL_NET_REACTOR_HPP

#include <chrono>
#include <functional>
#include <memory>
#include <vector>
#include <utility>

#include "zmq.hpp"
#include "coral/config.h"


namespace coral
{
namespace net
{


class Reactor
{
public:
#ifdef _WIN32
    typedef SOCKET NativeSocket;
#else
    typedef int NativeSocket;
#endif

    typedef std::chrono::system_clock::time_point TimePoint;
    typedef std::function<void(Reactor&, zmq::socket_t&)> SocketHandler;
    typedef std::function<void(Reactor&, NativeSocket)> NativeSocketHandler;
    typedef std::function<void(Reactor&, int)> TimerHandler;

    Reactor();

    void AddSocket(zmq::socket_t& socket, SocketHandler handler);

    void RemoveSocket(zmq::socket_t& socket) CORAL_NOEXCEPT;

    void AddNativeSocket(NativeSocket socket, NativeSocketHandler handler);

    void RemoveNativeSocket(NativeSocket socket) CORAL_NOEXCEPT;

    static const int invalidTimerID;

    int AddTimer(
        std::chrono::milliseconds interval,
        int count,
        TimerHandler handler);

    void RemoveTimer(int id);

    void RestartTimerInterval(int id);

    void Run();

    void Stop();

private:
    struct Timer
    {
        Timer(
            int id,
            TimePoint nextEventTime,
            std::chrono::milliseconds interval,
            int remaining,
            std::unique_ptr<TimerHandler> handler);

        CORAL_DEFINE_DEFAULT_MOVE(Timer, id, nextEventTime, interval, remaining, handler)

        int id;
        TimePoint nextEventTime;
        std::chrono::milliseconds interval;
        int remaining;
        std::unique_ptr<TimerHandler> handler;
    };

    void RestartTimerIntervals(
        std::vector<Timer>::iterator begin,
        std::vector<Timer>::iterator end);
    std::chrono::milliseconds TimeToNextEvent() const;
    void PerformNextEvent();

    // Rebuilds the list of poll items.
    void Rebuild();

    typedef std::pair<zmq::socket_t*, std::unique_ptr<SocketHandler>> SocketHandlerPair;
    typedef std::pair<NativeSocket, std::unique_ptr<NativeSocketHandler>> NativeSocketHandlerPair;
    std::vector<SocketHandlerPair> m_sockets;
    std::vector<NativeSocketHandlerPair> m_nativeSockets;
    std::vector<zmq::pollitem_t> m_pollItems;

    int m_nextTimerID;
    std::vector<Timer> m_timers;

    bool m_needsRebuild;
    bool m_continuePolling;
};


}}      // namespace
#endif  // header guard