generic-library/cppzmq
1
0
Fork 0
mirror of https://github.com/zeromq/cppzmq.git synced 2024-12-12 10:33:52 +01:00
Code Issues Projects Releases Wiki Activity

Problem: Poller size function missing

Browse Source 
Solution: Add it to poller_t
This commit is contained in:
Gudmundur Adalsteinsson 2020-10-10 21:22:16 +00:00
parent 03243ad64d
commit 2af0c0105a
2 changed files with 23 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
tests/poller.cpp
View File

@ -133,6 +133,20 @@ TEST_CASE("poller wait with no handlers throws", "[poller]")
const zmq::error_t&); const zmq::error_t&);
} }
#if ZMQ_VERSION >= ZMQ_MAKE_VERSION(4, 3, 3)
TEST_CASE("poller add/remove size checks", "[poller]")
{
zmq::context_t context;
zmq::socket_t socket{context, zmq::socket_type::router};
zmq::poller_t<> poller;
CHECK(poller.size() == 0);
poller.add(socket, zmq::event_flags::pollin);
CHECK(poller.size() == 1);
CHECK_NOTHROW(poller.remove(socket));
CHECK(poller.size() == 0);
}
#endif
TEST_CASE("poller remove unregistered throws", "[poller]") TEST_CASE("poller remove unregistered throws", "[poller]")
{ {
zmq::context_t context; zmq::context_t context;

9
zmq.hpp
View File

@ -2653,6 +2653,15 @@ template<typename T = no_user_data> class poller_t
throw error_t(); throw error_t();
} }
#if ZMQ_VERSION >= ZMQ_MAKE_VERSION(4, 3, 3)
size_t size() const noexcept
{
int rc = zmq_poller_size(const_cast<void *>(poller_ptr.get()));
ZMQ_ASSERT(rc >= 0);
return static_cast<size_t>(std::max(rc, 0));
}
#endif
private: private:
struct destroy_poller_t struct destroy_poller_t
{ {