Beispiel #1
0
    def test_poller_unregister(self):
        poller = Poller()
        socket1, socket2 = self.create_bound_pair(zmqpy.PAIR, zmqpy.PAIR)
        poller.register(socket1)

        #register with no flags unregister the socket
        poller.unregister(socket1)

        assert poller.sockets == {}
Beispiel #2
0
    def test_poller_modify(self):
        poller = Poller()
        socket1, socket2 = self.create_bound_pair(zmqpy.PAIR, zmqpy.PAIR)
        poller.register(socket1)

        #register with no flags unregister the socket
        poller.modify(socket1, flags=zmqpy.POLLOUT)

        assert poller.sockets[socket1] == zmqpy.POLLOUT
Beispiel #3
0
class ZMQPoller(object):
    """A poller that can be used in the tornado IOLoop.

    This simply wraps a regular zmq.Poller, scaling the timeout
    by 1000, so that it is in seconds rather than milliseconds.
    """

    def __init__(self):
        self._poller = Poller()

    @staticmethod
    def _map_events(events):
        """translate IOLoop.READ/WRITE/ERROR event masks into zmq.POLLIN/OUT/ERR"""
        z_events = 0
        if events & IOLoop.READ:
            z_events |= POLLIN
        if events & IOLoop.WRITE:
            z_events |= POLLOUT
        if events & IOLoop.ERROR:
            z_events |= POLLERR
        return z_events

    @staticmethod
    def _remap_events(z_events):
        """translate zmq.POLLIN/OUT/ERR event masks into IOLoop.READ/WRITE/ERROR"""
        events = 0
        if z_events & POLLIN:
            events |= IOLoop.READ
        if z_events & POLLOUT:
            events |= IOLoop.WRITE
        if z_events & POLLERR:
            events |= IOLoop.ERROR
        return events

    def register(self, fd, events):
        return self._poller.register(fd, self._map_events(events))

    def modify(self, fd, events):
        return self._poller.modify(fd, self._map_events(events))

    def unregister(self, fd):
        return self._poller.unregister(fd)

    def poll(self, timeout):
        """poll in seconds rather than milliseconds.

        Event masks will be IOLoop.READ/WRITE/ERROR
        """
        z_events = self._poller.poll(1000*timeout)
        return [ (fd,self._remap_events(evt)) for (fd,evt) in z_events ]

    def close(self):
        pass
Beispiel #4
0
 def __init__(self):
     self._poller = Poller()
Beispiel #5
0
    def test_poller_register(self):
        poller = Poller()
        socket1, socket2 = self.create_bound_pair(zmqpy.PAIR, zmqpy.PAIR)
        poller.register(socket1)

        assert poller.sockets[socket1] == zmqpy.POLLIN | zmqpy.POLLOUT