Source code for uzmq.poll

# -*- coding: utf-8 -
#
# This file is part of uzmq. See the NOTICE for more information.

"""
ZMQPoll: ZMQ Poll handle

"""
import pyuv
import zmq

from . import util

[docs]class ZMQPoll(object): """\ :param loop: loop object where this handle runs (accessible through :py:attr:`Poll.loop`). :param int socket: zmq socket to be monitored for readibility or writability. ``ZMQPoll`` ZMQPoll handles can be used to monitor any ZMQ sockets for readability or writability. .. py:attribute:: loop *Read only* :py:class:`pyuv.Loop` object where this handle runs. """ def __init__(self, loop, socket): self.loop = loop self.socket = socket # initialize private variable self.fd = socket.getsockopt(zmq.FD) self._poller = pyuv.Poll(loop, self.fd) self._callback = None self._started = False self._events = 0 @property
[docs] def active(self): """*Read only* Indicates if this handle is active.""" return self._poller.active
@property
[docs] def closed(self): """*Read only* Indicates if this handle is closing or already closed.""" return self._poller.closed
[docs] def start(self, events, callback): """\ :param events: int Mask of events that will be detected. The possible events are `pyuv.UV_READABLE` or `pyuv.UV_WRITABLE`. :param callback: callable Function that will be called when the ``Poll`` handle receives events. Callback signature: ``callback(poll_handle, events, errorno)``. Start or update the event mask of the ``ZMQPoll`` handle. """ if not util.is_callable(callback): raise TypeError("a callable is required") self._callback = callback self._events = events evs = pyuv.UV_READABLE | pyuv.UV_WRITABLE self._poller.start(evs, self._poll)
[docs] def stop(self): """ Stop the ``Poll`` handle. """ self._poller.stop()
[docs] def close(self, callback=None): """ :param callable callback: Function that will be called after the ``ZMQPoll`` handle is closed. Close the ``ZMQPoll`` handle. After a handle has been closed no other operations can be performed on it. """ self._poller.close() if util.is_callable(callback): callback(self)
def _poll(self, handle, evs, errno): # trick to use the last state. Fix a race condition z_events = self.socket.getsockopt(zmq.EVENTS) if not z_events: return events = 0 if z_events & zmq.POLLIN: events |= pyuv.UV_READABLE if z_events & zmq.POLLOUT: events |= pyuv.UV_WRITABLE if events & self._events: self._callback(self, events, errno)