Skip to content

Commit

Permalink
Problem: Missing swap functions
Browse files Browse the repository at this point in the history
Solution: Implement for socket_t, context_t, message_t and poller_t
Additionally remove dependency on <functional> by refactoring poller_t
and remove unused <unordered_map> include.
  • Loading branch information
gummif committed Apr 3, 2019
1 parent e1fe5e5 commit 83b91c8
Show file tree
Hide file tree
Showing 7 changed files with 108 additions and 9 deletions.
15 changes: 15 additions & 0 deletions tests/context.cpp
Original file line number Diff line number Diff line change
@@ -1,6 +1,11 @@
#include <catch.hpp>
#include <zmq.hpp>

#if (__cplusplus >= 201703L)
static_assert(std::is_nothrow_swappable<zmq::context_t>::value,
"context_t should be nothrow swappable");
#endif

TEST_CASE("context construct default and destroy", "[context]")
{
zmq::context_t context;
Expand All @@ -12,3 +17,13 @@ TEST_CASE("context create, close and destroy", "[context]")
context.close();
CHECK(NULL == (void *) context);
}

#ifdef ZMQ_CPP11
TEST_CASE("context swap", "[context]")
{
zmq::context_t context1;
zmq::context_t context2;
using std::swap;
swap(context1, context2);
}
#endif
20 changes: 20 additions & 0 deletions tests/message.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -8,13 +8,33 @@ static_assert(!std::is_copy_constructible<zmq::message_t>::value,
static_assert(!std::is_copy_assignable<zmq::message_t>::value,
"message_t should not be copy-assignable");
#endif
#if (__cplusplus >= 201703L)
static_assert(std::is_nothrow_swappable<zmq::message_t>::value,
"message_t should be nothrow swappable");
#endif

TEST_CASE("message default constructed", "[message]")
{
const zmq::message_t message;
CHECK(0u == message.size());
}

#ifdef ZMQ_CPP11
TEST_CASE("message swap", "[message]")
{
const std::string data = "foo";
zmq::message_t message1;
zmq::message_t message2(data.data(), data.size());
using std::swap;
swap(message1, message2);
CHECK(message1.size() == data.size());
CHECK(message2.size() == 0);
swap(message1, message2);
CHECK(message1.size() == 0);
CHECK(message2.size() == data.size());
}
#endif

namespace {
const char *const data = "Hi";
}
Expand Down
1 change: 1 addition & 0 deletions tests/monitor.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@
#include <thread>
#include <mutex>
#include <condition_variable>
#include <functional>

class mock_monitor_t : public zmq::monitor_t
{
Expand Down
13 changes: 13 additions & 0 deletions tests/poller.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,11 @@
#include <array>
#include <memory>

#if (__cplusplus >= 201703L)
static_assert(std::is_nothrow_swappable<zmq::poller_t<>>::value,
"poller_t should be nothrow swappable");
#endif

TEST_CASE("poller create destroy", "[poller]")
{
zmq::poller_t<> poller;
Expand All @@ -28,6 +33,14 @@ TEST_CASE("poller move assign empty", "[poller]")
b = std::move(a);
}

TEST_CASE("poller swap", "[poller]")
{
zmq::poller_t<> a;
zmq::poller_t<> b;
using std::swap;
swap(a, b);
}

TEST_CASE("poller move construct non empty", "[poller]")
{
zmq::context_t context;
Expand Down
14 changes: 14 additions & 0 deletions tests/socket.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,11 @@
#include <future>
#endif

#if (__cplusplus >= 201703L)
static_assert(std::is_nothrow_swappable<zmq::socket_t>::value,
"socket_t should be nothrow swappable");
#endif

TEST_CASE("socket create destroy", "[socket]")
{
zmq::context_t context;
Expand All @@ -16,6 +21,15 @@ TEST_CASE("socket create by enum and destroy", "[socket]")
zmq::context_t context;
zmq::socket_t socket(context, zmq::socket_type::router);
}

TEST_CASE("socket swap", "[socket]")
{
zmq::context_t context;
zmq::socket_t socket1(context, zmq::socket_type::router);
zmq::socket_t socket2(context, zmq::socket_type::dealer);
using std::swap;
swap(socket1, socket2);
}
#endif

TEST_CASE("socket sends and receives const buffer", "[socket]")
Expand Down
48 changes: 40 additions & 8 deletions zmq.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -73,8 +73,6 @@
#ifdef ZMQ_CPP11
#include <chrono>
#include <tuple>
#include <functional>
#include <unordered_map>
#include <memory>
#endif

Expand Down Expand Up @@ -476,6 +474,12 @@ class message_t
return os.str();
}

void swap(message_t &other) ZMQ_NOTHROW
{
// this assumes zmq::msg_t from libzmq is trivially relocatable
std::swap(msg, other.msg);
}

private:
// The underlying message
zmq_msg_t msg;
Expand All @@ -486,6 +490,11 @@ class message_t
void operator=(const message_t &) ZMQ_DELETED_FUNCTION;
};

inline void swap(message_t &a, message_t &b) ZMQ_NOTHROW
{
a.swap(b);
}

class context_t
{
public:
Expand Down Expand Up @@ -554,13 +563,22 @@ class context_t

operator bool() const ZMQ_NOTHROW { return ptr != ZMQ_NULLPTR; }

void swap(context_t &other) ZMQ_NOTHROW
{
std::swap(ptr, other.ptr);
}

private:
void *ptr;

context_t(const context_t &) ZMQ_DELETED_FUNCTION;
void operator=(const context_t &) ZMQ_DELETED_FUNCTION;
};

inline void swap(context_t &a, context_t &b) ZMQ_NOTHROW {
a.swap(b);
}

#ifdef ZMQ_CPP11
enum class socket_type : int
{
Expand Down Expand Up @@ -767,6 +785,12 @@ class socket_t
}
#endif

void swap(socket_t &other) ZMQ_NOTHROW
{
std::swap(ptr, other.ptr);
std::swap(ctxptr, other.ctxptr);
}

private:
void *ptr;
void *ctxptr;
Expand All @@ -775,6 +799,10 @@ class socket_t
void operator=(const socket_t &) ZMQ_DELETED_FUNCTION;
};

inline void swap(socket_t &a, socket_t &b) ZMQ_NOTHROW {
a.swap(b);
}

ZMQ_DEPRECATED("from 4.3.1, use proxy taking socket_t objects")
inline void proxy(void *frontend, void *backend, void *capture)
{
Expand Down Expand Up @@ -1119,6 +1147,8 @@ class monitor_t
template<typename T = void> class poller_t
{
public:
poller_t() = default;

void add(zmq::socket_t &socket, short events, T *user_data)
{
if (0
Expand Down Expand Up @@ -1164,17 +1194,19 @@ template<typename T = void> class poller_t
}

private:
std::unique_ptr<void, std::function<void(void *)>> poller_ptr{
std::unique_ptr<void, void(*)(void *)> poller_ptr{
[]() {
auto poller_new = zmq_poller_new();
if (poller_new)
return poller_new;
throw error_t();
}(),
[](void *ptr) {
int rc = zmq_poller_destroy(&ptr);
ZMQ_ASSERT(rc == 0);
}};
}(), &destroy_poller};

static void destroy_poller(void *ptr)
{
int rc = zmq_poller_destroy(&ptr);
ZMQ_ASSERT(rc == 0);
}
};
#endif // defined(ZMQ_BUILD_DRAFT_API) && defined(ZMQ_CPP11) && defined(ZMQ_HAVE_POLLER)

Expand Down
6 changes: 5 additions & 1 deletion zmq_addon.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -30,6 +30,10 @@
#include <iomanip>
#include <sstream>
#include <stdexcept>
#ifdef ZMQ_CPP11
#include <functional>
#include <unordered_map>
#endif

namespace zmq
{
Expand Down Expand Up @@ -245,7 +249,7 @@ class multipart_t
m_parts.pop_back();
return message;
}

// get message part from front
const message_t &front()
{
Expand Down

0 comments on commit 83b91c8

Please sign in to comment.