89 lines
2.2 KiB
Python
89 lines
2.2 KiB
Python
|
"""Blocking channels
|
||
|
|
|
||
|
|
Useful for test suites and blocking terminal interfaces.
|
||
|
|
"""
|
||
|
|
|
||
|
|
# Copyright (c) Jupyter Development Team.
|
||
|
|
# Distributed under the terms of the Modified BSD License.
|
||
|
|
|
||
|
|
from queue import Queue, Empty
|
||
|
|
|
||
|
|
|
||
|
|
class ZMQSocketChannel(object):
|
||
|
|
"""A ZMQ socket in a simple blocking API"""
|
||
|
|
session = None
|
||
|
|
socket = None
|
||
|
|
stream = None
|
||
|
|
_exiting = False
|
||
|
|
proxy_methods = []
|
||
|
|
|
||
|
|
def __init__(self, socket, session, loop=None):
|
||
|
|
"""Create a channel.
|
||
|
|
|
||
|
|
Parameters
|
||
|
|
----------
|
||
|
|
socket : :class:`zmq.Socket`
|
||
|
|
The ZMQ socket to use.
|
||
|
|
session : :class:`session.Session`
|
||
|
|
The session to use.
|
||
|
|
loop
|
||
|
|
Unused here, for other implementations
|
||
|
|
"""
|
||
|
|
super().__init__()
|
||
|
|
|
||
|
|
self.socket = socket
|
||
|
|
self.session = session
|
||
|
|
|
||
|
|
def _recv(self, **kwargs):
|
||
|
|
msg = self.socket.recv_multipart(**kwargs)
|
||
|
|
ident,smsg = self.session.feed_identities(msg)
|
||
|
|
return self.session.deserialize(smsg)
|
||
|
|
|
||
|
|
def get_msg(self, block=True, timeout=None):
|
||
|
|
""" Gets a message if there is one that is ready. """
|
||
|
|
if block:
|
||
|
|
if timeout is not None:
|
||
|
|
timeout *= 1000 # seconds to ms
|
||
|
|
ready = self.socket.poll(timeout)
|
||
|
|
else:
|
||
|
|
ready = self.socket.poll(timeout=0)
|
||
|
|
|
||
|
|
if ready:
|
||
|
|
return self._recv()
|
||
|
|
else:
|
||
|
|
raise Empty
|
||
|
|
|
||
|
|
def get_msgs(self):
|
||
|
|
""" Get all messages that are currently ready. """
|
||
|
|
msgs = []
|
||
|
|
while True:
|
||
|
|
try:
|
||
|
|
msgs.append(self.get_msg(block=False))
|
||
|
|
except Empty:
|
||
|
|
break
|
||
|
|
return msgs
|
||
|
|
|
||
|
|
def msg_ready(self):
|
||
|
|
""" Is there a message that has been received? """
|
||
|
|
return bool(self.socket.poll(timeout=0))
|
||
|
|
|
||
|
|
def close(self):
|
||
|
|
if self.socket is not None:
|
||
|
|
try:
|
||
|
|
self.socket.close(linger=0)
|
||
|
|
except Exception:
|
||
|
|
pass
|
||
|
|
self.socket = None
|
||
|
|
stop = close
|
||
|
|
|
||
|
|
def is_alive(self):
|
||
|
|
return (self.socket is not None)
|
||
|
|
|
||
|
|
def send(self, msg):
|
||
|
|
"""Pass a message to the ZMQ socket to send
|
||
|
|
"""
|
||
|
|
self.session.send(self.socket, msg)
|
||
|
|
|
||
|
|
def start(self):
|
||
|
|
pass
|