poller.h

#ifndef POLLER_H
#define POLLER_H
 
#include <map>
#include <mutex>
#include <functional>
#include "./tcp_listener.h"
 
namespace AsyncBsdSocketLib
{
    class Poller
    {
    private:
        std::mutex mMutex;
        int mFileDescriptor;
        size_t mEventCounter;
        std::map<int, std::function<void()>> mListeners;
        std::map<int, std::function<void()>> mSenders;
        std::map<int, std::function<void()>> mReceivers;
 
        bool tryModifyAsSenderReceiver(int connectionDescriptor);
 
    public:
        Poller();
 
        ~Poller() noexcept;
 
        bool TryAddListener(TcpListener *tcpListener, std::function<void()> callback);
 
        bool TryAddSender(Communicator *communicator, std::function<void()> callback);
 
        bool TryAddReceiver(Communicator *communicator, std::function<void()> callback);
 
        bool TryRemoveListener(TcpListener *tcpListener);
 
        bool TryRemoveSender(Communicator *communicator);
 
        bool TryRemoveReceiver(Communicator *communicator);
 
        bool TryPoll(int timeout = 0) const;
    };
}
 
#endif