Patch for issue LIBZMQ-275. Dealing with VSM in distribution when pipes fail to write.

Signed-off-by: Ben Gray <ben@benjamg.com>
This commit is contained in:
Ben Gray 2011-10-27 13:59:54 +02:00 committed by Martin Sustrik
parent 68ab5f87ed
commit 9e000c8f26
3 changed files with 17 additions and 0 deletions

View File

@ -137,6 +137,16 @@ void zmq::dist_t::distribute (msg_t *msg_, int flags_)
return;
}
if (msg_->is_vsm ()) {
for (pipes_t::size_type i = 0; i < matching; ++i)
write (pipes [i], msg_);
int rc = msg_->close();
errno_assert (rc == 0);
rc = msg_->init ();
errno_assert (rc == 0);
return;
}
// Add matching-1 references to the message. We already hold one reference,
// that's why -1.
msg_->add_refs ((int) matching - 1);

View File

@ -237,6 +237,11 @@ bool zmq::msg_t::is_delimiter ()
return u.base.type == type_delimiter;
}
bool zmq::msg_t::is_vsm ()
{
return u.base.type == type_vsm;
}
void zmq::msg_t::add_refs (int refs_)
{
zmq_assert (refs_ >= 0);
@ -279,3 +284,4 @@ bool zmq::msg_t::rm_refs (int refs_)
return true;
}

View File

@ -68,6 +68,7 @@ namespace zmq
void set_flags (unsigned char flags_);
void reset_flags (unsigned char flags_);
bool is_delimiter ();
bool is_vsm ();
// After calling this function you can copy the message in POD-style
// refs_ times. No need to call copy.