本文整理汇总了Python中zmq.eventloop.zmqstream.ZMQStream.on_send方法的典型用法代码示例。如果您正苦于以下问题:Python ZMQStream.on_send方法的具体用法?Python ZMQStream.on_send怎么用?Python ZMQStream.on_send使用的例子?那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类zmq.eventloop.zmqstream.ZMQStream
的用法示例。
在下文中一共展示了ZMQStream.on_send方法的3个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: PubQueue
# 需要导入模块: from zmq.eventloop.zmqstream import ZMQStream [as 别名]
# 或者: from zmq.eventloop.zmqstream.ZMQStream import on_send [as 别名]
class PubQueue(BaseQueue):
def __init__(self, bind, ioloop=None):
super().__init__(zmq.PUSH)
self.socket.bind(bind)
self.stream = ZMQStream(self.socket, ioloop)
self.stream.on_send(self.__on_send)
def __on_send(self, msq, status):
print(msq, status)
def send(self, data):
self.stream.send(data)
self.stream.flush()
def send_string(self, data):
self.stream.send_string(data)
self.stream.flush()
示例2: Channel
# 需要导入模块: from zmq.eventloop.zmqstream import ZMQStream [as 别名]
# 或者: from zmq.eventloop.zmqstream.ZMQStream import on_send [as 别名]
class Channel(object):
"""Mother of all channels. Defines the interface.
Callbacks:
The callbacks will receive the channel as first parameter and
the message as second parameter. The error callback will get
the stream where the error occured as second parameter.
Attributes:
* stream_in, stream_out : the streams for eventlopp handling
* serializer : the serializer used
"""
def __init__(self, socket_in, socket_out, serializer):
self.stream_in = ZMQStream(socket_in)
self.stream_out = ZMQStream(socket_out)
self.serializer = serializer
self._cb_receive = None
self._cb_send = None
self._cb_error = None
self._chan_id = id(self)
return
def on_receive(self, callback):
"""Set callback to invoke when a message was received.
"""
self.stream_in.stop_on_recv()
self._cb_receive = callback
if callback:
self.stream_in.on_recv(self._on_recv)
return
def on_send(self, callback):
"""Set callback to invoke when a message was sent.
"""
self.stream_out.stop_on_send()
self._cb_send = callback
if callback:
self.stream_out.on_send(self._on_send)
return
def on_error(self, callback):
"""Set callback to invoke when an error event occured.
"""
self.stream_in.stop_on_err()
self.stream_out.stop_on_err()
self._cb_error = callback
if callback:
self.stream_in.on_err(self._on_err_in)
self.stream_out.on_err(self._on_err_out)
return
def send(self, message):
"""Send given message.
"""
m = self.serializer.serialize(message)
if self.serializer.multipart:
self.stream_out.send_multipart(m)
else:
self.stream_out.send(m)
return
def _on_recv(self, msg):
"""Helper interfacing w/ streams.
"""
if self.serializer.multipart:
msg = self.serializer.deserialize(msg)
else:
msg = self.serializer.deserialize(msg[0])
self._cb_receive(self, msg)
return
def _on_send(self, sent, _):
"""Helper interfacing w/ streams.
"""
msg = sent[0]
if self.serializer.multipart:
msg = self.serializer.deserialize(msg)
else:
msg = self.serializer.deserialize(msg[0])
self._cb_send(self, msg)
return
def _on_err_in(self):
self._cb_error(self, self.stream_in)
return
def _on_err_out(self):
self._cb_error(self, self.stream_out)
return
def __hash__(self):
return self._chan_id
def __eq__(self, other):
if not isinstance(other, self.__class__):
return False
return self._chan_id == other._chan_id
#.........这里部分代码省略.........
示例3: __init__
# 需要导入模块: from zmq.eventloop.zmqstream import ZMQStream [as 别名]
# 或者: from zmq.eventloop.zmqstream.ZMQStream import on_send [as 别名]
class Stream:
'''
This is the main class that interacts with the zmq library to send, and
receive messages.
'''
def __init__(self, socket, stream_info, path, on_recv=None, on_send=None, loop=None):
'''
Initializes instance of Stream.
@param socket - zmq socket that has alredy been bound
@param stream_info - this streams definition from the yaml config
@param path - path to the socket on the disk
@parma on_recv - callback that processes received messages
@parma on_send - callback that processes sent messages
@param loop - loop this socket will belong to. Default is global async loop.
'''
self._path = path
self._recv_type = stream_info.recv_type
self._send_type = stream_info.send_type
self._reply_type = stream_info.reply_type
self._on_recv = on_recv
self._on_send = on_send
self._stream = ZMQStream(socket, io_loop=loop)
if self._on_recv is not None:
self._stream.on_recv(self._recv_wrapper)
if self._on_send is not None:
self._stream.on_send(self._send_wrapper)
@property
def recv_type(self):
return self._recv_type
@property
def send_type(self):
return self._send_type
@property
def reply_type(self):
return self._reply_type
def on_send(self, callback):
'''
Set the callback to be invoked on every send command. on_send(None)
disables this callback.
@param callback - Callback must take exactly two arguments, which will
be the message being sent (always a list), and the
return result of socket.send_multipart(msg) -
MessageTracker or None.
'''
self._on_send = callback
if callback is None:
self._stream.on_send(None)
else:
self._stream.on_send(self._send_wrapper)
def on_recv(self, callback):
'''
Register a callback for when a message is ready to recv. There can be
only one callback registered at a time, so each call to on_recv
replaces previously registered callbacks. on_recv(None) disables recv
event polling.
@param callback - callback must take exactly one argument, which will
be a list, as returned by socket.recv_multipart()
if callback is None, recv callbacks are disabled.
'''
self._on_recv = callback
if callback is None:
self._stream.on_recv(None)
else:
self._stream.on_recv(self._recv_wrapper)
def flush(self, flag=3, limit=None):
'''
Flush pending messages.
This method safely handles all pending incoming and/or outgoing
messages, bypassing the inner loop, passing them to the registered
callbacks.
A limit can be specified, to prevent blocking under high load.
flush will return the first time ANY of these conditions are met:
No more events matching the flag are pending.
the total number of events handled reaches the limit.
@param flag - 0MQ poll flags. If flag|POLLIN, recv events will be
flushed. If flag|POLLOUT, send events will be flushed.
Both flags can be set at once, which is the default.
@param limit - None, or int. Optional. The maximum number of messages
to send or receive. Both send and receive count against
this limit
@returns int - count of events handled
'''
return self._stream.flush(flag, limit)
def send(self, msg, **kwds):
#.........这里部分代码省略.........