mirror of
https://github.com/zeromq/libzmq.git
synced 2025-03-03 04:38:42 +01:00
New style ROUTER socket removed.
Signed-off-by: Martin Sustrik <sustrik@turist.(none)>
This commit is contained in:
parent
541b83bc02
commit
dee8b2360a
@ -162,7 +162,6 @@ ZMQ_EXPORT int zmq_term (void *context);
|
|||||||
#define ZMQ_PUSH 8
|
#define ZMQ_PUSH 8
|
||||||
#define ZMQ_XPUB 9
|
#define ZMQ_XPUB 9
|
||||||
#define ZMQ_XSUB 10
|
#define ZMQ_XSUB 10
|
||||||
#define ZMQ_ROUTER 13
|
|
||||||
|
|
||||||
/* Socket options. */
|
/* Socket options. */
|
||||||
#define ZMQ_AFFINITY 4
|
#define ZMQ_AFFINITY 4
|
||||||
|
@ -55,7 +55,6 @@ libzmq_la_SOURCES = \
|
|||||||
reaper.hpp \
|
reaper.hpp \
|
||||||
rep.hpp \
|
rep.hpp \
|
||||||
req.hpp \
|
req.hpp \
|
||||||
router.hpp \
|
|
||||||
select.hpp \
|
select.hpp \
|
||||||
session_base.hpp \
|
session_base.hpp \
|
||||||
signaler.hpp \
|
signaler.hpp \
|
||||||
@ -113,7 +112,6 @@ libzmq_la_SOURCES = \
|
|||||||
reaper.cpp \
|
reaper.cpp \
|
||||||
pub.cpp \
|
pub.cpp \
|
||||||
random.cpp \
|
random.cpp \
|
||||||
router.cpp \
|
|
||||||
rep.cpp \
|
rep.cpp \
|
||||||
req.cpp \
|
req.cpp \
|
||||||
select.cpp \
|
select.cpp \
|
||||||
|
285
src/router.cpp
285
src/router.cpp
@ -1,285 +0,0 @@
|
|||||||
/*
|
|
||||||
Copyright (c) 2007-2011 iMatix Corporation
|
|
||||||
Copyright (c) 2007-2011 Other contributors as noted in the AUTHORS file
|
|
||||||
|
|
||||||
This file is part of 0MQ.
|
|
||||||
|
|
||||||
0MQ is free software; you can redistribute it and/or modify it under
|
|
||||||
the terms of the GNU Lesser General Public License as published by
|
|
||||||
the Free Software Foundation; either version 3 of the License, or
|
|
||||||
(at your option) any later version.
|
|
||||||
|
|
||||||
0MQ is distributed in the hope that it will be useful,
|
|
||||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
||||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
||||||
GNU Lesser General Public License for more details.
|
|
||||||
|
|
||||||
You should have received a copy of the GNU Lesser General Public License
|
|
||||||
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
||||||
*/
|
|
||||||
|
|
||||||
#include "router.hpp"
|
|
||||||
#include "pipe.hpp"
|
|
||||||
#include "wire.hpp"
|
|
||||||
#include "random.hpp"
|
|
||||||
#include "likely.hpp"
|
|
||||||
#include "wire.hpp"
|
|
||||||
#include "err.hpp"
|
|
||||||
|
|
||||||
zmq::router_t::router_t (class ctx_t *parent_, uint32_t tid_) :
|
|
||||||
socket_base_t (parent_, tid_),
|
|
||||||
prefetched (false),
|
|
||||||
more_in (false),
|
|
||||||
current_out (NULL),
|
|
||||||
more_out (false),
|
|
||||||
next_peer_id (generate_random ())
|
|
||||||
{
|
|
||||||
options.type = ZMQ_ROUTER;
|
|
||||||
|
|
||||||
prefetched_msg.init ();
|
|
||||||
}
|
|
||||||
|
|
||||||
zmq::router_t::~router_t ()
|
|
||||||
{
|
|
||||||
zmq_assert (outpipes.empty ());
|
|
||||||
prefetched_msg.close ();
|
|
||||||
}
|
|
||||||
|
|
||||||
void zmq::router_t::xattach_pipe (pipe_t *pipe_)
|
|
||||||
{
|
|
||||||
zmq_assert (pipe_);
|
|
||||||
|
|
||||||
// Generate a new peer ID. Take care to avoid duplicates.
|
|
||||||
outpipes_t::iterator it = outpipes.lower_bound (next_peer_id);
|
|
||||||
if (!outpipes.empty ()) {
|
|
||||||
while (true) {
|
|
||||||
if (it == outpipes.end ())
|
|
||||||
it = outpipes.begin ();
|
|
||||||
if (it->first != next_peer_id)
|
|
||||||
break;
|
|
||||||
++next_peer_id;
|
|
||||||
++it;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Add the pipe to the map out outbound pipes.
|
|
||||||
outpipe_t outpipe = {pipe_, true};
|
|
||||||
bool ok = outpipes.insert (outpipes_t::value_type (
|
|
||||||
next_peer_id, outpipe)).second;
|
|
||||||
zmq_assert (ok);
|
|
||||||
|
|
||||||
// Add the pipe to the list of inbound pipes.
|
|
||||||
pipe_->set_pipe_id (next_peer_id);
|
|
||||||
fq.attach (pipe_);
|
|
||||||
|
|
||||||
// Queue the connection command.
|
|
||||||
pending_command_t cmd = {1, next_peer_id};
|
|
||||||
pending_commands.push_back (cmd);
|
|
||||||
|
|
||||||
// Advance next peer ID so that if new connection is dropped shortly after
|
|
||||||
// its creation we don't accidentally get two subsequent peers with
|
|
||||||
// the same ID.
|
|
||||||
++next_peer_id;
|
|
||||||
}
|
|
||||||
|
|
||||||
void zmq::router_t::xterminated (pipe_t *pipe_)
|
|
||||||
{
|
|
||||||
fq.terminated (pipe_);
|
|
||||||
|
|
||||||
for (outpipes_t::iterator it = outpipes.begin ();
|
|
||||||
it != outpipes.end (); ++it) {
|
|
||||||
if (it->second.pipe == pipe_) {
|
|
||||||
|
|
||||||
// Queue the disconnection command.
|
|
||||||
pending_command_t cmd = {2, it->first};
|
|
||||||
pending_commands.push_back (cmd);
|
|
||||||
|
|
||||||
// Remove the pipe.
|
|
||||||
outpipes.erase (it);
|
|
||||||
if (pipe_ == current_out)
|
|
||||||
current_out = NULL;
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
zmq_assert (false);
|
|
||||||
}
|
|
||||||
|
|
||||||
void zmq::router_t::xread_activated (pipe_t *pipe_)
|
|
||||||
{
|
|
||||||
fq.activated (pipe_);
|
|
||||||
}
|
|
||||||
|
|
||||||
void zmq::router_t::xwrite_activated (pipe_t *pipe_)
|
|
||||||
{
|
|
||||||
for (outpipes_t::iterator it = outpipes.begin ();
|
|
||||||
it != outpipes.end (); ++it) {
|
|
||||||
if (it->second.pipe == pipe_) {
|
|
||||||
zmq_assert (!it->second.active);
|
|
||||||
it->second.active = true;
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
zmq_assert (false);
|
|
||||||
}
|
|
||||||
|
|
||||||
int zmq::router_t::xsend (msg_t *msg_, int flags_)
|
|
||||||
{
|
|
||||||
// If this is the first part of the message it's the ID of the
|
|
||||||
// peer to send the message to.
|
|
||||||
if (!more_out) {
|
|
||||||
zmq_assert (!current_out);
|
|
||||||
|
|
||||||
// The first message part has to be label.
|
|
||||||
if (unlikely (!(msg_->flags () & msg_t::label))) {
|
|
||||||
errno = EFSM;
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Find the pipe associated with the peer ID stored in the message.
|
|
||||||
if (unlikely (msg_->size () != 4)) {
|
|
||||||
errno = ECANTROUTE;
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
uint32_t peer_id = get_uint32 ((unsigned char*) msg_->data ());
|
|
||||||
outpipes_t::iterator it = outpipes.find (peer_id);
|
|
||||||
if (unlikely (it == outpipes.end ())) {
|
|
||||||
errno = ECANTROUTE;
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Check whether the pipe is available for writing.
|
|
||||||
msg_t empty;
|
|
||||||
int rc = empty.init ();
|
|
||||||
errno_assert (rc == 0);
|
|
||||||
if (!it->second.pipe->check_write (&empty)) {
|
|
||||||
rc = empty.close ();
|
|
||||||
errno_assert (rc == 0);
|
|
||||||
it->second.active = false;
|
|
||||||
errno = EAGAIN;
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
rc = empty.close ();
|
|
||||||
errno_assert (rc == 0);
|
|
||||||
|
|
||||||
// Mark the pipe to send the message to.
|
|
||||||
current_out = it->second.pipe;
|
|
||||||
more_out = true;
|
|
||||||
|
|
||||||
// Clean up the message object.
|
|
||||||
rc = msg_->close ();
|
|
||||||
errno_assert (rc == 0);
|
|
||||||
rc = msg_->init ();
|
|
||||||
errno_assert (rc == 0);
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Check whether this is the last part of the message.
|
|
||||||
more_out = msg_->flags () & (msg_t::more | msg_t::label) ? true : false;
|
|
||||||
|
|
||||||
// Push the message into the pipe. If there's no out pipe, just drop it.
|
|
||||||
if (current_out) {
|
|
||||||
bool ok = current_out->write (msg_);
|
|
||||||
if (unlikely (!ok))
|
|
||||||
current_out = NULL;
|
|
||||||
else if (!more_out) {
|
|
||||||
current_out->flush ();
|
|
||||||
current_out = NULL;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
else {
|
|
||||||
int rc = msg_->close ();
|
|
||||||
errno_assert (rc == 0);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Detach the message from the data buffer.
|
|
||||||
int rc = msg_->init ();
|
|
||||||
errno_assert (rc == 0);
|
|
||||||
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
int zmq::router_t::xrecv (msg_t *msg_, int flags_)
|
|
||||||
{
|
|
||||||
// If there's a queued command, pass it to the caller.
|
|
||||||
if (unlikely (!more_in && !pending_commands.empty ())) {
|
|
||||||
msg_->init_size (5);
|
|
||||||
unsigned char *data = (unsigned char*) msg_->data ();
|
|
||||||
put_uint8 (data, pending_commands.front ().cmd);
|
|
||||||
put_uint32 (data + 1, pending_commands.front ().peer);
|
|
||||||
msg_->set_flags (msg_t::command);
|
|
||||||
pending_commands.pop_front ();
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
// If there is a prefetched message, return it.
|
|
||||||
if (prefetched) {
|
|
||||||
int rc = msg_->move (prefetched_msg);
|
|
||||||
errno_assert (rc == 0);
|
|
||||||
more_in = msg_->flags () & (msg_t::more | msg_t::label) ? true : false;
|
|
||||||
prefetched = false;
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Get next message part.
|
|
||||||
pipe_t *pipe;
|
|
||||||
int rc = fq.recvpipe (msg_, flags_, &pipe);
|
|
||||||
if (rc != 0)
|
|
||||||
return -1;
|
|
||||||
|
|
||||||
// If we are in the middle of reading a message, just return the next part.
|
|
||||||
if (more_in) {
|
|
||||||
more_in = msg_->flags () & (msg_t::more | msg_t::label) ? true : false;
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
// We are at the beginning of a new message. Move the message part we
|
|
||||||
// have to the prefetched and return the ID of the peer instead.
|
|
||||||
rc = prefetched_msg.move (*msg_);
|
|
||||||
errno_assert (rc == 0);
|
|
||||||
prefetched = true;
|
|
||||||
rc = msg_->close ();
|
|
||||||
errno_assert (rc == 0);
|
|
||||||
rc = msg_->init_size (4);
|
|
||||||
errno_assert (rc == 0);
|
|
||||||
put_uint32 ((unsigned char*) msg_->data (), pipe->get_pipe_id ());
|
|
||||||
msg_->set_flags (msg_t::label);
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
int zmq::router_t::rollback (void)
|
|
||||||
{
|
|
||||||
if (current_out) {
|
|
||||||
current_out->rollback ();
|
|
||||||
current_out = NULL;
|
|
||||||
more_out = false;
|
|
||||||
}
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool zmq::router_t::xhas_in ()
|
|
||||||
{
|
|
||||||
if (prefetched)
|
|
||||||
return true;
|
|
||||||
return fq.has_in () || !pending_commands.empty();
|
|
||||||
}
|
|
||||||
|
|
||||||
bool zmq::router_t::xhas_out ()
|
|
||||||
{
|
|
||||||
// In theory, GENERIC socket is always ready for writing. Whether actual
|
|
||||||
// attempt to write succeeds depends on whitch pipe the message is going
|
|
||||||
// to be routed to.
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
zmq::router_session_t::router_session_t (io_thread_t *io_thread_, bool connect_,
|
|
||||||
socket_base_t *socket_, const options_t &options_,
|
|
||||||
const char *protocol_, const char *address_) :
|
|
||||||
session_base_t (io_thread_, connect_, socket_, options_, protocol_,
|
|
||||||
address_)
|
|
||||||
{
|
|
||||||
}
|
|
||||||
|
|
||||||
zmq::router_session_t::~router_session_t ()
|
|
||||||
{
|
|
||||||
}
|
|
||||||
|
|
123
src/router.hpp
123
src/router.hpp
@ -1,123 +0,0 @@
|
|||||||
/*
|
|
||||||
Copyright (c) 2007-2011 iMatix Corporation
|
|
||||||
Copyright (c) 2007-2011 Other contributors as noted in the AUTHORS file
|
|
||||||
|
|
||||||
This file is part of 0MQ.
|
|
||||||
|
|
||||||
0MQ is free software; you can redistribute it and/or modify it under
|
|
||||||
the terms of the GNU Lesser General Public License as published by
|
|
||||||
the Free Software Foundation; either version 3 of the License, or
|
|
||||||
(at your option) any later version.
|
|
||||||
|
|
||||||
0MQ is distributed in the hope that it will be useful,
|
|
||||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
||||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
||||||
GNU Lesser General Public License for more details.
|
|
||||||
|
|
||||||
You should have received a copy of the GNU Lesser General Public License
|
|
||||||
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
||||||
*/
|
|
||||||
|
|
||||||
#ifndef __ZMQ_ROUTER_HPP_INCLUDED__
|
|
||||||
#define __ZMQ_ROUTER_HPP_INCLUDED__
|
|
||||||
|
|
||||||
#include <map>
|
|
||||||
#include <deque>
|
|
||||||
|
|
||||||
#include "socket_base.hpp"
|
|
||||||
#include "session_base.hpp"
|
|
||||||
#include "stdint.hpp"
|
|
||||||
#include "msg.hpp"
|
|
||||||
#include "fq.hpp"
|
|
||||||
|
|
||||||
namespace zmq
|
|
||||||
{
|
|
||||||
|
|
||||||
class router_t :
|
|
||||||
public socket_base_t
|
|
||||||
{
|
|
||||||
public:
|
|
||||||
|
|
||||||
router_t (class ctx_t *parent_, uint32_t tid_);
|
|
||||||
~router_t ();
|
|
||||||
|
|
||||||
// Overloads of functions from socket_base_t.
|
|
||||||
void xattach_pipe (class pipe_t *pipe_);
|
|
||||||
int xsend (class msg_t *msg_, int flags_);
|
|
||||||
int xrecv (class msg_t *msg_, int flags_);
|
|
||||||
bool xhas_in ();
|
|
||||||
bool xhas_out ();
|
|
||||||
void xread_activated (class pipe_t *pipe_);
|
|
||||||
void xwrite_activated (class pipe_t *pipe_);
|
|
||||||
void xterminated (class pipe_t *pipe_);
|
|
||||||
|
|
||||||
protected:
|
|
||||||
|
|
||||||
// Rollback any message parts that were sent but not yet flushed.
|
|
||||||
int rollback ();
|
|
||||||
|
|
||||||
private:
|
|
||||||
|
|
||||||
// Fair queueing object for inbound pipes.
|
|
||||||
fq_t fq;
|
|
||||||
|
|
||||||
// Have we prefetched a message.
|
|
||||||
bool prefetched;
|
|
||||||
|
|
||||||
// Holds the prefetched message.
|
|
||||||
msg_t prefetched_msg;
|
|
||||||
|
|
||||||
// If true, more incoming message parts are expected.
|
|
||||||
bool more_in;
|
|
||||||
|
|
||||||
struct outpipe_t
|
|
||||||
{
|
|
||||||
class pipe_t *pipe;
|
|
||||||
bool active;
|
|
||||||
};
|
|
||||||
|
|
||||||
// Outbound pipes indexed by the peer IDs.
|
|
||||||
typedef std::map <uint32_t, outpipe_t> outpipes_t;
|
|
||||||
outpipes_t outpipes;
|
|
||||||
|
|
||||||
// The pipe we are currently writing to.
|
|
||||||
class pipe_t *current_out;
|
|
||||||
|
|
||||||
// If true, more outgoing message parts are expected.
|
|
||||||
bool more_out;
|
|
||||||
|
|
||||||
// Peer ID are generated. It's a simple increment and wrap-over
|
|
||||||
// algorithm. This value is the next ID to use (if not used already).
|
|
||||||
uint32_t next_peer_id;
|
|
||||||
|
|
||||||
// Commands to be delivered to the user.
|
|
||||||
struct pending_command_t
|
|
||||||
{
|
|
||||||
uint8_t cmd;
|
|
||||||
uint32_t peer;
|
|
||||||
};
|
|
||||||
typedef std::deque <pending_command_t> pending_commands_t;
|
|
||||||
pending_commands_t pending_commands;
|
|
||||||
|
|
||||||
router_t (const router_t&);
|
|
||||||
const router_t &operator = (const router_t&);
|
|
||||||
};
|
|
||||||
|
|
||||||
class router_session_t : public session_base_t
|
|
||||||
{
|
|
||||||
public:
|
|
||||||
|
|
||||||
router_session_t (class io_thread_t *io_thread_, bool connect_,
|
|
||||||
class socket_base_t *socket_, const options_t &options_,
|
|
||||||
const char *protocol_, const char *address_);
|
|
||||||
~router_session_t ();
|
|
||||||
|
|
||||||
private:
|
|
||||||
|
|
||||||
router_session_t (const router_session_t&);
|
|
||||||
const router_session_t &operator = (const router_session_t&);
|
|
||||||
};
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
#endif
|
|
@ -40,7 +40,6 @@
|
|||||||
#include "xsub.hpp"
|
#include "xsub.hpp"
|
||||||
#include "push.hpp"
|
#include "push.hpp"
|
||||||
#include "pull.hpp"
|
#include "pull.hpp"
|
||||||
#include "router.hpp"
|
|
||||||
#include "pair.hpp"
|
#include "pair.hpp"
|
||||||
|
|
||||||
zmq::session_base_t *zmq::session_base_t::create (class io_thread_t *io_thread_,
|
zmq::session_base_t *zmq::session_base_t::create (class io_thread_t *io_thread_,
|
||||||
@ -88,10 +87,6 @@ zmq::session_base_t *zmq::session_base_t::create (class io_thread_t *io_thread_,
|
|||||||
s = new (std::nothrow) pull_session_t (io_thread_, connect_,
|
s = new (std::nothrow) pull_session_t (io_thread_, connect_,
|
||||||
socket_, options_, protocol_, address_);
|
socket_, options_, protocol_, address_);
|
||||||
break;
|
break;
|
||||||
case ZMQ_ROUTER:
|
|
||||||
s = new (std::nothrow) router_session_t (io_thread_, connect_,
|
|
||||||
socket_, options_, protocol_, address_);
|
|
||||||
break;
|
|
||||||
case ZMQ_PAIR:
|
case ZMQ_PAIR:
|
||||||
s = new (std::nothrow) pair_session_t (io_thread_, connect_,
|
s = new (std::nothrow) pair_session_t (io_thread_, connect_,
|
||||||
socket_, options_, protocol_, address_);
|
socket_, options_, protocol_, address_);
|
||||||
|
@ -60,7 +60,6 @@
|
|||||||
#include "xrep.hpp"
|
#include "xrep.hpp"
|
||||||
#include "xpub.hpp"
|
#include "xpub.hpp"
|
||||||
#include "xsub.hpp"
|
#include "xsub.hpp"
|
||||||
#include "router.hpp"
|
|
||||||
|
|
||||||
bool zmq::socket_base_t::check_tag ()
|
bool zmq::socket_base_t::check_tag ()
|
||||||
{
|
{
|
||||||
@ -106,9 +105,6 @@ zmq::socket_base_t *zmq::socket_base_t::create (int type_, class ctx_t *parent_,
|
|||||||
case ZMQ_XSUB:
|
case ZMQ_XSUB:
|
||||||
s = new (std::nothrow) xsub_t (parent_, tid_);
|
s = new (std::nothrow) xsub_t (parent_, tid_);
|
||||||
break;
|
break;
|
||||||
case ZMQ_ROUTER:
|
|
||||||
s = new (std::nothrow) router_t (parent_, tid_);
|
|
||||||
break;
|
|
||||||
default:
|
default:
|
||||||
errno = EINVAL;
|
errno = EINVAL;
|
||||||
return NULL;
|
return NULL;
|
||||||
|
Loading…
x
Reference in New Issue
Block a user