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; 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, // Add matching-1 references to the message. We already hold one reference,
// that's why -1. // that's why -1.
msg_->add_refs ((int) matching - 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; 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_) void zmq::msg_t::add_refs (int refs_)
{ {
zmq_assert (refs_ >= 0); zmq_assert (refs_ >= 0);
@@ -279,3 +284,4 @@ bool zmq::msg_t::rm_refs (int refs_)
return true; return true;
} }

View File

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