ZeroMQ Worker example

worker.py

import zmq
from  multiprocessing import Process

def worker(wrk_num):
    print "Getting ready to work", wrk_num
    context = zmq.Context()

    work_receiver = context.socket(zmq.PULL) # Channel to receive work
    work_receiver.connect("tcp://127.0.0.1:5557")

    results_sender = context.socket(zmq.PUSH) # Channel to send result of work
    results_sender.connect("tcp://127.0.0.1:5558")

    poller = zmq.Poller() # Poller to multiplex the work receiver and control receiver channels
    poller.register(work_receiver, zmq.POLLIN)

    while True: # Loop and accept messages from both channels, acting accordingly
        socks = dict(poller.poll())
        
        if socks.get(work_receiver) == zmq.POLLIN:
            work_message = work_receiver.recv_json()
            product = work_message['num'] * work_message['num']
            answer_message = { 'worker' : wrk_num, 'result' : product }
            print "Completed work", answer_message
            results_sender.send_json(answer_message)

if __name__ == "__main__":
    worker_pool = range(10)
    for wrk_num in range(len(worker_pool)):
        Process(target=worker, args=(wrk_num,)).start()