From ed291b02516ac5c9fe01f328d505305d36fe6319 Mon Sep 17 00:00:00 2001 From: Martin Sustrik Date: Sat, 27 Mar 2010 09:24:38 +0100 Subject: multi-part messages work with PUB/SUB --- src/pipe.cpp | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) (limited to 'src/pipe.cpp') diff --git a/src/pipe.cpp b/src/pipe.cpp index 14a6ef4..e844865 100644 --- a/src/pipe.cpp +++ b/src/pipe.cpp @@ -77,7 +77,9 @@ bool zmq::reader_t::read (zmq_msg_t *msg_) return false; } - msgs_read++; + if (!(msg_->flags & ZMQ_MSG_TBC)) + msgs_read++; + if (lwm > 0 && msgs_read % lwm == 0) send_reader_info (peer, msgs_read); @@ -161,7 +163,8 @@ bool zmq::writer_t::write (zmq_msg_t *msg_) } pipe->write (*msg_); - msgs_written++; + if (!(msg_->flags & ZMQ_MSG_TBC)) + msgs_written++; return true; } @@ -194,6 +197,9 @@ void zmq::writer_t::term () { endpoint = NULL; + // Rollback any unfinished messages. + rollback (); + // Push delimiter into the pipe. // Trick the compiler to belive that the tag is a valid pointer. zmq_msg_t msg; -- cgit v1.2.3