python - ZeroMQ、Redis 和 Gevent

Question

我正在开发一个必须处理大量小数据请求的服务器。
服务器是用ZeroMQ开发的，使用PULL/PUSH模式。
基本上每个请求过程都包括将其数据(消息)存储在 Redis 中。

我希望通过使用 gevent 处理每个请求来获得更好的性能。
以下代码不起作用(消息不存储在 Redis 中)。

from gevent import monkey
monkey.patch_all()
from redis import Redis, StrictRedis
from redis import connection
import zmq
import gevent


context = zmq.Context()
socket = context.socket(zmq.PULL)
socket.bind("tcp://*:5000")

connection.socket = gevent.socket
redis = Redis()

def enqueue(message):
    redis.lpush('work_queue', message)


while True:
    message = socket.recv()
    #print message
    gevent.spawn(enqueue, message)

如果我删除 gevent 的内容，那么代码就可以工作(消息可以正确存储)。

附注我刚刚开始使用 Zeromq 和 gevent。

更新:我想实现一个简单的任务队列。最终使用 celery 。服务器(使用 zmq)异步启动 celery 任务(并且效果很好)。

Answer 1

要将 Zeromq 与 gevent 结合使用，您必须导入 de zmq.green 模块。

将 zmq.green 导入为 zmq

http://zeromq.github.io/pyzmq/api/zmq.green.html