Skip to content

Commit 93af23a

Browse files
committed
Enable passing flags in to send() and recv()
1 parent 92d2af6 commit 93af23a

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

zmq_addon.hpp

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -99,29 +99,30 @@ class multipart_t
9999
return m_parts.empty();
100100
}
101101

102-
bool recv(socket_t& socket)
102+
bool recv(socket_t& socket, int flags = 0)
103103
{
104104
clear();
105105
bool more = true;
106106
while (more)
107107
{
108108
message_t message;
109-
if (!socket.recv(&message))
109+
if (!socket.recv(&message, flags))
110110
return false;
111111
more = message.more();
112112
add(std::move(message));
113113
}
114114
return true;
115115
}
116116

117-
bool send(socket_t& socket)
117+
bool send(socket_t& socket, int flags = 0)
118118
{
119+
flags &= ~(ZMQ_SNDMORE);
119120
bool more = size() > 0;
120121
while (more)
121122
{
122123
message_t message = pop();
123124
more = size() > 0;
124-
if (!socket.send(message, more ? ZMQ_SNDMORE : 0))
125+
if (!socket.send(message, (more ? ZMQ_SNDMORE : 0) | flags))
125126
return false;
126127
}
127128
clear();

0 commit comments

Comments
 (0)