eventlet + zmq + multiprocessing poc (similar to gevent_poc)

eventlet_poc.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

"""Simple eventlet POC to check how to get ZMQ sockets working with
subprocesses spawned by a simple process."""

import os
import eventlet
import multiprocessing

from eventlet.green import zmq

def log(msg):
    print "(%s) %s" % (os.getpid(), msg)

def subprocess(ip, port):
    c = zmq._Context()
    log('Cid: %s' % id(c))
    socket = c.socket(zmq.REP)
    socket.connect("tcp://%s:%s" % (ip, port))
    log("Connected to %s:%s" % (ip, port))
    msg = socket.recv()
    log("Received msg: %s" % msg)
    socket.send("Reply %s" % (os.getpid()))
    eventlet.sleep(0)

c = zmq.Context()
log("Cid: %s" % id(c))
socket = c.socket(zmq.REQ)
port = socket.bind_to_random_port('tcp://127.0.0.1')

# start subprocess
proc = multiprocessing.Process(target=subprocess, args=('127.0.0.1', port))
proc2 = multiprocessing.Process(target=subprocess, args=('127.0.0.1', port))
proc.start()
proc2.start()

log("Sending msg")
socket.send("Hello, world")
reply = socket.recv()
log("Reply received: %s" % reply)
# try it again;  second worker should still be waiting on recv?
assert any([proc.is_alive(), proc2.is_alive()])
socket.send("Hello, world")
reply = socket.recv()
log("Reply received: %s" % reply)