Poller.hpp

Go to the documentation of this file.

/*
 * This file is part of the DUDS project. It is subject to the BSD-style
 * license terms in the LICENSE file found in the top-level directory of this
 * distribution and at https://github.com/jjackowski/duds/blob/master/LICENSE.
 * No part of DUDS, including this file, may be copied, modified, propagated,
 * or distributed except according to the terms contained in the LICENSE file.
 *
 * Copyright (C) 2020  Jeff Jackowski
 */
#ifndef POLLER_HPP
#define POLLER_HPP

#include <sys/epoll.h>
#include <boost/exception/info.hpp>
#include <boost/noncopyable.hpp>
#include <mutex>
#include <vector>
#include <memory>

#ifdef linux
// !@?!#?!#?
#undef linux
#endif

namespace duds {

namespace os {

namespace linux {

struct PollerError : virtual std::exception, virtual boost::exception { };

struct PollerCreateError : PollerError { };

struct PollerLacksFileDescriptor : PollerError { };

struct PollResponderDoesNotExist : PollerError { };

typedef boost::error_info<struct Info_PollerFileDescriptor, int>
    PollerFileDescriptor;

class Poller;

class PollResponder {
public:
    virtual void respond(Poller *poller, int fd) = 0;
};

typedef std::shared_ptr<PollResponder>  PollResponderSptr;

class Poller : boost::noncopyable {
    struct ResponderRecord {
        std::weak_ptr<PollResponder> responder;
        int fd;
        ResponderRecord() = default;
        ResponderRecord(const PollResponderSptr &prs, int f) :
        responder(prs), fd(f) { }
    };
    typedef std::vector<ResponderRecord>  ResponderVec;
    ResponderVec responders;
    std::vector<int> flist;
    std::mutex block;
    int epfd;
public:
    static constexpr int maxEvents = 32;
    Poller(int reserveSize = 0);
    Poller(Poller &&p);
    ~Poller();
    Poller &operator=(Poller &&p) noexcept;
    void add(const PollResponderSptr &prs, int fd, int events = EPOLLIN);
    void remove(int fd);
    int wait(std::chrono::milliseconds timeout);
    int wait() {  // indefinite
        return wait(std::chrono::milliseconds(-1));
    }
    int respond() { // no block
        return wait(std::chrono::milliseconds(0));
    }
};

} } }

#endif        //  #ifndef POLLER_HPP