mirror of
https://github.com/zeromq/cppzmq.git
synced 2025-03-01 02:56:56 +01:00
Merge pull request #226 from kurdybacha/clang-format
Problem: inconsistent code style
This commit is contained in:
commit
bc82352c95
53
.clang-format
Normal file
53
.clang-format
Normal file
@ -0,0 +1,53 @@
|
||||
BasedOnStyle: LLVM
|
||||
IndentWidth: 4
|
||||
UseTab: Never
|
||||
BreakBeforeBraces: Custom
|
||||
BraceWrapping:
|
||||
AfterClass: true
|
||||
AfterControlStatement: false
|
||||
AfterEnum: true
|
||||
AfterFunction: true
|
||||
AfterNamespace: true
|
||||
AfterObjCDeclaration: true
|
||||
AfterStruct: true
|
||||
AfterUnion: true
|
||||
BeforeCatch: true
|
||||
BeforeElse: false
|
||||
IndentBraces: false
|
||||
|
||||
AlignConsecutiveAssignments: false
|
||||
AlignConsecutiveDeclarations: false
|
||||
AllowShortIfStatementsOnASingleLine: false
|
||||
IndentCaseLabels: true
|
||||
BinPackArguments: true
|
||||
BinPackParameters: false
|
||||
AlignTrailingComments: true
|
||||
AllowShortBlocksOnASingleLine: false
|
||||
AllowAllParametersOfDeclarationOnNextLine: true
|
||||
AllowShortFunctionsOnASingleLine: InlineOnly
|
||||
AlwaysBreakTemplateDeclarations: false
|
||||
ColumnLimit: 80
|
||||
MaxEmptyLinesToKeep: 2
|
||||
KeepEmptyLinesAtTheStartOfBlocks: false
|
||||
ContinuationIndentWidth: 2
|
||||
PointerAlignment: Right
|
||||
ReflowComments: false
|
||||
SpaceBeforeAssignmentOperators: true
|
||||
SpaceBeforeParens: Always
|
||||
SpaceInEmptyParentheses: false
|
||||
SpacesInAngles: false
|
||||
SpacesInParentheses: false
|
||||
SpacesInSquareBrackets: false
|
||||
Standard: Cpp11
|
||||
|
||||
SortIncludes: false
|
||||
|
||||
FixNamespaceComments: false
|
||||
BreakBeforeBinaryOperators: NonAssignment
|
||||
SpaceAfterTemplateKeyword: true
|
||||
AlignAfterOpenBracket: Align
|
||||
AlignOperands: true
|
||||
BreakConstructorInitializers: AfterColon
|
||||
ConstructorInitializerAllOnOneLineOrOnePerLine: true
|
||||
SpaceAfterCStyleCast: true
|
||||
BreakBeforeTernaryOperators: true
|
@ -1,8 +1,7 @@
|
||||
#include <zmq.hpp>
|
||||
|
||||
int main(int argc, char** argv)
|
||||
int main (int argc, char **argv)
|
||||
{
|
||||
zmq::context_t context;
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
@ -7,83 +7,85 @@
|
||||
#include <array>
|
||||
#include <memory>
|
||||
|
||||
TEST(active_poller, create_destroy)
|
||||
TEST (active_poller, create_destroy)
|
||||
{
|
||||
zmq::active_poller_t active_poller;
|
||||
ASSERT_TRUE(active_poller.empty ());
|
||||
ASSERT_TRUE (active_poller.empty ());
|
||||
}
|
||||
|
||||
static_assert(!std::is_copy_constructible<zmq::active_poller_t>::value, "active_active_poller_t should not be copy-constructible");
|
||||
static_assert(!std::is_copy_assignable<zmq::active_poller_t>::value, "active_active_poller_t should not be copy-assignable");
|
||||
static_assert (!std::is_copy_constructible<zmq::active_poller_t>::value,
|
||||
"active_active_poller_t should not be copy-constructible");
|
||||
static_assert (!std::is_copy_assignable<zmq::active_poller_t>::value,
|
||||
"active_active_poller_t should not be copy-assignable");
|
||||
|
||||
TEST(active_poller, move_construct_empty)
|
||||
TEST (active_poller, move_construct_empty)
|
||||
{
|
||||
zmq::active_poller_t a;
|
||||
ASSERT_TRUE(a.empty ());
|
||||
ASSERT_TRUE (a.empty ());
|
||||
zmq::active_poller_t b = std::move (a);
|
||||
ASSERT_TRUE(b.empty ());
|
||||
ASSERT_EQ(0u, a.size ());
|
||||
ASSERT_EQ(0u, b.size ());
|
||||
ASSERT_TRUE (b.empty ());
|
||||
ASSERT_EQ (0u, a.size ());
|
||||
ASSERT_EQ (0u, b.size ());
|
||||
}
|
||||
|
||||
TEST(active_poller, move_assign_empty)
|
||||
TEST (active_poller, move_assign_empty)
|
||||
{
|
||||
zmq::active_poller_t a;
|
||||
ASSERT_TRUE(a.empty());
|
||||
ASSERT_TRUE (a.empty ());
|
||||
zmq::active_poller_t b;
|
||||
ASSERT_TRUE(b.empty());
|
||||
b = std::move(a);
|
||||
ASSERT_EQ(0u, a.size ());
|
||||
ASSERT_EQ(0u, b.size ());
|
||||
ASSERT_TRUE(a.empty());
|
||||
ASSERT_TRUE(b.empty());
|
||||
ASSERT_TRUE (b.empty ());
|
||||
b = std::move (a);
|
||||
ASSERT_EQ (0u, a.size ());
|
||||
ASSERT_EQ (0u, b.size ());
|
||||
ASSERT_TRUE (a.empty ());
|
||||
ASSERT_TRUE (b.empty ());
|
||||
}
|
||||
|
||||
TEST(active_poller, move_construct_non_empty)
|
||||
TEST (active_poller, move_construct_non_empty)
|
||||
{
|
||||
zmq::context_t context;
|
||||
zmq::socket_t socket{context, zmq::socket_type::router};
|
||||
|
||||
zmq::active_poller_t a;
|
||||
a.add(socket, ZMQ_POLLIN, [](short) {});
|
||||
ASSERT_FALSE(a.empty ());
|
||||
ASSERT_EQ(1u, a.size ());
|
||||
a.add (socket, ZMQ_POLLIN, [](short) {});
|
||||
ASSERT_FALSE (a.empty ());
|
||||
ASSERT_EQ (1u, a.size ());
|
||||
zmq::active_poller_t b = std::move (a);
|
||||
ASSERT_TRUE(a.empty ());
|
||||
ASSERT_EQ(0u, a.size ());
|
||||
ASSERT_FALSE(b.empty ());
|
||||
ASSERT_EQ(1u, b.size ());
|
||||
ASSERT_TRUE (a.empty ());
|
||||
ASSERT_EQ (0u, a.size ());
|
||||
ASSERT_FALSE (b.empty ());
|
||||
ASSERT_EQ (1u, b.size ());
|
||||
}
|
||||
|
||||
TEST(active_poller, move_assign_non_empty)
|
||||
TEST (active_poller, move_assign_non_empty)
|
||||
{
|
||||
zmq::context_t context;
|
||||
zmq::socket_t socket{context, zmq::socket_type::router};
|
||||
|
||||
zmq::active_poller_t a;
|
||||
a.add(socket, ZMQ_POLLIN, [](short) {});
|
||||
ASSERT_FALSE(a.empty());
|
||||
ASSERT_EQ(1u, a.size ());
|
||||
a.add (socket, ZMQ_POLLIN, [](short) {});
|
||||
ASSERT_FALSE (a.empty ());
|
||||
ASSERT_EQ (1u, a.size ());
|
||||
zmq::active_poller_t b;
|
||||
b = std::move(a);
|
||||
ASSERT_TRUE(a.empty ());
|
||||
ASSERT_EQ(0u, a.size ());
|
||||
ASSERT_FALSE(b.empty ());
|
||||
ASSERT_EQ(1u, b.size ());
|
||||
b = std::move (a);
|
||||
ASSERT_TRUE (a.empty ());
|
||||
ASSERT_EQ (0u, a.size ());
|
||||
ASSERT_FALSE (b.empty ());
|
||||
ASSERT_EQ (1u, b.size ());
|
||||
}
|
||||
|
||||
TEST(active_poller, add_handler)
|
||||
TEST (active_poller, add_handler)
|
||||
{
|
||||
zmq::context_t context;
|
||||
zmq::socket_t socket{context, zmq::socket_type::router};
|
||||
zmq::active_poller_t active_poller;
|
||||
zmq::active_poller_t::handler_t handler;
|
||||
ASSERT_NO_THROW(active_poller.add(socket, ZMQ_POLLIN, handler));
|
||||
ASSERT_NO_THROW (active_poller.add (socket, ZMQ_POLLIN, handler));
|
||||
}
|
||||
|
||||
TEST(active_poller, add_handler_invalid_events_type)
|
||||
TEST (active_poller, add_handler_invalid_events_type)
|
||||
{
|
||||
/// \todo is it good that this is accepted? should probably already be
|
||||
/// \todo is it good that this is accepted? should probably already be
|
||||
/// checked by zmq_poller_add/modify in libzmq:
|
||||
/// https://github.com/zeromq/libzmq/issues/3088
|
||||
zmq::context_t context;
|
||||
@ -91,86 +93,88 @@ TEST(active_poller, add_handler_invalid_events_type)
|
||||
zmq::active_poller_t active_poller;
|
||||
zmq::active_poller_t::handler_t handler;
|
||||
short invalid_events_type = 2 << 10;
|
||||
ASSERT_NO_THROW(active_poller.add(socket, invalid_events_type, handler));
|
||||
ASSERT_FALSE(active_poller.empty ());
|
||||
ASSERT_EQ(1u, active_poller.size ());
|
||||
ASSERT_NO_THROW (active_poller.add (socket, invalid_events_type, handler));
|
||||
ASSERT_FALSE (active_poller.empty ());
|
||||
ASSERT_EQ (1u, active_poller.size ());
|
||||
}
|
||||
|
||||
TEST(active_poller, add_handler_twice_throws)
|
||||
TEST (active_poller, add_handler_twice_throws)
|
||||
{
|
||||
zmq::context_t context;
|
||||
zmq::socket_t socket{context, zmq::socket_type::router};
|
||||
zmq::active_poller_t active_poller;
|
||||
zmq::active_poller_t::handler_t handler;
|
||||
active_poller.add(socket, ZMQ_POLLIN, handler);
|
||||
active_poller.add (socket, ZMQ_POLLIN, handler);
|
||||
/// \todo the actual error code should be checked
|
||||
ASSERT_THROW(active_poller.add(socket, ZMQ_POLLIN, handler), zmq::error_t);
|
||||
ASSERT_THROW (active_poller.add (socket, ZMQ_POLLIN, handler),
|
||||
zmq::error_t);
|
||||
}
|
||||
|
||||
TEST(active_poller, wait_with_no_handlers_throws)
|
||||
TEST (active_poller, wait_with_no_handlers_throws)
|
||||
{
|
||||
zmq::active_poller_t active_poller;
|
||||
/// \todo the actual error code should be checked
|
||||
ASSERT_THROW(active_poller.wait(std::chrono::milliseconds{10}), zmq::error_t);
|
||||
ASSERT_THROW (active_poller.wait (std::chrono::milliseconds{10}),
|
||||
zmq::error_t);
|
||||
}
|
||||
|
||||
TEST(active_poller, remove_unregistered_throws)
|
||||
TEST (active_poller, remove_unregistered_throws)
|
||||
{
|
||||
zmq::context_t context;
|
||||
zmq::socket_t socket{context, zmq::socket_type::router};
|
||||
zmq::active_poller_t active_poller;
|
||||
/// \todo the actual error code should be checked
|
||||
ASSERT_THROW(active_poller.remove(socket), zmq::error_t);
|
||||
ASSERT_THROW (active_poller.remove (socket), zmq::error_t);
|
||||
}
|
||||
|
||||
TEST(active_poller, remove_registered_empty)
|
||||
TEST (active_poller, remove_registered_empty)
|
||||
{
|
||||
zmq::context_t context;
|
||||
zmq::socket_t socket{context, zmq::socket_type::router};
|
||||
zmq::active_poller_t active_poller;
|
||||
active_poller.add(socket, ZMQ_POLLIN, zmq::active_poller_t::handler_t{});
|
||||
ASSERT_NO_THROW(active_poller.remove(socket));
|
||||
active_poller.add (socket, ZMQ_POLLIN, zmq::active_poller_t::handler_t{});
|
||||
ASSERT_NO_THROW (active_poller.remove (socket));
|
||||
}
|
||||
|
||||
TEST(active_poller, remove_registered_non_empty)
|
||||
TEST (active_poller, remove_registered_non_empty)
|
||||
{
|
||||
zmq::context_t context;
|
||||
zmq::socket_t socket{context, zmq::socket_type::router};
|
||||
zmq::active_poller_t active_poller;
|
||||
active_poller.add(socket, ZMQ_POLLIN, [](short) {});
|
||||
ASSERT_NO_THROW(active_poller.remove(socket));
|
||||
active_poller.add (socket, ZMQ_POLLIN, [](short) {});
|
||||
ASSERT_NO_THROW (active_poller.remove (socket));
|
||||
}
|
||||
|
||||
namespace {
|
||||
namespace
|
||||
{
|
||||
struct server_client_setup : common_server_client_setup
|
||||
{
|
||||
zmq::active_poller_t::handler_t handler = [&](short e) {
|
||||
events = e;
|
||||
};
|
||||
zmq::active_poller_t::handler_t handler = [&](short e) { events = e; };
|
||||
|
||||
short events = 0;
|
||||
};
|
||||
}
|
||||
|
||||
TEST(active_poller, poll_basic)
|
||||
TEST (active_poller, poll_basic)
|
||||
{
|
||||
server_client_setup s;
|
||||
|
||||
ASSERT_NO_THROW(s.client.send("Hi"));
|
||||
ASSERT_NO_THROW (s.client.send ("Hi"));
|
||||
|
||||
zmq::active_poller_t active_poller;
|
||||
bool message_received = false;
|
||||
zmq::active_poller_t::handler_t handler = [&message_received](short events) {
|
||||
ASSERT_TRUE(0 != (events & ZMQ_POLLIN));
|
||||
message_received = true;
|
||||
};
|
||||
ASSERT_NO_THROW(active_poller.add(s.server, ZMQ_POLLIN, handler));
|
||||
ASSERT_EQ(1, active_poller.wait(std::chrono::milliseconds{-1}));
|
||||
ASSERT_TRUE(message_received);
|
||||
zmq::active_poller_t::handler_t handler =
|
||||
[&message_received](short events) {
|
||||
ASSERT_TRUE (0 != (events & ZMQ_POLLIN));
|
||||
message_received = true;
|
||||
};
|
||||
ASSERT_NO_THROW (active_poller.add (s.server, ZMQ_POLLIN, handler));
|
||||
ASSERT_EQ (1, active_poller.wait (std::chrono::milliseconds{-1}));
|
||||
ASSERT_TRUE (message_received);
|
||||
}
|
||||
|
||||
/// \todo this contains multiple test cases that should be split up
|
||||
TEST(active_poller, client_server)
|
||||
TEST (active_poller, client_server)
|
||||
{
|
||||
const std::string send_msg = "Hi";
|
||||
|
||||
@ -183,47 +187,49 @@ TEST(active_poller, client_server)
|
||||
zmq::active_poller_t::handler_t handler = [&](short e) {
|
||||
if (0 != (e & ZMQ_POLLIN)) {
|
||||
zmq::message_t zmq_msg;
|
||||
ASSERT_NO_THROW(s.server.recv(&zmq_msg)); // get message
|
||||
std::string recv_msg(zmq_msg.data<char>(),
|
||||
zmq_msg.size());
|
||||
ASSERT_EQ(send_msg, recv_msg);
|
||||
ASSERT_NO_THROW (s.server.recv (&zmq_msg)); // get message
|
||||
std::string recv_msg (zmq_msg.data<char> (), zmq_msg.size ());
|
||||
ASSERT_EQ (send_msg, recv_msg);
|
||||
} else if (0 != (e & ~ZMQ_POLLOUT)) {
|
||||
ASSERT_TRUE(false) << "Unexpected event type " << events;
|
||||
ASSERT_TRUE (false) << "Unexpected event type " << events;
|
||||
}
|
||||
events = e;
|
||||
};
|
||||
|
||||
ASSERT_NO_THROW(active_poller.add(s.server, ZMQ_POLLIN, handler));
|
||||
ASSERT_NO_THROW (active_poller.add (s.server, ZMQ_POLLIN, handler));
|
||||
|
||||
// client sends message
|
||||
ASSERT_NO_THROW(s.client.send(send_msg));
|
||||
ASSERT_NO_THROW (s.client.send (send_msg));
|
||||
|
||||
ASSERT_EQ(1, active_poller.wait(std::chrono::milliseconds{-1}));
|
||||
ASSERT_EQ(events, ZMQ_POLLIN);
|
||||
ASSERT_EQ (1, active_poller.wait (std::chrono::milliseconds{-1}));
|
||||
ASSERT_EQ (events, ZMQ_POLLIN);
|
||||
|
||||
// Re-add server socket with pollout flag
|
||||
ASSERT_NO_THROW(active_poller.remove(s.server));
|
||||
ASSERT_NO_THROW(active_poller.add(s.server, ZMQ_POLLIN | ZMQ_POLLOUT, handler));
|
||||
ASSERT_EQ(1, active_poller.wait(std::chrono::milliseconds{-1}));
|
||||
ASSERT_EQ(events, ZMQ_POLLOUT);
|
||||
ASSERT_NO_THROW (active_poller.remove (s.server));
|
||||
ASSERT_NO_THROW (
|
||||
active_poller.add (s.server, ZMQ_POLLIN | ZMQ_POLLOUT, handler));
|
||||
ASSERT_EQ (1, active_poller.wait (std::chrono::milliseconds{-1}));
|
||||
ASSERT_EQ (events, ZMQ_POLLOUT);
|
||||
}
|
||||
|
||||
TEST(active_poller, add_invalid_socket_throws)
|
||||
TEST (active_poller, add_invalid_socket_throws)
|
||||
{
|
||||
zmq::context_t context;
|
||||
zmq::active_poller_t active_poller;
|
||||
zmq::socket_t a {context, zmq::socket_type::router};
|
||||
zmq::socket_t b {std::move (a)};
|
||||
ASSERT_THROW (active_poller.add (a, ZMQ_POLLIN, zmq::active_poller_t::handler_t {}),
|
||||
zmq::error_t);
|
||||
zmq::socket_t a{context, zmq::socket_type::router};
|
||||
zmq::socket_t b{std::move (a)};
|
||||
ASSERT_THROW (
|
||||
active_poller.add (a, ZMQ_POLLIN, zmq::active_poller_t::handler_t{}),
|
||||
zmq::error_t);
|
||||
}
|
||||
|
||||
TEST(active_poller, remove_invalid_socket_throws)
|
||||
TEST (active_poller, remove_invalid_socket_throws)
|
||||
{
|
||||
zmq::context_t context;
|
||||
zmq::socket_t socket {context, zmq::socket_type::router};
|
||||
zmq::socket_t socket{context, zmq::socket_type::router};
|
||||
zmq::active_poller_t active_poller;
|
||||
ASSERT_NO_THROW (active_poller.add (socket, ZMQ_POLLIN, zmq::active_poller_t::handler_t {}));
|
||||
ASSERT_NO_THROW (active_poller.add (socket, ZMQ_POLLIN,
|
||||
zmq::active_poller_t::handler_t{}));
|
||||
ASSERT_EQ (1u, active_poller.size ());
|
||||
std::vector<zmq::socket_t> sockets;
|
||||
sockets.emplace_back (std::move (socket));
|
||||
@ -231,75 +237,77 @@ TEST(active_poller, remove_invalid_socket_throws)
|
||||
ASSERT_EQ (1u, active_poller.size ());
|
||||
}
|
||||
|
||||
TEST(active_poller, wait_on_added_empty_handler)
|
||||
TEST (active_poller, wait_on_added_empty_handler)
|
||||
{
|
||||
server_client_setup s;
|
||||
ASSERT_NO_THROW (s.client.send ("Hi"));
|
||||
zmq::active_poller_t active_poller;
|
||||
zmq::active_poller_t::handler_t handler;
|
||||
ASSERT_NO_THROW (active_poller.add (s.server, ZMQ_POLLIN, handler));
|
||||
ASSERT_NO_THROW (active_poller.wait (std::chrono::milliseconds {-1}));
|
||||
ASSERT_NO_THROW (active_poller.wait (std::chrono::milliseconds{-1}));
|
||||
}
|
||||
|
||||
TEST(active_poller, modify_empty_throws)
|
||||
TEST (active_poller, modify_empty_throws)
|
||||
{
|
||||
zmq::context_t context;
|
||||
zmq::socket_t socket {context, zmq::socket_type::push};
|
||||
zmq::socket_t socket{context, zmq::socket_type::push};
|
||||
zmq::active_poller_t active_poller;
|
||||
ASSERT_THROW (active_poller.modify (socket, ZMQ_POLLIN), zmq::error_t);
|
||||
}
|
||||
|
||||
TEST(active_poller, modify_invalid_socket_throws)
|
||||
TEST (active_poller, modify_invalid_socket_throws)
|
||||
{
|
||||
zmq::context_t context;
|
||||
zmq::socket_t a {context, zmq::socket_type::push};
|
||||
zmq::socket_t b {std::move (a)};
|
||||
zmq::socket_t a{context, zmq::socket_type::push};
|
||||
zmq::socket_t b{std::move (a)};
|
||||
zmq::active_poller_t active_poller;
|
||||
ASSERT_THROW (active_poller.modify (a, ZMQ_POLLIN), zmq::error_t);
|
||||
}
|
||||
|
||||
TEST(active_poller, modify_not_added_throws)
|
||||
TEST (active_poller, modify_not_added_throws)
|
||||
{
|
||||
zmq::context_t context;
|
||||
zmq::socket_t a {context, zmq::socket_type::push};
|
||||
zmq::socket_t b {context, zmq::socket_type::push};
|
||||
zmq::socket_t a{context, zmq::socket_type::push};
|
||||
zmq::socket_t b{context, zmq::socket_type::push};
|
||||
zmq::active_poller_t active_poller;
|
||||
ASSERT_NO_THROW (active_poller.add (a, ZMQ_POLLIN, zmq::active_poller_t::handler_t {}));
|
||||
ASSERT_NO_THROW (
|
||||
active_poller.add (a, ZMQ_POLLIN, zmq::active_poller_t::handler_t{}));
|
||||
ASSERT_THROW (active_poller.modify (b, ZMQ_POLLIN), zmq::error_t);
|
||||
}
|
||||
|
||||
TEST(active_poller, modify_simple)
|
||||
TEST (active_poller, modify_simple)
|
||||
{
|
||||
zmq::context_t context;
|
||||
zmq::socket_t a {context, zmq::socket_type::push};
|
||||
zmq::socket_t a{context, zmq::socket_type::push};
|
||||
zmq::active_poller_t active_poller;
|
||||
ASSERT_NO_THROW (active_poller.add (a, ZMQ_POLLIN, zmq::active_poller_t::handler_t {}));
|
||||
ASSERT_NO_THROW (active_poller.modify (a, ZMQ_POLLIN|ZMQ_POLLOUT));
|
||||
ASSERT_NO_THROW (
|
||||
active_poller.add (a, ZMQ_POLLIN, zmq::active_poller_t::handler_t{}));
|
||||
ASSERT_NO_THROW (active_poller.modify (a, ZMQ_POLLIN | ZMQ_POLLOUT));
|
||||
}
|
||||
|
||||
TEST(active_poller, poll_client_server)
|
||||
TEST (active_poller, poll_client_server)
|
||||
{
|
||||
// Setup server and client
|
||||
server_client_setup s;
|
||||
|
||||
// Setup active_poller
|
||||
zmq::active_poller_t active_poller;
|
||||
ASSERT_NO_THROW(active_poller.add(s.server, ZMQ_POLLIN, s.handler));
|
||||
ASSERT_NO_THROW (active_poller.add (s.server, ZMQ_POLLIN, s.handler));
|
||||
|
||||
// client sends message
|
||||
ASSERT_NO_THROW(s.client.send("Hi"));
|
||||
ASSERT_NO_THROW (s.client.send ("Hi"));
|
||||
|
||||
// wait for message and verify events
|
||||
ASSERT_NO_THROW(active_poller.wait(std::chrono::milliseconds{500}));
|
||||
ASSERT_EQ(s.events, ZMQ_POLLIN);
|
||||
ASSERT_NO_THROW (active_poller.wait (std::chrono::milliseconds{500}));
|
||||
ASSERT_EQ (s.events, ZMQ_POLLIN);
|
||||
|
||||
// Modify server socket with pollout flag
|
||||
ASSERT_NO_THROW(active_poller.modify(s.server, ZMQ_POLLIN | ZMQ_POLLOUT));
|
||||
ASSERT_EQ(1, active_poller.wait(std::chrono::milliseconds{500}));
|
||||
ASSERT_EQ(s.events, ZMQ_POLLIN | ZMQ_POLLOUT);
|
||||
ASSERT_NO_THROW (active_poller.modify (s.server, ZMQ_POLLIN | ZMQ_POLLOUT));
|
||||
ASSERT_EQ (1, active_poller.wait (std::chrono::milliseconds{500}));
|
||||
ASSERT_EQ (s.events, ZMQ_POLLIN | ZMQ_POLLOUT);
|
||||
}
|
||||
|
||||
TEST(active_poller, wait_one_return)
|
||||
TEST (active_poller, wait_one_return)
|
||||
{
|
||||
// Setup server and client
|
||||
server_client_setup s;
|
||||
@ -308,33 +316,32 @@ TEST(active_poller, wait_one_return)
|
||||
|
||||
// Setup active_poller
|
||||
zmq::active_poller_t active_poller;
|
||||
ASSERT_NO_THROW(active_poller.add(s.server, ZMQ_POLLIN, [&count](short) {
|
||||
++count;
|
||||
}));
|
||||
ASSERT_NO_THROW (
|
||||
active_poller.add (s.server, ZMQ_POLLIN, [&count](short) { ++count; }));
|
||||
|
||||
// client sends message
|
||||
ASSERT_NO_THROW(s.client.send("Hi"));
|
||||
ASSERT_NO_THROW (s.client.send ("Hi"));
|
||||
|
||||
// wait for message and verify events
|
||||
ASSERT_EQ(1, active_poller.wait(std::chrono::milliseconds{500}));
|
||||
ASSERT_EQ(1u, count);
|
||||
ASSERT_EQ (1, active_poller.wait (std::chrono::milliseconds{500}));
|
||||
ASSERT_EQ (1u, count);
|
||||
}
|
||||
|
||||
TEST(active_poller, wait_on_move_constructed_active_poller)
|
||||
TEST (active_poller, wait_on_move_constructed_active_poller)
|
||||
{
|
||||
server_client_setup s;
|
||||
ASSERT_NO_THROW (s.client.send ("Hi"));
|
||||
zmq::active_poller_t a;
|
||||
zmq::active_poller_t::handler_t handler;
|
||||
ASSERT_NO_THROW (a.add (s.server, ZMQ_POLLIN, handler));
|
||||
zmq::active_poller_t b {std::move (a)};
|
||||
ASSERT_EQ(1u, b.size ());
|
||||
zmq::active_poller_t b{std::move (a)};
|
||||
ASSERT_EQ (1u, b.size ());
|
||||
/// \todo the actual error code should be checked
|
||||
ASSERT_THROW(a.wait(std::chrono::milliseconds{10}), zmq::error_t);
|
||||
ASSERT_TRUE (b.wait (std::chrono::milliseconds {-1}));
|
||||
ASSERT_THROW (a.wait (std::chrono::milliseconds{10}), zmq::error_t);
|
||||
ASSERT_TRUE (b.wait (std::chrono::milliseconds{-1}));
|
||||
}
|
||||
|
||||
TEST(active_poller, wait_on_move_assigned_active_poller)
|
||||
TEST (active_poller, wait_on_move_assigned_active_poller)
|
||||
{
|
||||
server_client_setup s;
|
||||
ASSERT_NO_THROW (s.client.send ("Hi"));
|
||||
@ -343,38 +350,37 @@ TEST(active_poller, wait_on_move_assigned_active_poller)
|
||||
ASSERT_NO_THROW (a.add (s.server, ZMQ_POLLIN, handler));
|
||||
zmq::active_poller_t b;
|
||||
b = {std::move (a)};
|
||||
ASSERT_EQ(1u, b.size ());
|
||||
ASSERT_EQ (1u, b.size ());
|
||||
/// \todo the actual error code should be checked
|
||||
ASSERT_THROW(a.wait(std::chrono::milliseconds{10}), zmq::error_t);
|
||||
ASSERT_TRUE (b.wait (std::chrono::milliseconds {-1}));
|
||||
ASSERT_THROW (a.wait (std::chrono::milliseconds{10}), zmq::error_t);
|
||||
ASSERT_TRUE (b.wait (std::chrono::milliseconds{-1}));
|
||||
}
|
||||
|
||||
TEST(active_poller, received_on_move_constructed_active_poller)
|
||||
TEST (active_poller, received_on_move_constructed_active_poller)
|
||||
{
|
||||
// Setup server and client
|
||||
server_client_setup s;
|
||||
int count = 0;
|
||||
// Setup active_poller a
|
||||
zmq::active_poller_t a;
|
||||
ASSERT_NO_THROW(a.add(s.server, ZMQ_POLLIN, [&count](short) {
|
||||
++count;
|
||||
}));
|
||||
ASSERT_NO_THROW (
|
||||
a.add (s.server, ZMQ_POLLIN, [&count](short) { ++count; }));
|
||||
// client sends message
|
||||
ASSERT_NO_THROW(s.client.send("Hi"));
|
||||
ASSERT_NO_THROW (s.client.send ("Hi"));
|
||||
// wait for message and verify it is received
|
||||
ASSERT_EQ(1, a.wait(std::chrono::milliseconds{500}));
|
||||
ASSERT_EQ(1u, count);
|
||||
ASSERT_EQ (1, a.wait (std::chrono::milliseconds{500}));
|
||||
ASSERT_EQ (1u, count);
|
||||
// Move construct active_poller b
|
||||
zmq::active_poller_t b{std::move(a)};
|
||||
zmq::active_poller_t b{std::move (a)};
|
||||
// client sends message again
|
||||
ASSERT_NO_THROW(s.client.send("Hi"));
|
||||
ASSERT_NO_THROW (s.client.send ("Hi"));
|
||||
// wait for message and verify it is received
|
||||
ASSERT_EQ(1, b.wait(std::chrono::milliseconds{500}));
|
||||
ASSERT_EQ(2u, count);
|
||||
ASSERT_EQ (1, b.wait (std::chrono::milliseconds{500}));
|
||||
ASSERT_EQ (2u, count);
|
||||
}
|
||||
|
||||
|
||||
TEST(active_poller, remove_from_handler)
|
||||
TEST (active_poller, remove_from_handler)
|
||||
{
|
||||
constexpr auto ITER_NO = 10;
|
||||
|
||||
@ -387,28 +393,29 @@ TEST(active_poller, remove_from_handler)
|
||||
zmq::active_poller_t active_poller;
|
||||
int count = 0;
|
||||
for (auto i = 0; i < ITER_NO; ++i) {
|
||||
ASSERT_NO_THROW(active_poller.add(setup_list[i].server, ZMQ_POLLIN, [&,i](short events) {
|
||||
ASSERT_EQ(events, ZMQ_POLLIN);
|
||||
active_poller.remove(setup_list[ITER_NO-i-1].server);
|
||||
ASSERT_EQ(ITER_NO-i-1, active_poller.size());
|
||||
}));
|
||||
ASSERT_NO_THROW (active_poller.add (
|
||||
setup_list[i].server, ZMQ_POLLIN, [&, i](short events) {
|
||||
ASSERT_EQ (events, ZMQ_POLLIN);
|
||||
active_poller.remove (setup_list[ITER_NO - i - 1].server);
|
||||
ASSERT_EQ (ITER_NO - i - 1, active_poller.size ());
|
||||
}));
|
||||
++count;
|
||||
}
|
||||
ASSERT_EQ(ITER_NO, active_poller.size());
|
||||
ASSERT_EQ (ITER_NO, active_poller.size ());
|
||||
// Clients send messages
|
||||
for (auto & s : setup_list) {
|
||||
ASSERT_NO_THROW(s.client.send("Hi"));
|
||||
for (auto &s : setup_list) {
|
||||
ASSERT_NO_THROW (s.client.send ("Hi"));
|
||||
}
|
||||
|
||||
// Wait for all servers to receive a message
|
||||
for (auto & s : setup_list) {
|
||||
zmq::pollitem_t items [] = { { s.server, 0, ZMQ_POLLIN, 0 } };
|
||||
zmq::poll (&items [0], 1);
|
||||
for (auto &s : setup_list) {
|
||||
zmq::pollitem_t items[] = {{s.server, 0, ZMQ_POLLIN, 0}};
|
||||
zmq::poll (&items[0], 1);
|
||||
}
|
||||
|
||||
// Fire all handlers in one wait
|
||||
ASSERT_EQ(ITER_NO, active_poller.wait (std::chrono::milliseconds{-1}));
|
||||
ASSERT_EQ(ITER_NO, count);
|
||||
ASSERT_EQ (ITER_NO, active_poller.wait (std::chrono::milliseconds{-1}));
|
||||
ASSERT_EQ (ITER_NO, count);
|
||||
}
|
||||
|
||||
#endif
|
||||
|
@ -1,15 +1,15 @@
|
||||
#include <gtest/gtest.h>
|
||||
#include <zmq.hpp>
|
||||
|
||||
TEST(context, create_default_destroy)
|
||||
TEST (context, create_default_destroy)
|
||||
{
|
||||
zmq::context_t context;
|
||||
}
|
||||
|
||||
TEST(context, create_close)
|
||||
TEST (context, create_close)
|
||||
{
|
||||
zmq::context_t context;
|
||||
context.close();
|
||||
|
||||
ASSERT_EQ(NULL, (void*)context);
|
||||
context.close ();
|
||||
|
||||
ASSERT_EQ (NULL, (void *) context);
|
||||
}
|
||||
|
@ -1,7 +1,7 @@
|
||||
#include "gtest/gtest.h"
|
||||
|
||||
int main(int argc, char** argv)
|
||||
int main (int argc, char **argv)
|
||||
{
|
||||
::testing::InitGoogleTest(&argc, argv);
|
||||
return RUN_ALL_TESTS();
|
||||
::testing::InitGoogleTest (&argc, argv);
|
||||
return RUN_ALL_TESTS ();
|
||||
}
|
||||
|
@ -2,8 +2,10 @@
|
||||
#include <zmq.hpp>
|
||||
|
||||
#if defined(ZMQ_CPP11)
|
||||
static_assert(!std::is_copy_constructible<zmq::message_t>::value, "message_t should not be copy-constructible");
|
||||
static_assert(!std::is_copy_assignable<zmq::message_t>::value, "message_t should not be copy-assignable");
|
||||
static_assert (!std::is_copy_constructible<zmq::message_t>::value,
|
||||
"message_t should not be copy-constructible");
|
||||
static_assert (!std::is_copy_assignable<zmq::message_t>::value,
|
||||
"message_t should not be copy-assignable");
|
||||
#endif
|
||||
|
||||
TEST (message, constructor_default)
|
||||
@ -12,7 +14,7 @@ TEST (message, constructor_default)
|
||||
ASSERT_EQ (0u, message.size ());
|
||||
}
|
||||
|
||||
const char* const data = "Hi";
|
||||
const char *const data = "Hi";
|
||||
|
||||
TEST (message, constructor_iterators)
|
||||
{
|
||||
@ -30,7 +32,8 @@ TEST (message, constructor_pointer_size)
|
||||
ASSERT_EQ (0, memcmp (data, hi_msg.data (), 2));
|
||||
}
|
||||
|
||||
TEST (message, constructor_char_array) {
|
||||
TEST (message, constructor_char_array)
|
||||
{
|
||||
const zmq::message_t hi_msg (data, strlen (data));
|
||||
ASSERT_EQ (2u, hi_msg.size ());
|
||||
ASSERT_EQ (0, memcmp (data, hi_msg.data (), 2));
|
||||
@ -49,7 +52,7 @@ TEST (message, constructor_container)
|
||||
#ifdef ZMQ_HAS_RVALUE_REFS
|
||||
TEST (message, constructor_move)
|
||||
{
|
||||
zmq::message_t hi_msg (zmq::message_t(data, strlen (data)));
|
||||
zmq::message_t hi_msg (zmq::message_t (data, strlen (data)));
|
||||
}
|
||||
|
||||
TEST (message, assign_move_empty_before)
|
||||
@ -63,50 +66,55 @@ TEST (message, assign_move_empty_before)
|
||||
TEST (message, assign_move_empty_after)
|
||||
{
|
||||
zmq::message_t hi_msg (data, strlen (data));
|
||||
hi_msg = zmq::message_t();
|
||||
hi_msg = zmq::message_t ();
|
||||
ASSERT_EQ (0u, hi_msg.size ());
|
||||
}
|
||||
|
||||
TEST (message, assign_move_empty_before_and_after)
|
||||
{
|
||||
zmq::message_t hi_msg;
|
||||
hi_msg = zmq::message_t();
|
||||
hi_msg = zmq::message_t ();
|
||||
ASSERT_EQ (0u, hi_msg.size ());
|
||||
}
|
||||
#endif
|
||||
|
||||
TEST (message, equality_self) {
|
||||
TEST (message, equality_self)
|
||||
{
|
||||
const zmq::message_t hi_msg (data, strlen (data));
|
||||
ASSERT_EQ (hi_msg, hi_msg);
|
||||
}
|
||||
|
||||
TEST (message, equality_equal) {
|
||||
TEST (message, equality_equal)
|
||||
{
|
||||
const zmq::message_t hi_msg_a (data, strlen (data));
|
||||
const zmq::message_t hi_msg_b (data, strlen (data));
|
||||
ASSERT_EQ (hi_msg_a, hi_msg_b);
|
||||
}
|
||||
|
||||
TEST (message, equality_equal_empty) {
|
||||
TEST (message, equality_equal_empty)
|
||||
{
|
||||
const zmq::message_t msg_a;
|
||||
const zmq::message_t msg_b;
|
||||
ASSERT_EQ (msg_a, msg_b);
|
||||
}
|
||||
|
||||
TEST (message, equality_non_equal) {
|
||||
TEST (message, equality_non_equal)
|
||||
{
|
||||
const zmq::message_t msg_a ("Hi", 2);
|
||||
const zmq::message_t msg_b ("Hello", 5);
|
||||
ASSERT_NE (msg_a, msg_b);
|
||||
}
|
||||
|
||||
TEST (message, equality_non_equal_rhs_empty) {
|
||||
TEST (message, equality_non_equal_rhs_empty)
|
||||
{
|
||||
const zmq::message_t msg_a ("Hi", 2);
|
||||
const zmq::message_t msg_b;
|
||||
ASSERT_NE (msg_a, msg_b);
|
||||
}
|
||||
|
||||
TEST (message, equality_non_equal_lhs_empty) {
|
||||
TEST (message, equality_non_equal_lhs_empty)
|
||||
{
|
||||
const zmq::message_t msg_a;
|
||||
const zmq::message_t msg_b ("Hi", 2);
|
||||
ASSERT_NE (msg_a, msg_b);
|
||||
}
|
||||
|
||||
|
@ -4,168 +4,170 @@
|
||||
#ifdef ZMQ_HAS_RVALUE_REFS
|
||||
/// \todo split this up into separate test cases
|
||||
///
|
||||
TEST(multipart, legacy_test)
|
||||
TEST (multipart, legacy_test)
|
||||
{
|
||||
using namespace zmq;
|
||||
|
||||
bool ok = true; (void) ok;
|
||||
float num = 0; (void) num;
|
||||
|
||||
bool ok = true;
|
||||
(void) ok;
|
||||
float num = 0;
|
||||
(void) num;
|
||||
std::string str = "";
|
||||
message_t msg;
|
||||
|
||||
// Create two PAIR sockets and connect over inproc
|
||||
context_t context(1);
|
||||
socket_t output(context, ZMQ_PAIR);
|
||||
socket_t input(context, ZMQ_PAIR);
|
||||
output.bind("inproc://multipart.test");
|
||||
input.connect("inproc://multipart.test");
|
||||
context_t context (1);
|
||||
socket_t output (context, ZMQ_PAIR);
|
||||
socket_t input (context, ZMQ_PAIR);
|
||||
output.bind ("inproc://multipart.test");
|
||||
input.connect ("inproc://multipart.test");
|
||||
|
||||
// Test send and receive of single-frame message
|
||||
multipart_t multipart;
|
||||
assert(multipart.empty());
|
||||
assert (multipart.empty ());
|
||||
|
||||
multipart.push(message_t("Hello", 5));
|
||||
assert(multipart.size() == 1);
|
||||
multipart.push (message_t ("Hello", 5));
|
||||
assert (multipart.size () == 1);
|
||||
|
||||
ok = multipart.send(output);
|
||||
assert(multipart.empty());
|
||||
assert(ok);
|
||||
ok = multipart.send (output);
|
||||
assert (multipart.empty ());
|
||||
assert (ok);
|
||||
|
||||
ok = multipart.recv(input);
|
||||
assert(multipart.size() == 1);
|
||||
assert(ok);
|
||||
ok = multipart.recv (input);
|
||||
assert (multipart.size () == 1);
|
||||
assert (ok);
|
||||
|
||||
msg = multipart.pop();
|
||||
assert(multipart.empty());
|
||||
assert(std::string(msg.data<char>(), msg.size()) == "Hello");
|
||||
msg = multipart.pop ();
|
||||
assert (multipart.empty ());
|
||||
assert (std::string (msg.data<char> (), msg.size ()) == "Hello");
|
||||
|
||||
// Test send and receive of multi-frame message
|
||||
multipart.addstr("A");
|
||||
multipart.addstr("BB");
|
||||
multipart.addstr("CCC");
|
||||
assert(multipart.size() == 3);
|
||||
multipart.addstr ("A");
|
||||
multipart.addstr ("BB");
|
||||
multipart.addstr ("CCC");
|
||||
assert (multipart.size () == 3);
|
||||
|
||||
multipart_t copy = multipart.clone();
|
||||
assert(copy.size() == 3);
|
||||
multipart_t copy = multipart.clone ();
|
||||
assert (copy.size () == 3);
|
||||
|
||||
ok = copy.send(output);
|
||||
assert(copy.empty());
|
||||
assert(ok);
|
||||
ok = copy.send (output);
|
||||
assert (copy.empty ());
|
||||
assert (ok);
|
||||
|
||||
ok = copy.recv(input);
|
||||
assert(copy.size() == 3);
|
||||
assert(ok);
|
||||
assert(copy.equal(&multipart));
|
||||
ok = copy.recv (input);
|
||||
assert (copy.size () == 3);
|
||||
assert (ok);
|
||||
assert (copy.equal (&multipart));
|
||||
|
||||
multipart.clear();
|
||||
assert(multipart.empty());
|
||||
multipart.clear ();
|
||||
assert (multipart.empty ());
|
||||
|
||||
// Test message frame manipulation
|
||||
multipart.add(message_t("Frame5", 6));
|
||||
multipart.addstr("Frame6");
|
||||
multipart.addstr("Frame7");
|
||||
multipart.addtyp(8.0f);
|
||||
multipart.addmem("Frame9", 6);
|
||||
multipart.push(message_t("Frame4", 6));
|
||||
multipart.pushstr("Frame3");
|
||||
multipart.pushstr("Frame2");
|
||||
multipart.pushtyp(1.0f);
|
||||
multipart.pushmem("Frame0", 6);
|
||||
assert(multipart.size() == 10);
|
||||
multipart.add (message_t ("Frame5", 6));
|
||||
multipart.addstr ("Frame6");
|
||||
multipart.addstr ("Frame7");
|
||||
multipart.addtyp (8.0f);
|
||||
multipart.addmem ("Frame9", 6);
|
||||
multipart.push (message_t ("Frame4", 6));
|
||||
multipart.pushstr ("Frame3");
|
||||
multipart.pushstr ("Frame2");
|
||||
multipart.pushtyp (1.0f);
|
||||
multipart.pushmem ("Frame0", 6);
|
||||
assert (multipart.size () == 10);
|
||||
|
||||
msg = multipart.remove();
|
||||
assert(multipart.size() == 9);
|
||||
assert(std::string(msg.data<char>(), msg.size()) == "Frame9");
|
||||
msg = multipart.remove ();
|
||||
assert (multipart.size () == 9);
|
||||
assert (std::string (msg.data<char> (), msg.size ()) == "Frame9");
|
||||
|
||||
msg = multipart.pop();
|
||||
assert(multipart.size() == 8);
|
||||
assert(std::string(msg.data<char>(), msg.size()) == "Frame0");
|
||||
msg = multipart.pop ();
|
||||
assert (multipart.size () == 8);
|
||||
assert (std::string (msg.data<char> (), msg.size ()) == "Frame0");
|
||||
|
||||
num = multipart.poptyp<float>();
|
||||
assert(multipart.size() == 7);
|
||||
assert(num == 1.0f);
|
||||
num = multipart.poptyp<float> ();
|
||||
assert (multipart.size () == 7);
|
||||
assert (num == 1.0f);
|
||||
|
||||
str = multipart.popstr();
|
||||
assert(multipart.size() == 6);
|
||||
assert(str == "Frame2");
|
||||
str = multipart.popstr ();
|
||||
assert (multipart.size () == 6);
|
||||
assert (str == "Frame2");
|
||||
|
||||
str = multipart.popstr();
|
||||
assert(multipart.size() == 5);
|
||||
assert(str == "Frame3");
|
||||
str = multipart.popstr ();
|
||||
assert (multipart.size () == 5);
|
||||
assert (str == "Frame3");
|
||||
|
||||
str = multipart.popstr();
|
||||
assert(multipart.size() == 4);
|
||||
assert(str == "Frame4");
|
||||
str = multipart.popstr ();
|
||||
assert (multipart.size () == 4);
|
||||
assert (str == "Frame4");
|
||||
|
||||
str = multipart.popstr();
|
||||
assert(multipart.size() == 3);
|
||||
assert(str == "Frame5");
|
||||
str = multipart.popstr ();
|
||||
assert (multipart.size () == 3);
|
||||
assert (str == "Frame5");
|
||||
|
||||
str = multipart.popstr();
|
||||
assert(multipart.size() == 2);
|
||||
assert(str == "Frame6");
|
||||
str = multipart.popstr ();
|
||||
assert (multipart.size () == 2);
|
||||
assert (str == "Frame6");
|
||||
|
||||
str = multipart.popstr();
|
||||
assert(multipart.size() == 1);
|
||||
assert(str == "Frame7");
|
||||
str = multipart.popstr ();
|
||||
assert (multipart.size () == 1);
|
||||
assert (str == "Frame7");
|
||||
|
||||
num = multipart.poptyp<float>();
|
||||
assert(multipart.empty());
|
||||
assert(num == 8.0f);
|
||||
num = multipart.poptyp<float> ();
|
||||
assert (multipart.empty ());
|
||||
assert (num == 8.0f);
|
||||
|
||||
// Test message constructors and concatenation
|
||||
multipart_t head("One", 3);
|
||||
head.addstr("Two");
|
||||
assert(head.size() == 2);
|
||||
multipart_t head ("One", 3);
|
||||
head.addstr ("Two");
|
||||
assert (head.size () == 2);
|
||||
|
||||
multipart_t tail(std::string("One-hundred"));
|
||||
tail.pushstr("Ninety-nine");
|
||||
assert(tail.size() == 2);
|
||||
multipart_t tail (std::string ("One-hundred"));
|
||||
tail.pushstr ("Ninety-nine");
|
||||
assert (tail.size () == 2);
|
||||
|
||||
multipart_t tmp(message_t("Fifty", 5));
|
||||
assert(tmp.size() == 1);
|
||||
multipart_t tmp (message_t ("Fifty", 5));
|
||||
assert (tmp.size () == 1);
|
||||
|
||||
multipart_t mid = multipart_t::create(49.0f);
|
||||
mid.append(std::move(tmp));
|
||||
assert(mid.size() == 2);
|
||||
assert(tmp.empty());
|
||||
multipart_t mid = multipart_t::create (49.0f);
|
||||
mid.append (std::move (tmp));
|
||||
assert (mid.size () == 2);
|
||||
assert (tmp.empty ());
|
||||
|
||||
multipart_t merged(std::move(mid));
|
||||
merged.prepend(std::move(head));
|
||||
merged.append(std::move(tail));
|
||||
assert(merged.size() == 6);
|
||||
assert(head.empty());
|
||||
assert(tail.empty());
|
||||
multipart_t merged (std::move (mid));
|
||||
merged.prepend (std::move (head));
|
||||
merged.append (std::move (tail));
|
||||
assert (merged.size () == 6);
|
||||
assert (head.empty ());
|
||||
assert (tail.empty ());
|
||||
|
||||
ok = merged.send(output);
|
||||
assert(merged.empty());
|
||||
assert(ok);
|
||||
ok = merged.send (output);
|
||||
assert (merged.empty ());
|
||||
assert (ok);
|
||||
|
||||
multipart_t received(input);
|
||||
assert(received.size() == 6);
|
||||
multipart_t received (input);
|
||||
assert (received.size () == 6);
|
||||
|
||||
str = received.popstr();
|
||||
assert(received.size() == 5);
|
||||
assert(str == "One");
|
||||
str = received.popstr ();
|
||||
assert (received.size () == 5);
|
||||
assert (str == "One");
|
||||
|
||||
str = received.popstr();
|
||||
assert(received.size() == 4);
|
||||
assert(str == "Two");
|
||||
str = received.popstr ();
|
||||
assert (received.size () == 4);
|
||||
assert (str == "Two");
|
||||
|
||||
num = received.poptyp<float>();
|
||||
assert(received.size() == 3);
|
||||
assert(num == 49.0f);
|
||||
num = received.poptyp<float> ();
|
||||
assert (received.size () == 3);
|
||||
assert (num == 49.0f);
|
||||
|
||||
str = received.popstr();
|
||||
assert(received.size() == 2);
|
||||
assert(str == "Fifty");
|
||||
str = received.popstr ();
|
||||
assert (received.size () == 2);
|
||||
assert (str == "Fifty");
|
||||
|
||||
str = received.popstr();
|
||||
assert(received.size() == 1);
|
||||
assert(str == "Ninety-nine");
|
||||
str = received.popstr ();
|
||||
assert (received.size () == 1);
|
||||
assert (str == "Ninety-nine");
|
||||
|
||||
str = received.popstr();
|
||||
assert(received.empty());
|
||||
assert(str == "One-hundred");
|
||||
str = received.popstr ();
|
||||
assert (received.empty ());
|
||||
assert (str == "One-hundred");
|
||||
}
|
||||
#endif
|
||||
|
180
tests/poller.cpp
180
tests/poller.cpp
@ -5,151 +5,153 @@
|
||||
#include <array>
|
||||
#include <memory>
|
||||
|
||||
TEST(poller, create_destroy)
|
||||
TEST (poller, create_destroy)
|
||||
{
|
||||
zmq::poller_t<> poller;
|
||||
}
|
||||
|
||||
static_assert(!std::is_copy_constructible<zmq::poller_t<>>::value, "poller_t should not be copy-constructible");
|
||||
static_assert(!std::is_copy_assignable<zmq::poller_t<>>::value, "poller_t should not be copy-assignable");
|
||||
static_assert (!std::is_copy_constructible<zmq::poller_t<>>::value,
|
||||
"poller_t should not be copy-constructible");
|
||||
static_assert (!std::is_copy_assignable<zmq::poller_t<>>::value,
|
||||
"poller_t should not be copy-assignable");
|
||||
|
||||
TEST(poller, move_construct_empty)
|
||||
TEST (poller, move_construct_empty)
|
||||
{
|
||||
zmq::poller_t<> a;
|
||||
zmq::poller_t<> b = std::move (a);
|
||||
}
|
||||
|
||||
TEST(poller, move_assign_empty)
|
||||
TEST (poller, move_assign_empty)
|
||||
{
|
||||
zmq::poller_t<> a;
|
||||
zmq::poller_t<> b;
|
||||
b = std::move(a);
|
||||
b = std::move (a);
|
||||
}
|
||||
|
||||
TEST(poller, move_construct_non_empty)
|
||||
TEST (poller, move_construct_non_empty)
|
||||
{
|
||||
zmq::context_t context;
|
||||
zmq::socket_t socket{context, zmq::socket_type::router};
|
||||
|
||||
zmq::poller_t<> a;
|
||||
a.add(socket, ZMQ_POLLIN, nullptr);
|
||||
a.add (socket, ZMQ_POLLIN, nullptr);
|
||||
zmq::poller_t<> b = std::move (a);
|
||||
}
|
||||
|
||||
TEST(poller, move_assign_non_empty)
|
||||
TEST (poller, move_assign_non_empty)
|
||||
{
|
||||
zmq::context_t context;
|
||||
zmq::socket_t socket{context, zmq::socket_type::router};
|
||||
|
||||
zmq::poller_t<> a;
|
||||
a.add(socket, ZMQ_POLLIN, nullptr);
|
||||
a.add (socket, ZMQ_POLLIN, nullptr);
|
||||
zmq::poller_t<> b;
|
||||
b = std::move(a);
|
||||
b = std::move (a);
|
||||
}
|
||||
|
||||
TEST(poller, add_nullptr)
|
||||
TEST (poller, add_nullptr)
|
||||
{
|
||||
zmq::context_t context;
|
||||
zmq::socket_t socket{context, zmq::socket_type::router};
|
||||
zmq::poller_t<> poller;
|
||||
ASSERT_NO_THROW(poller.add(socket, ZMQ_POLLIN, nullptr));
|
||||
ASSERT_NO_THROW (poller.add (socket, ZMQ_POLLIN, nullptr));
|
||||
}
|
||||
|
||||
TEST(poller, add_non_nullptr)
|
||||
TEST (poller, add_non_nullptr)
|
||||
{
|
||||
zmq::context_t context;
|
||||
zmq::socket_t socket{context, zmq::socket_type::router};
|
||||
zmq::poller_t<> poller;
|
||||
int i;
|
||||
ASSERT_NO_THROW(poller.add(socket, ZMQ_POLLIN, &i));
|
||||
ASSERT_NO_THROW (poller.add (socket, ZMQ_POLLIN, &i));
|
||||
}
|
||||
|
||||
TEST(poller, add_handler_invalid_events_type)
|
||||
TEST (poller, add_handler_invalid_events_type)
|
||||
{
|
||||
/// \todo is it good that this is accepted? should probably already be
|
||||
/// \todo is it good that this is accepted? should probably already be
|
||||
/// checked by zmq_poller_add/modify in libzmq:
|
||||
/// https://github.com/zeromq/libzmq/issues/3088
|
||||
zmq::context_t context;
|
||||
zmq::socket_t socket{context, zmq::socket_type::router};
|
||||
zmq::poller_t<> poller;
|
||||
short invalid_events_type = 2 << 10;
|
||||
ASSERT_NO_THROW(poller.add(socket, invalid_events_type, nullptr));
|
||||
ASSERT_NO_THROW (poller.add (socket, invalid_events_type, nullptr));
|
||||
}
|
||||
|
||||
TEST(poller, add_handler_twice_throws)
|
||||
TEST (poller, add_handler_twice_throws)
|
||||
{
|
||||
zmq::context_t context;
|
||||
zmq::socket_t socket{context, zmq::socket_type::router};
|
||||
zmq::poller_t<> poller;
|
||||
poller.add(socket, ZMQ_POLLIN, nullptr);
|
||||
poller.add (socket, ZMQ_POLLIN, nullptr);
|
||||
/// \todo the actual error code should be checked
|
||||
ASSERT_THROW(poller.add(socket, ZMQ_POLLIN, nullptr), zmq::error_t);
|
||||
ASSERT_THROW (poller.add (socket, ZMQ_POLLIN, nullptr), zmq::error_t);
|
||||
}
|
||||
|
||||
TEST(poller, wait_with_no_handlers_throws)
|
||||
TEST (poller, wait_with_no_handlers_throws)
|
||||
{
|
||||
zmq::poller_t<> poller;
|
||||
std::vector<zmq_poller_event_t> events;
|
||||
/// \todo the actual error code should be checked
|
||||
ASSERT_THROW(poller.wait_all(events, std::chrono::milliseconds{10}), zmq::error_t);
|
||||
ASSERT_THROW (poller.wait_all (events, std::chrono::milliseconds{10}),
|
||||
zmq::error_t);
|
||||
}
|
||||
|
||||
TEST(poller, remove_unregistered_throws)
|
||||
TEST (poller, remove_unregistered_throws)
|
||||
{
|
||||
zmq::context_t context;
|
||||
zmq::socket_t socket{context, zmq::socket_type::router};
|
||||
zmq::poller_t<> poller;
|
||||
/// \todo the actual error code should be checked
|
||||
ASSERT_THROW(poller.remove(socket), zmq::error_t);
|
||||
ASSERT_THROW (poller.remove (socket), zmq::error_t);
|
||||
}
|
||||
|
||||
TEST(poller, remove_registered_empty)
|
||||
TEST (poller, remove_registered_empty)
|
||||
{
|
||||
zmq::context_t context;
|
||||
zmq::socket_t socket{context, zmq::socket_type::router};
|
||||
zmq::poller_t<> poller;
|
||||
poller.add(socket, ZMQ_POLLIN, nullptr);
|
||||
ASSERT_NO_THROW(poller.remove(socket));
|
||||
poller.add (socket, ZMQ_POLLIN, nullptr);
|
||||
ASSERT_NO_THROW (poller.remove (socket));
|
||||
}
|
||||
|
||||
TEST(poller, remove_registered_non_empty)
|
||||
TEST (poller, remove_registered_non_empty)
|
||||
{
|
||||
zmq::context_t context;
|
||||
zmq::socket_t socket{context, zmq::socket_type::router};
|
||||
zmq::poller_t<> poller;
|
||||
poller.add(socket, ZMQ_POLLIN, nullptr);
|
||||
ASSERT_NO_THROW(poller.remove(socket));
|
||||
poller.add (socket, ZMQ_POLLIN, nullptr);
|
||||
ASSERT_NO_THROW (poller.remove (socket));
|
||||
}
|
||||
|
||||
TEST(poller, poll_basic)
|
||||
TEST (poller, poll_basic)
|
||||
{
|
||||
common_server_client_setup s;
|
||||
|
||||
ASSERT_NO_THROW(s.client.send("Hi"));
|
||||
ASSERT_NO_THROW (s.client.send ("Hi"));
|
||||
|
||||
zmq::poller_t<int> poller;
|
||||
std::vector<zmq_poller_event_t> events{1};
|
||||
int i = 0;
|
||||
ASSERT_NO_THROW(poller.add(s.server, ZMQ_POLLIN, &i));
|
||||
ASSERT_EQ(1, poller.wait_all(events, std::chrono::milliseconds{-1}));
|
||||
ASSERT_EQ(s.server, events[0].socket);
|
||||
ASSERT_EQ(&i, events[0].user_data);
|
||||
ASSERT_NO_THROW (poller.add (s.server, ZMQ_POLLIN, &i));
|
||||
ASSERT_EQ (1, poller.wait_all (events, std::chrono::milliseconds{-1}));
|
||||
ASSERT_EQ (s.server, events[0].socket);
|
||||
ASSERT_EQ (&i, events[0].user_data);
|
||||
}
|
||||
|
||||
TEST(poller, add_invalid_socket_throws)
|
||||
TEST (poller, add_invalid_socket_throws)
|
||||
{
|
||||
zmq::context_t context;
|
||||
zmq::poller_t<> poller;
|
||||
zmq::socket_t a {context, zmq::socket_type::router};
|
||||
zmq::socket_t b {std::move (a)};
|
||||
ASSERT_THROW (poller.add (a, ZMQ_POLLIN, nullptr),
|
||||
zmq::error_t);
|
||||
zmq::socket_t a{context, zmq::socket_type::router};
|
||||
zmq::socket_t b{std::move (a)};
|
||||
ASSERT_THROW (poller.add (a, ZMQ_POLLIN, nullptr), zmq::error_t);
|
||||
}
|
||||
|
||||
TEST(poller, remove_invalid_socket_throws)
|
||||
TEST (poller, remove_invalid_socket_throws)
|
||||
{
|
||||
zmq::context_t context;
|
||||
zmq::socket_t socket {context, zmq::socket_type::router};
|
||||
zmq::socket_t socket{context, zmq::socket_type::router};
|
||||
zmq::poller_t<> poller;
|
||||
ASSERT_NO_THROW (poller.add (socket, ZMQ_POLLIN, nullptr));
|
||||
std::vector<zmq::socket_t> sockets;
|
||||
@ -158,96 +160,97 @@ TEST(poller, remove_invalid_socket_throws)
|
||||
ASSERT_NO_THROW (poller.remove (sockets[0]));
|
||||
}
|
||||
|
||||
TEST(poller, modify_empty_throws)
|
||||
TEST (poller, modify_empty_throws)
|
||||
{
|
||||
zmq::context_t context;
|
||||
zmq::socket_t socket {context, zmq::socket_type::push};
|
||||
zmq::socket_t socket{context, zmq::socket_type::push};
|
||||
zmq::poller_t<> poller;
|
||||
ASSERT_THROW (poller.modify (socket, ZMQ_POLLIN), zmq::error_t);
|
||||
}
|
||||
|
||||
TEST(poller, modify_invalid_socket_throws)
|
||||
TEST (poller, modify_invalid_socket_throws)
|
||||
{
|
||||
zmq::context_t context;
|
||||
zmq::socket_t a {context, zmq::socket_type::push};
|
||||
zmq::socket_t b {std::move (a)};
|
||||
zmq::socket_t a{context, zmq::socket_type::push};
|
||||
zmq::socket_t b{std::move (a)};
|
||||
zmq::poller_t<> poller;
|
||||
ASSERT_THROW (poller.modify (a, ZMQ_POLLIN), zmq::error_t);
|
||||
}
|
||||
|
||||
TEST(poller, modify_not_added_throws)
|
||||
TEST (poller, modify_not_added_throws)
|
||||
{
|
||||
zmq::context_t context;
|
||||
zmq::socket_t a {context, zmq::socket_type::push};
|
||||
zmq::socket_t b {context, zmq::socket_type::push};
|
||||
zmq::socket_t a{context, zmq::socket_type::push};
|
||||
zmq::socket_t b{context, zmq::socket_type::push};
|
||||
zmq::poller_t<> poller;
|
||||
ASSERT_NO_THROW (poller.add (a, ZMQ_POLLIN, nullptr));
|
||||
ASSERT_THROW (poller.modify (b, ZMQ_POLLIN), zmq::error_t);
|
||||
}
|
||||
|
||||
TEST(poller, modify_simple)
|
||||
TEST (poller, modify_simple)
|
||||
{
|
||||
zmq::context_t context;
|
||||
zmq::socket_t a {context, zmq::socket_type::push};
|
||||
zmq::socket_t a{context, zmq::socket_type::push};
|
||||
zmq::poller_t<> poller;
|
||||
ASSERT_NO_THROW (poller.add (a, ZMQ_POLLIN, nullptr));
|
||||
ASSERT_NO_THROW (poller.modify (a, ZMQ_POLLIN|ZMQ_POLLOUT));
|
||||
ASSERT_NO_THROW (poller.modify (a, ZMQ_POLLIN | ZMQ_POLLOUT));
|
||||
}
|
||||
|
||||
TEST(poller, poll_client_server)
|
||||
TEST (poller, poll_client_server)
|
||||
{
|
||||
// Setup server and client
|
||||
common_server_client_setup s;
|
||||
|
||||
// Setup poller
|
||||
zmq::poller_t<> poller;
|
||||
ASSERT_NO_THROW(poller.add(s.server, ZMQ_POLLIN, s.server));
|
||||
ASSERT_NO_THROW (poller.add (s.server, ZMQ_POLLIN, s.server));
|
||||
|
||||
// client sends message
|
||||
ASSERT_NO_THROW(s.client.send("Hi"));
|
||||
ASSERT_NO_THROW (s.client.send ("Hi"));
|
||||
|
||||
// wait for message and verify events
|
||||
std::vector<zmq_poller_event_t> events(1);
|
||||
ASSERT_EQ(1, poller.wait_all(events, std::chrono::milliseconds{500}));
|
||||
ASSERT_EQ(ZMQ_POLLIN, events[0].events);
|
||||
std::vector<zmq_poller_event_t> events (1);
|
||||
ASSERT_EQ (1, poller.wait_all (events, std::chrono::milliseconds{500}));
|
||||
ASSERT_EQ (ZMQ_POLLIN, events[0].events);
|
||||
|
||||
// Modify server socket with pollout flag
|
||||
ASSERT_NO_THROW(poller.modify(s.server, ZMQ_POLLIN | ZMQ_POLLOUT));
|
||||
ASSERT_EQ(1, poller.wait_all(events, std::chrono::milliseconds{500}));
|
||||
ASSERT_EQ(ZMQ_POLLIN | ZMQ_POLLOUT, events[0].events);
|
||||
ASSERT_NO_THROW (poller.modify (s.server, ZMQ_POLLIN | ZMQ_POLLOUT));
|
||||
ASSERT_EQ (1, poller.wait_all (events, std::chrono::milliseconds{500}));
|
||||
ASSERT_EQ (ZMQ_POLLIN | ZMQ_POLLOUT, events[0].events);
|
||||
}
|
||||
|
||||
TEST(poller, wait_one_return)
|
||||
TEST (poller, wait_one_return)
|
||||
{
|
||||
// Setup server and client
|
||||
common_server_client_setup s;
|
||||
|
||||
// Setup poller
|
||||
zmq::poller_t<> poller;
|
||||
ASSERT_NO_THROW(poller.add(s.server, ZMQ_POLLIN, nullptr));
|
||||
ASSERT_NO_THROW (poller.add (s.server, ZMQ_POLLIN, nullptr));
|
||||
|
||||
// client sends message
|
||||
ASSERT_NO_THROW(s.client.send("Hi"));
|
||||
ASSERT_NO_THROW (s.client.send ("Hi"));
|
||||
|
||||
// wait for message and verify events
|
||||
std::vector<zmq_poller_event_t> events(1);
|
||||
ASSERT_EQ(1, poller.wait_all(events, std::chrono::milliseconds{500}));
|
||||
std::vector<zmq_poller_event_t> events (1);
|
||||
ASSERT_EQ (1, poller.wait_all (events, std::chrono::milliseconds{500}));
|
||||
}
|
||||
|
||||
TEST(poller, wait_on_move_constructed_poller)
|
||||
TEST (poller, wait_on_move_constructed_poller)
|
||||
{
|
||||
common_server_client_setup s;
|
||||
ASSERT_NO_THROW (s.client.send ("Hi"));
|
||||
zmq::poller_t<> a;
|
||||
ASSERT_NO_THROW (a.add (s.server, ZMQ_POLLIN, nullptr));
|
||||
zmq::poller_t<> b {std::move (a)};
|
||||
std::vector<zmq_poller_event_t> events(1);
|
||||
zmq::poller_t<> b{std::move (a)};
|
||||
std::vector<zmq_poller_event_t> events (1);
|
||||
/// \todo the actual error code should be checked
|
||||
ASSERT_THROW(a.wait_all (events, std::chrono::milliseconds{10}), zmq::error_t);
|
||||
ASSERT_EQ (1, b.wait_all (events, std::chrono::milliseconds {-1}));
|
||||
ASSERT_THROW (a.wait_all (events, std::chrono::milliseconds{10}),
|
||||
zmq::error_t);
|
||||
ASSERT_EQ (1, b.wait_all (events, std::chrono::milliseconds{-1}));
|
||||
}
|
||||
|
||||
TEST(poller, wait_on_move_assigned_poller)
|
||||
TEST (poller, wait_on_move_assigned_poller)
|
||||
{
|
||||
common_server_client_setup s;
|
||||
ASSERT_NO_THROW (s.client.send ("Hi"));
|
||||
@ -256,12 +259,13 @@ TEST(poller, wait_on_move_assigned_poller)
|
||||
zmq::poller_t<> b;
|
||||
b = {std::move (a)};
|
||||
/// \todo the actual error code should be checked
|
||||
std::vector<zmq_poller_event_t> events(1);
|
||||
ASSERT_THROW(a.wait_all (events, std::chrono::milliseconds{10}), zmq::error_t);
|
||||
ASSERT_EQ (1, b.wait_all (events, std::chrono::milliseconds {-1}));
|
||||
std::vector<zmq_poller_event_t> events (1);
|
||||
ASSERT_THROW (a.wait_all (events, std::chrono::milliseconds{10}),
|
||||
zmq::error_t);
|
||||
ASSERT_EQ (1, b.wait_all (events, std::chrono::milliseconds{-1}));
|
||||
}
|
||||
|
||||
TEST(poller, remove_from_handler)
|
||||
TEST (poller, remove_from_handler)
|
||||
{
|
||||
constexpr auto ITER_NO = 10;
|
||||
|
||||
@ -273,22 +277,24 @@ TEST(poller, remove_from_handler)
|
||||
// Setup poller
|
||||
zmq::poller_t<> poller;
|
||||
for (auto i = 0; i < ITER_NO; ++i) {
|
||||
ASSERT_NO_THROW(poller.add(setup_list[i].server, ZMQ_POLLIN, nullptr));
|
||||
ASSERT_NO_THROW (
|
||||
poller.add (setup_list[i].server, ZMQ_POLLIN, nullptr));
|
||||
}
|
||||
// Clients send messages
|
||||
for (auto & s : setup_list) {
|
||||
ASSERT_NO_THROW(s.client.send("Hi"));
|
||||
for (auto &s : setup_list) {
|
||||
ASSERT_NO_THROW (s.client.send ("Hi"));
|
||||
}
|
||||
|
||||
// Wait for all servers to receive a message
|
||||
for (auto & s : setup_list) {
|
||||
zmq::pollitem_t items [] = { { s.server, 0, ZMQ_POLLIN, 0 } };
|
||||
zmq::poll (&items [0], 1);
|
||||
for (auto &s : setup_list) {
|
||||
zmq::pollitem_t items[] = {{s.server, 0, ZMQ_POLLIN, 0}};
|
||||
zmq::poll (&items[0], 1);
|
||||
}
|
||||
|
||||
// Fire all handlers in one wait
|
||||
std::vector<zmq_poller_event_t> events(ITER_NO);
|
||||
ASSERT_EQ(ITER_NO, poller.wait_all (events, std::chrono::milliseconds{-1}));
|
||||
std::vector<zmq_poller_event_t> events (ITER_NO);
|
||||
ASSERT_EQ (ITER_NO,
|
||||
poller.wait_all (events, std::chrono::milliseconds{-1}));
|
||||
}
|
||||
|
||||
#endif
|
||||
|
@ -1,16 +1,16 @@
|
||||
#include <gtest/gtest.h>
|
||||
#include <zmq.hpp>
|
||||
|
||||
TEST(socket, create_destroy)
|
||||
TEST (socket, create_destroy)
|
||||
{
|
||||
zmq::context_t context;
|
||||
zmq::socket_t socket(context, ZMQ_ROUTER);
|
||||
zmq::socket_t socket (context, ZMQ_ROUTER);
|
||||
}
|
||||
|
||||
#ifdef ZMQ_CPP11
|
||||
TEST(socket, create_by_enum_destroy)
|
||||
TEST (socket, create_by_enum_destroy)
|
||||
{
|
||||
zmq::context_t context;
|
||||
zmq::socket_t socket(context, zmq::socket_type::router);
|
||||
zmq::socket_t socket (context, zmq::socket_type::router);
|
||||
}
|
||||
#endif
|
||||
|
496
zmq_addon.hpp
496
zmq_addon.hpp
@ -31,8 +31,8 @@
|
||||
#include <sstream>
|
||||
#include <stdexcept>
|
||||
|
||||
namespace zmq {
|
||||
|
||||
namespace zmq
|
||||
{
|
||||
#ifdef ZMQ_HAS_RVALUE_REFS
|
||||
|
||||
/*
|
||||
@ -44,479 +44,399 @@ namespace zmq {
|
||||
*/
|
||||
class multipart_t
|
||||
{
|
||||
private:
|
||||
private:
|
||||
std::deque<message_t> m_parts;
|
||||
|
||||
public:
|
||||
|
||||
public:
|
||||
typedef std::deque<message_t>::iterator iterator;
|
||||
typedef std::deque<message_t>::const_iterator const_iterator;
|
||||
|
||||
typedef std::deque<message_t>::reverse_iterator reverse_iterator;
|
||||
typedef std::deque<message_t>::const_reverse_iterator const_reverse_iterator;
|
||||
typedef std::deque<message_t>::const_reverse_iterator
|
||||
const_reverse_iterator;
|
||||
|
||||
// Default constructor
|
||||
multipart_t()
|
||||
{}
|
||||
multipart_t () {}
|
||||
|
||||
// Construct from socket receive
|
||||
multipart_t(socket_t& socket)
|
||||
{
|
||||
recv(socket);
|
||||
}
|
||||
multipart_t (socket_t &socket) { recv (socket); }
|
||||
|
||||
// Construct from memory block
|
||||
multipart_t(const void *src, size_t size)
|
||||
{
|
||||
addmem(src, size);
|
||||
}
|
||||
multipart_t (const void *src, size_t size) { addmem (src, size); }
|
||||
|
||||
// Construct from string
|
||||
multipart_t(const std::string& string)
|
||||
{
|
||||
addstr(string);
|
||||
}
|
||||
multipart_t (const std::string &string) { addstr (string); }
|
||||
|
||||
// Construct from message part
|
||||
multipart_t(message_t&& message)
|
||||
{
|
||||
add(std::move(message));
|
||||
}
|
||||
multipart_t (message_t &&message) { add (std::move (message)); }
|
||||
|
||||
// Move constructor
|
||||
multipart_t(multipart_t&& other)
|
||||
{
|
||||
m_parts = std::move(other.m_parts);
|
||||
}
|
||||
multipart_t (multipart_t &&other) { m_parts = std::move (other.m_parts); }
|
||||
|
||||
// Move assignment operator
|
||||
multipart_t& operator=(multipart_t&& other)
|
||||
multipart_t &operator= (multipart_t &&other)
|
||||
{
|
||||
m_parts = std::move(other.m_parts);
|
||||
m_parts = std::move (other.m_parts);
|
||||
return *this;
|
||||
}
|
||||
|
||||
// Destructor
|
||||
virtual ~multipart_t()
|
||||
{
|
||||
clear();
|
||||
}
|
||||
virtual ~multipart_t () { clear (); }
|
||||
|
||||
message_t& operator[] (size_t n)
|
||||
{
|
||||
return m_parts[n];
|
||||
}
|
||||
message_t &operator[] (size_t n) { return m_parts[n]; }
|
||||
|
||||
const message_t& operator[] (size_t n) const
|
||||
{
|
||||
return m_parts[n];
|
||||
}
|
||||
const message_t &operator[] (size_t n) const { return m_parts[n]; }
|
||||
|
||||
message_t& at (size_t n)
|
||||
{
|
||||
return m_parts.at(n);
|
||||
}
|
||||
message_t &at (size_t n) { return m_parts.at (n); }
|
||||
|
||||
const message_t& at (size_t n) const
|
||||
{
|
||||
return m_parts.at(n);
|
||||
}
|
||||
const message_t &at (size_t n) const { return m_parts.at (n); }
|
||||
|
||||
iterator begin()
|
||||
{
|
||||
return m_parts.begin();
|
||||
}
|
||||
iterator begin () { return m_parts.begin (); }
|
||||
|
||||
const_iterator begin() const
|
||||
{
|
||||
return m_parts.begin();
|
||||
}
|
||||
const_iterator begin () const { return m_parts.begin (); }
|
||||
|
||||
const_iterator cbegin() const
|
||||
{
|
||||
return m_parts.cbegin();
|
||||
}
|
||||
const_iterator cbegin () const { return m_parts.cbegin (); }
|
||||
|
||||
reverse_iterator rbegin()
|
||||
{
|
||||
return m_parts.rbegin();
|
||||
}
|
||||
reverse_iterator rbegin () { return m_parts.rbegin (); }
|
||||
|
||||
const_reverse_iterator rbegin() const
|
||||
{
|
||||
return m_parts.rbegin();
|
||||
}
|
||||
const_reverse_iterator rbegin () const { return m_parts.rbegin (); }
|
||||
|
||||
iterator end()
|
||||
{
|
||||
return m_parts.end();
|
||||
}
|
||||
iterator end () { return m_parts.end (); }
|
||||
|
||||
const_iterator end() const
|
||||
{
|
||||
return m_parts.end();
|
||||
}
|
||||
const_iterator end () const { return m_parts.end (); }
|
||||
|
||||
const_iterator cend() const
|
||||
{
|
||||
return m_parts.cend();
|
||||
}
|
||||
const_iterator cend () const { return m_parts.cend (); }
|
||||
|
||||
reverse_iterator rend()
|
||||
{
|
||||
return m_parts.rend();
|
||||
}
|
||||
reverse_iterator rend () { return m_parts.rend (); }
|
||||
|
||||
const_reverse_iterator rend() const
|
||||
{
|
||||
return m_parts.rend();
|
||||
}
|
||||
const_reverse_iterator rend () const { return m_parts.rend (); }
|
||||
|
||||
// Delete all parts
|
||||
void clear()
|
||||
{
|
||||
m_parts.clear();
|
||||
}
|
||||
void clear () { m_parts.clear (); }
|
||||
|
||||
// Get number of parts
|
||||
size_t size() const
|
||||
{
|
||||
return m_parts.size();
|
||||
}
|
||||
size_t size () const { return m_parts.size (); }
|
||||
|
||||
// Check if number of parts is zero
|
||||
bool empty() const
|
||||
{
|
||||
return m_parts.empty();
|
||||
}
|
||||
bool empty () const { return m_parts.empty (); }
|
||||
|
||||
// Receive multipart message from socket
|
||||
bool recv(socket_t& socket, int flags = 0)
|
||||
bool recv (socket_t &socket, int flags = 0)
|
||||
{
|
||||
clear();
|
||||
clear ();
|
||||
bool more = true;
|
||||
while (more)
|
||||
{
|
||||
while (more) {
|
||||
message_t message;
|
||||
if (!socket.recv(&message, flags))
|
||||
if (!socket.recv (&message, flags))
|
||||
return false;
|
||||
more = message.more();
|
||||
add(std::move(message));
|
||||
more = message.more ();
|
||||
add (std::move (message));
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
// Send multipart message to socket
|
||||
bool send(socket_t& socket, int flags = 0)
|
||||
bool send (socket_t &socket, int flags = 0)
|
||||
{
|
||||
flags &= ~(ZMQ_SNDMORE);
|
||||
bool more = size() > 0;
|
||||
while (more)
|
||||
{
|
||||
message_t message = pop();
|
||||
more = size() > 0;
|
||||
if (!socket.send(message, (more ? ZMQ_SNDMORE : 0) | flags))
|
||||
bool more = size () > 0;
|
||||
while (more) {
|
||||
message_t message = pop ();
|
||||
more = size () > 0;
|
||||
if (!socket.send (message, (more ? ZMQ_SNDMORE : 0) | flags))
|
||||
return false;
|
||||
}
|
||||
clear();
|
||||
clear ();
|
||||
return true;
|
||||
}
|
||||
|
||||
// Concatenate other multipart to front
|
||||
void prepend(multipart_t&& other)
|
||||
void prepend (multipart_t &&other)
|
||||
{
|
||||
while (!other.empty())
|
||||
push(other.remove());
|
||||
while (!other.empty ())
|
||||
push (other.remove ());
|
||||
}
|
||||
|
||||
// Concatenate other multipart to back
|
||||
void append(multipart_t&& other)
|
||||
void append (multipart_t &&other)
|
||||
{
|
||||
while (!other.empty())
|
||||
add(other.pop());
|
||||
while (!other.empty ())
|
||||
add (other.pop ());
|
||||
}
|
||||
|
||||
// Push memory block to front
|
||||
void pushmem(const void *src, size_t size)
|
||||
void pushmem (const void *src, size_t size)
|
||||
{
|
||||
m_parts.push_front(message_t(src, size));
|
||||
m_parts.push_front (message_t (src, size));
|
||||
}
|
||||
|
||||
// Push memory block to back
|
||||
void addmem(const void *src, size_t size)
|
||||
void addmem (const void *src, size_t size)
|
||||
{
|
||||
m_parts.push_back(message_t(src, size));
|
||||
m_parts.push_back (message_t (src, size));
|
||||
}
|
||||
|
||||
// Push string to front
|
||||
void pushstr(const std::string& string)
|
||||
void pushstr (const std::string &string)
|
||||
{
|
||||
m_parts.push_front(message_t(string.data(), string.size()));
|
||||
m_parts.push_front (message_t (string.data (), string.size ()));
|
||||
}
|
||||
|
||||
// Push string to back
|
||||
void addstr(const std::string& string)
|
||||
void addstr (const std::string &string)
|
||||
{
|
||||
m_parts.push_back(message_t(string.data(), string.size()));
|
||||
m_parts.push_back (message_t (string.data (), string.size ()));
|
||||
}
|
||||
|
||||
// Push type (fixed-size) to front
|
||||
template<typename T>
|
||||
void pushtyp(const T& type)
|
||||
template <typename T> void pushtyp (const T &type)
|
||||
{
|
||||
static_assert(!std::is_same<T, std::string>::value, "Use pushstr() instead of pushtyp<std::string>()");
|
||||
m_parts.push_front(message_t(&type, sizeof(type)));
|
||||
static_assert (!std::is_same<T, std::string>::value,
|
||||
"Use pushstr() instead of pushtyp<std::string>()");
|
||||
m_parts.push_front (message_t (&type, sizeof (type)));
|
||||
}
|
||||
|
||||
// Push type (fixed-size) to back
|
||||
template<typename T>
|
||||
void addtyp(const T& type)
|
||||
template <typename T> void addtyp (const T &type)
|
||||
{
|
||||
static_assert(!std::is_same<T, std::string>::value, "Use addstr() instead of addtyp<std::string>()");
|
||||
m_parts.push_back(message_t(&type, sizeof(type)));
|
||||
static_assert (!std::is_same<T, std::string>::value,
|
||||
"Use addstr() instead of addtyp<std::string>()");
|
||||
m_parts.push_back (message_t (&type, sizeof (type)));
|
||||
}
|
||||
|
||||
// Push message part to front
|
||||
void push(message_t&& message)
|
||||
void push (message_t &&message)
|
||||
{
|
||||
m_parts.push_front(std::move(message));
|
||||
m_parts.push_front (std::move (message));
|
||||
}
|
||||
|
||||
// Push message part to back
|
||||
void add(message_t&& message)
|
||||
{
|
||||
m_parts.push_back(std::move(message));
|
||||
}
|
||||
void add (message_t &&message) { m_parts.push_back (std::move (message)); }
|
||||
|
||||
// Pop string from front
|
||||
std::string popstr()
|
||||
std::string popstr ()
|
||||
{
|
||||
std::string string(m_parts.front().data<char>(), m_parts.front().size());
|
||||
m_parts.pop_front();
|
||||
std::string string (m_parts.front ().data<char> (),
|
||||
m_parts.front ().size ());
|
||||
m_parts.pop_front ();
|
||||
return string;
|
||||
}
|
||||
|
||||
// Pop type (fixed-size) from front
|
||||
template<typename T>
|
||||
T poptyp()
|
||||
template <typename T> T poptyp ()
|
||||
{
|
||||
static_assert(!std::is_same<T, std::string>::value, "Use popstr() instead of poptyp<std::string>()");
|
||||
if (sizeof(T) != m_parts.front().size())
|
||||
throw std::runtime_error("Invalid type, size does not match the message size");
|
||||
T type = *m_parts.front().data<T>();
|
||||
m_parts.pop_front();
|
||||
static_assert (!std::is_same<T, std::string>::value,
|
||||
"Use popstr() instead of poptyp<std::string>()");
|
||||
if (sizeof (T) != m_parts.front ().size ())
|
||||
throw std::runtime_error (
|
||||
"Invalid type, size does not match the message size");
|
||||
T type = *m_parts.front ().data<T> ();
|
||||
m_parts.pop_front ();
|
||||
return type;
|
||||
}
|
||||
|
||||
// Pop message part from front
|
||||
message_t pop()
|
||||
message_t pop ()
|
||||
{
|
||||
message_t message = std::move(m_parts.front());
|
||||
m_parts.pop_front();
|
||||
message_t message = std::move (m_parts.front ());
|
||||
m_parts.pop_front ();
|
||||
return message;
|
||||
}
|
||||
|
||||
// Pop message part from back
|
||||
message_t remove()
|
||||
message_t remove ()
|
||||
{
|
||||
message_t message = std::move(m_parts.back());
|
||||
m_parts.pop_back();
|
||||
message_t message = std::move (m_parts.back ());
|
||||
m_parts.pop_back ();
|
||||
return message;
|
||||
}
|
||||
|
||||
// Get pointer to a specific message part
|
||||
const message_t* peek(size_t index) const
|
||||
{
|
||||
return &m_parts[index];
|
||||
}
|
||||
const message_t *peek (size_t index) const { return &m_parts[index]; }
|
||||
|
||||
// Get a string copy of a specific message part
|
||||
std::string peekstr(size_t index) const
|
||||
std::string peekstr (size_t index) const
|
||||
{
|
||||
std::string string(m_parts[index].data<char>(), m_parts[index].size());
|
||||
std::string string (m_parts[index].data<char> (),
|
||||
m_parts[index].size ());
|
||||
return string;
|
||||
}
|
||||
|
||||
// Peek type (fixed-size) from front
|
||||
template<typename T>
|
||||
T peektyp(size_t index) const
|
||||
template <typename T> T peektyp (size_t index) const
|
||||
{
|
||||
static_assert(!std::is_same<T, std::string>::value, "Use peekstr() instead of peektyp<std::string>()");
|
||||
if(sizeof(T) != m_parts[index].size())
|
||||
throw std::runtime_error("Invalid type, size does not match the message size");
|
||||
T type = *m_parts[index].data<T>();
|
||||
static_assert (!std::is_same<T, std::string>::value,
|
||||
"Use peekstr() instead of peektyp<std::string>()");
|
||||
if (sizeof (T) != m_parts[index].size ())
|
||||
throw std::runtime_error (
|
||||
"Invalid type, size does not match the message size");
|
||||
T type = *m_parts[index].data<T> ();
|
||||
return type;
|
||||
}
|
||||
|
||||
// Create multipart from type (fixed-size)
|
||||
template<typename T>
|
||||
static multipart_t create(const T& type)
|
||||
template <typename T> static multipart_t create (const T &type)
|
||||
{
|
||||
multipart_t multipart;
|
||||
multipart.addtyp(type);
|
||||
multipart.addtyp (type);
|
||||
return multipart;
|
||||
}
|
||||
|
||||
// Copy multipart
|
||||
multipart_t clone() const
|
||||
multipart_t clone () const
|
||||
{
|
||||
multipart_t multipart;
|
||||
for (size_t i = 0; i < size(); i++)
|
||||
multipart.addmem(m_parts[i].data(), m_parts[i].size());
|
||||
for (size_t i = 0; i < size (); i++)
|
||||
multipart.addmem (m_parts[i].data (), m_parts[i].size ());
|
||||
return multipart;
|
||||
}
|
||||
|
||||
// Dump content to string
|
||||
std::string str() const
|
||||
std::string str () const
|
||||
{
|
||||
std::stringstream ss;
|
||||
for (size_t i = 0; i < m_parts.size(); i++)
|
||||
{
|
||||
const unsigned char* data = m_parts[i].data<unsigned char>();
|
||||
size_t size = m_parts[i].size();
|
||||
for (size_t i = 0; i < m_parts.size (); i++) {
|
||||
const unsigned char *data = m_parts[i].data<unsigned char> ();
|
||||
size_t size = m_parts[i].size ();
|
||||
|
||||
// Dump the message as text or binary
|
||||
bool isText = true;
|
||||
for (size_t j = 0; j < size; j++)
|
||||
{
|
||||
if (data[j] < 32 || data[j] > 127)
|
||||
{
|
||||
for (size_t j = 0; j < size; j++) {
|
||||
if (data[j] < 32 || data[j] > 127) {
|
||||
isText = false;
|
||||
break;
|
||||
}
|
||||
}
|
||||
ss << "\n[" << std::dec << std::setw(3) << std::setfill('0') << size << "] ";
|
||||
if (size >= 1000)
|
||||
{
|
||||
ss << "\n[" << std::dec << std::setw (3) << std::setfill ('0')
|
||||
<< size << "] ";
|
||||
if (size >= 1000) {
|
||||
ss << "... (to big to print)";
|
||||
continue;
|
||||
}
|
||||
for (size_t j = 0; j < size; j++)
|
||||
{
|
||||
for (size_t j = 0; j < size; j++) {
|
||||
if (isText)
|
||||
ss << static_cast<char>(data[j]);
|
||||
ss << static_cast<char> (data[j]);
|
||||
else
|
||||
ss << std::hex << std::setw(2) << std::setfill('0') << static_cast<short>(data[j]);
|
||||
ss << std::hex << std::setw (2) << std::setfill ('0')
|
||||
<< static_cast<short> (data[j]);
|
||||
}
|
||||
}
|
||||
return ss.str();
|
||||
return ss.str ();
|
||||
}
|
||||
|
||||
// Check if equal to other multipart
|
||||
bool equal(const multipart_t* other) const
|
||||
bool equal (const multipart_t *other) const
|
||||
{
|
||||
if (size() != other->size())
|
||||
if (size () != other->size ())
|
||||
return false;
|
||||
for (size_t i = 0; i < size(); i++)
|
||||
if (*peek(i) != *other->peek(i))
|
||||
for (size_t i = 0; i < size (); i++)
|
||||
if (*peek (i) != *other->peek (i))
|
||||
return false;
|
||||
return true;
|
||||
}
|
||||
|
||||
private:
|
||||
private:
|
||||
// Disable implicit copying (moving is more efficient)
|
||||
multipart_t(const multipart_t& other) ZMQ_DELETED_FUNCTION;
|
||||
void operator=(const multipart_t& other) ZMQ_DELETED_FUNCTION;
|
||||
}; // class multipart_t
|
||||
multipart_t (const multipart_t &other) ZMQ_DELETED_FUNCTION;
|
||||
void operator= (const multipart_t &other) ZMQ_DELETED_FUNCTION;
|
||||
}; // class multipart_t
|
||||
|
||||
inline std::ostream& operator<<(std::ostream& os, const multipart_t& msg)
|
||||
inline std::ostream &operator<< (std::ostream &os, const multipart_t &msg)
|
||||
{
|
||||
return os << msg.str();
|
||||
return os << msg.str ();
|
||||
}
|
||||
|
||||
#endif // ZMQ_HAS_RVALUE_REFS
|
||||
#endif // ZMQ_HAS_RVALUE_REFS
|
||||
|
||||
#if defined(ZMQ_BUILD_DRAFT_API) && defined(ZMQ_CPP11) && defined(ZMQ_HAVE_POLLER)
|
||||
class active_poller_t
|
||||
#if defined(ZMQ_BUILD_DRAFT_API) && defined(ZMQ_CPP11) \
|
||||
&& defined(ZMQ_HAVE_POLLER)
|
||||
class active_poller_t
|
||||
{
|
||||
public:
|
||||
active_poller_t () = default;
|
||||
~active_poller_t () = default;
|
||||
|
||||
active_poller_t (const active_poller_t &) = delete;
|
||||
active_poller_t &operator= (const active_poller_t &) = delete;
|
||||
|
||||
active_poller_t (active_poller_t &&src) = default;
|
||||
active_poller_t &operator= (active_poller_t &&src) = default;
|
||||
|
||||
using handler_t = std::function<void(short)>;
|
||||
|
||||
void add (zmq::socket_t &socket, short events, handler_t handler)
|
||||
{
|
||||
public:
|
||||
active_poller_t () = default;
|
||||
~active_poller_t () = default;
|
||||
|
||||
active_poller_t(const active_poller_t&) = delete;
|
||||
active_poller_t &operator=(const active_poller_t&) = delete;
|
||||
|
||||
active_poller_t(active_poller_t&& src) = default;
|
||||
active_poller_t &operator=(active_poller_t&& src) = default;
|
||||
|
||||
using handler_t = std::function<void(short)>;
|
||||
|
||||
void add (zmq::socket_t &socket, short events, handler_t handler)
|
||||
{
|
||||
auto it = decltype (handlers)::iterator {};
|
||||
auto inserted = bool {};
|
||||
std::tie(it, inserted) = handlers.emplace (static_cast<void*>(socket), std::make_shared<handler_t> (std::move (handler)));
|
||||
try
|
||||
{
|
||||
base_poller.add (socket, events, inserted && *(it->second) ? it->second.get() : nullptr);
|
||||
need_rebuild |= inserted;
|
||||
auto it = decltype (handlers)::iterator{};
|
||||
auto inserted = bool{};
|
||||
std::tie (it, inserted) =
|
||||
handlers.emplace (static_cast<void *> (socket),
|
||||
std::make_shared<handler_t> (std::move (handler)));
|
||||
try {
|
||||
base_poller.add (socket, events,
|
||||
inserted && *(it->second) ? it->second.get ()
|
||||
: nullptr);
|
||||
need_rebuild |= inserted;
|
||||
}
|
||||
catch (const zmq::error_t &) {
|
||||
// rollback
|
||||
if (inserted) {
|
||||
handlers.erase (static_cast<void *> (socket));
|
||||
}
|
||||
catch (const zmq::error_t&)
|
||||
{
|
||||
// rollback
|
||||
if (inserted)
|
||||
{
|
||||
handlers.erase (static_cast<void*>(socket));
|
||||
}
|
||||
throw;
|
||||
throw;
|
||||
}
|
||||
}
|
||||
|
||||
void remove (zmq::socket_t &socket)
|
||||
{
|
||||
base_poller.remove (socket);
|
||||
handlers.erase (static_cast<void *> (socket));
|
||||
need_rebuild = true;
|
||||
}
|
||||
|
||||
void modify (zmq::socket_t &socket, short events)
|
||||
{
|
||||
base_poller.modify (socket, events);
|
||||
}
|
||||
|
||||
int wait (std::chrono::milliseconds timeout)
|
||||
{
|
||||
if (need_rebuild) {
|
||||
poller_events.resize (handlers.size ());
|
||||
poller_handlers.clear ();
|
||||
poller_handlers.reserve (handlers.size ());
|
||||
for (const auto &handler : handlers) {
|
||||
poller_handlers.push_back (handler.second);
|
||||
}
|
||||
need_rebuild = false;
|
||||
}
|
||||
|
||||
void remove (zmq::socket_t &socket)
|
||||
{
|
||||
base_poller.remove (socket);
|
||||
handlers.erase (static_cast<void*>(socket));
|
||||
need_rebuild = true;
|
||||
const int count = base_poller.wait_all (poller_events, timeout);
|
||||
if (count != 0) {
|
||||
std::for_each (poller_events.begin (),
|
||||
poller_events.begin () + count,
|
||||
[](zmq_poller_event_t &event) {
|
||||
if (event.user_data != NULL)
|
||||
(*reinterpret_cast<handler_t *> (
|
||||
event.user_data)) (event.events);
|
||||
});
|
||||
}
|
||||
return count;
|
||||
}
|
||||
|
||||
void modify (zmq::socket_t &socket, short events)
|
||||
{
|
||||
base_poller.modify (socket, events);
|
||||
}
|
||||
bool empty () const { return handlers.empty (); }
|
||||
|
||||
int wait (std::chrono::milliseconds timeout)
|
||||
{
|
||||
if (need_rebuild) {
|
||||
poller_events.resize (handlers.size ());
|
||||
poller_handlers.clear ();
|
||||
poller_handlers.reserve (handlers.size ());
|
||||
for (const auto &handler : handlers) {
|
||||
poller_handlers.push_back (handler.second);
|
||||
}
|
||||
need_rebuild = false;
|
||||
}
|
||||
const int count = base_poller.wait_all (poller_events, timeout);
|
||||
if (count != 0) {
|
||||
std::for_each (poller_events.begin (), poller_events.begin () + count,
|
||||
[](zmq_poller_event_t& event) {
|
||||
if (event.user_data != NULL)
|
||||
(*reinterpret_cast<handler_t*> (event.user_data)) (event.events);
|
||||
});
|
||||
}
|
||||
return count;
|
||||
}
|
||||
size_t size () const { return handlers.size (); }
|
||||
|
||||
bool empty () const
|
||||
{
|
||||
return handlers.empty ();
|
||||
}
|
||||
private:
|
||||
bool need_rebuild{false};
|
||||
|
||||
size_t size () const
|
||||
{
|
||||
return handlers.size ();
|
||||
}
|
||||
|
||||
private:
|
||||
bool need_rebuild {false};
|
||||
|
||||
poller_t<handler_t> base_poller {};
|
||||
std::unordered_map<void*, std::shared_ptr<handler_t>> handlers {};
|
||||
std::vector<zmq_poller_event_t> poller_events {};
|
||||
std::vector<std::shared_ptr<handler_t>> poller_handlers {};
|
||||
}; // class active_poller_t
|
||||
poller_t<handler_t> base_poller{};
|
||||
std::unordered_map<void *, std::shared_ptr<handler_t>> handlers{};
|
||||
std::vector<zmq_poller_event_t> poller_events{};
|
||||
std::vector<std::shared_ptr<handler_t>> poller_handlers{};
|
||||
}; // class active_poller_t
|
||||
#endif // defined(ZMQ_BUILD_DRAFT_API) && defined(ZMQ_CPP11) && defined(ZMQ_HAVE_POLLER)
|
||||
|
||||
|
||||
} // namespace zmq
|
||||
} // namespace zmq
|
||||
|
||||
#endif // __ZMQ_ADDON_HPP_INCLUDED__
|
||||
#endif // __ZMQ_ADDON_HPP_INCLUDED__
|
||||
|
Loading…
x
Reference in New Issue
Block a user