generic-library/cppzmq
1
0
Fork 0
mirror of https://github.com/zeromq/cppzmq.git synced 2025-04-21 08:19:00 +02:00
Code Issues Projects Releases Wiki Activity

Merge pull request #82 from staticfloat/master

Browse Source 
Enable passing flags in to `send()` and `recv()`
This commit is contained in:
Constantin Rack 2016-10-22 09:27:47 +02:00 committed by GitHub
commit 0791496622
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
zmq_addon.hpp
View File

@ -99,14 +99,14 @@ public:
return m_parts.empty(); return m_parts.empty();
} }
bool recv(socket_t& socket) bool recv(socket_t& socket, int flags = 0)
{ {
clear(); clear();
bool more = true; bool more = true;
while (more) while (more)
{ {
message_t message; message_t message;
if (!socket.recv(&message)) if (!socket.recv(&message, flags))
return false; return false;
more = message.more(); more = message.more();
add(std::move(message)); add(std::move(message));
@ -114,14 +114,15 @@ public:
return true; return true;
} }
bool send(socket_t& socket) bool send(socket_t& socket, int flags = 0)
{ {
flags &= ~(ZMQ_SNDMORE);
bool more = size() > 0; bool more = size() > 0;
while (more) while (more)
{ {
message_t message = pop(); message_t message = pop();
more = size() > 0; more = size() > 0;
if (!socket.send(message, more ? ZMQ_SNDMORE : 0)) if (!socket.send(message, (more ? ZMQ_SNDMORE : 0) | flags))
return false; return false;
} }
clear(); clear();