pipe_t::rollback removes only unfinished message from the pipe rather than all unflushed messages

1 files changed, 4 insertions, 0 deletions

diff --git a/src/pipe.cpp b/src/pipe.cpp
index 53dfb21..14a6ef4 100644
--- a/src/pipe.cpp
+++ b/src/pipe.cpp
@@ -170,6 +170,10 @@ void zmq::writer_t::rollback ()
     zmq_msg_t msg;
 
     while (pipe->unwrite (&msg)) {
+        if (!(msg.flags & ZMQ_MSG_TBC)) {
+            pipe->write (msg);
+            break;
+        }
         zmq_msg_close (&msg);
         msgs_written--;
     }