Round-trip demonstrator in Python

"""Round-trip demonstrator

While this example runs in a single process, that is just to make
it easier to start and stop the example. Client thread signals to
main when it's ready.
"""

import sys
import threading
import time

import zmq

from zhelpers import zpipe

def client_task (ctx, pipe):
client = ctx.socket(zmq.DEALER)
client.identity = 'C'
client.connect("tcp://localhost:5555")

print "Setting up test…\n",
time.sleep(0.1)

print "Synchronous round-trip test…\n",
start = time.time()
requests = 10000
for r in xrange(requests):
client.send("hello")
client.recv()
print " %d calls/second\n" % (requests / (time.time()-start)),

print "Asynchronous round-trip test…\n",
start = time.time()
for r in xrange(requests):
client.send("hello")
for r in xrange(requests):
client.recv()
print " %d calls/second\n" % (requests / (time.time()-start)),

# signal done:
pipe.send("done")

def worker_task():
ctx = zmq.Context()
worker = ctx.socket(zmq.DEALER)
worker.identity = 'W'
worker.connect("tcp://localhost:5556")

while True:
msg = worker.recv_multipart()
worker.send_multipart(msg)
ctx.destroy(0)

def broker_task():
# Prepare our context and sockets
ctx = zmq.Context()
frontend = ctx.socket(zmq.ROUTER)
backend = ctx.socket(zmq.ROUTER)
frontend.bind("tcp://*:5555")
backend.bind("tcp://*:5556")

# Initialize poll set
poller = zmq.Poller()
poller.register(backend, zmq.POLLIN)
poller.register(frontend, zmq.POLLIN)

while True:
try:
items = dict(poller.poll())
except:
break # Interrupted

if frontend in items:
msg = frontend.recv_multipart()
msg[0] = 'W'
backend.send_multipart(msg)
if backend in items:
msg = backend.recv_multipart()
msg[0] = 'C'
frontend.send_multipart(msg)

def main():
# Create threads
ctx = zmq.Context()
client,pipe = zpipe(ctx)

client_thread = threading.Thread(target=client_task, args=(ctx, pipe))
worker_thread = threading.Thread(target=worker_task)
worker_thread.daemon=True
broker_thread = threading.Thread(target=broker_task)
broker_thread.daemon=True

worker_thread.start()
broker_thread.start()
client_thread.start()

# Wait for signal on client pipe
client.recv()

if __name__ == '__main__':
main()