generic-library/cppzmq
1
0
Fork 0
mirror of https://github.com/zeromq/cppzmq.git synced 2025-05-20 13:05:34 +02:00
Code Issues Projects Releases Wiki Activity

poller_t class based on new libzmq poller API

Browse Source
This commit is contained in:
Pawel Kurdybacha 2017-01-31 21:57:51 +00:00
parent 178a910ae1
commit bafbf2889d
1 changed files with 58 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

58
zmq.hpp
View File

@ -52,6 +52,7 @@
#ifdef ZMQ_CPP11
#include <chrono>
#include <tuple>
#include <functional>
#endif
// Detect whether the compiler supports C++11 rvalue references.
@ -490,6 +491,7 @@ namespace zmq
class socket_t
{
friend class monitor_t;
friend class poller_t;
public:
inline socket_t(context_t& context_, int type_)
{
@ -827,6 +829,62 @@ namespace zmq
private:
void* socketPtr;
};
#if defined(ZMQ_BUILD_DRAFT_API) && defined(ZMQ_CPP11) && defined(ZMQ_HAVE_POLLER)
class poller_t
{
public:
poller_t () : poller_ptr (zmq_poller_new ())
{
if (!poller_ptr)
throw error_t ();
}
~poller_t ()
{
zmq_poller_destroy (&poller_ptr);
}
bool add (zmq::socket_t &socket, short events, std::function<void(void)> &handler)
{
if (0 == zmq_poller_add (poller_ptr, socket.ptr, &handler, events)) {
poller_events.emplace_back (zmq_poller_event_t ());
return true;
}
return false;
}
bool remove (zmq::socket_t &socket)
{
if (0 == zmq_poller_remove (poller_ptr, socket.ptr)) {
poller_events.pop_back ();
return true;
}
return false;
}
bool wait (std::chrono::milliseconds timeout)
{
int rc = zmq_poller_wait_all (poller_ptr, poller_events.data (), poller_events.size (), static_cast<long>(timeout.count ()));
if (rc >= 0) {
std::for_each (poller_events.begin (), poller_events.begin () + rc, [](zmq_poller_event_t& event) {
(*reinterpret_cast<std::function<void(void)>*> (event.user_data)) ();
});
return true;
}
if (zmq_errno () == ETIMEDOUT)
return false;
throw error_t ();
}
private:
void *poller_ptr;
std::vector<zmq_poller_event_t> poller_events;
};
#endif // defined(ZMQ_BUILD_DRAFT_API) && defined(ZMQ_CPP11) && defined(ZMQ_HAVE_POLLER)
}
#endif