poller.hpp

Go to the documentation of this file.

/*
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/.
 *
 * This file is part of zmqpp.
 * Copyright (c) 2011-2015 Contributors as noted in the AUTHORS file.
 */

#ifndef ZMQPP_POLLER_HPP_
#define ZMQPP_POLLER_HPP_

#include <unordered_map>
#include <vector>

#include "compatibility.hpp"

namespace zmqpp
{

class socket;
typedef socket socket_t;

class poller
{
public:
    static const long wait_forever; 
    static const short poll_none;   
    static const short poll_in;     
    static const short poll_out;    
    static const short poll_error;  
#if ((ZMQ_VERSION_MAJOR == 4 && ZMQ_VERSION_MINOR >= 2) || ZMQ_VERSION_MAJOR > 4)
        static const short poll_pri;    
#endif

    poller();

    ~poller();

    void add(socket_t& socket, short const event = poll_in);

    void add(raw_socket_t const descriptor, short const event = poll_in | poll_error);

    void add(zmq_pollitem_t const& item);

     bool has(socket_t const& socket);

     bool has(raw_socket_t const descriptor);

    bool has(zmq_pollitem_t const& item);

    void remove(socket_t const& socket);
    
    void remove(raw_socket_t const descriptor);

    void remove(zmq_pollitem_t const& item);

    void check_for(socket_t const& socket, short const event);
    
    void check_for(raw_socket_t const descriptor, short const event);
    
    void check_for(zmq_pollitem_t const& item, short const event);
    
    bool poll(long timeout = wait_forever);
    
    short events(socket_t const& socket) const;
    
    short events(raw_socket_t const descriptor) const;
    
    short events(zmq_pollitem_t const& item) const;

    template<typename Watched>
    bool has_input(Watched const& watchable) const { return (events(watchable) & poll_in) != 0; }

    template<typename Watched>
    bool has_output(Watched const& watchable) const { return (events(watchable) & poll_out) != 0; }

    template<typename Watched>
    bool has_error(Watched const& watchable) const { return (events(watchable) & poll_error) != 0; }

private:
    std::vector<zmq_pollitem_t> _items;
    std::unordered_map<void *, size_t> _index;
    std::unordered_map<raw_socket_t, size_t> _fdindex;

    void reindex(size_t const index);
};

}

#endif /* ZMQPP_POLLER_HPP_ */