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

Enable passing flags in to send() and recv()

Browse Source
This commit is contained in:
Elliot Saba 2016-08-14 16:46:43 -07:00
parent 92d2af6def
commit 93af23a58e
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();