diff options
author | Dan Klein <danklei@cisco.com> | 2015-08-24 09:28:43 +0300 |
---|---|---|
committer | Dan Klein <danklei@cisco.com> | 2015-08-24 09:28:43 +0300 |
commit | 20eb7d362f9bce1951bd61ad3f78cf8f4267d1d5 (patch) | |
tree | b2c3fe1c721d5f6898b4bb8e81f7c68502e313e9 /src/console/zmq/green/device.py | |
parent | 49f6b00b58c3ec734218fcd69259771a42c157bd (diff) | |
parent | 651a7d779551e193bd9dbadbe8b2a02bdab231b4 (diff) |
Merge branch 'master' into dan_stateless
Diffstat (limited to 'src/console/zmq/green/device.py')
-rwxr-xr-x | src/console/zmq/green/device.py | 32 |
1 files changed, 32 insertions, 0 deletions
diff --git a/src/console/zmq/green/device.py b/src/console/zmq/green/device.py new file mode 100755 index 00000000..4b070237 --- /dev/null +++ b/src/console/zmq/green/device.py @@ -0,0 +1,32 @@ +# Copyright (C) PyZMQ Developers +# Distributed under the terms of the Modified BSD License. + +import zmq +from zmq.green import Poller + +def device(device_type, isocket, osocket): + """Start a zeromq device (gevent-compatible). + + Unlike the true zmq.device, this does not release the GIL. + + Parameters + ---------- + device_type : (QUEUE, FORWARDER, STREAMER) + The type of device to start (ignored). + isocket : Socket + The Socket instance for the incoming traffic. + osocket : Socket + The Socket instance for the outbound traffic. + """ + p = Poller() + if osocket == -1: + osocket = isocket + p.register(isocket, zmq.POLLIN) + p.register(osocket, zmq.POLLIN) + + while True: + events = dict(p.poll()) + if isocket in events: + osocket.send_multipart(isocket.recv_multipart()) + if osocket in events: + isocket.send_multipart(osocket.recv_multipart()) |