鳞片Gevent Socketio

时间:2013-03-27 11:43:05

标签: django redis gevent gevent-socketio

我目前使用Django进行网站设置。我添加了Gevent Socketio来添加聊天功能。我需要扩展它,因为网站上已有很多用户,但无法找到这样做的方法。

我尝试了https://github.com/abourget/gevent-socketio/tree/master/examples/django_chat/chat

我正在使用Gunicorn& socketio.sgunicorn.GeventSocketIOWorker工人阶级,所以一开始我想到增加工人数量。不幸的是,这似乎间歇性地失败了。我已经开始重写它以使用我发现的几个源中的redis,并且每个服务器上有1个工作器,现在正在进行负载平衡。然而,这似乎有同样的问题。我想知道gevent socketio代码本身是否存在一些问题,不允许它扩展。

以下是我的启动方式,它只是提交消息代码。

def redis_client():
    """Get a redis client."""
    return Redis(settings.REDIS_HOST, settings.REDIS_PORT, settings.REDIS_DB)

class PubSub(object):
    """
    Very simple Pub/Sub pattern wrapper
    using simplified Redis Pub/Sub functionality.

    Usage (publisher)::

        import redis

        r = redis.Redis()

        q = PubSub(r, "channel")
        q.publish("test data")


    Usage (listener)::

        import redis

        r = redis.Redis()
        q = PubSub(r, "channel")

        def handler(data):
            print "Data received: %r" % data

        q.subscribe(handler)

    """
    def __init__(self, redis, channel="default"):
        self.redis = redis
        self.channel = channel

    def publish(self, data):
        self.redis.publish(self.channel, simplejson.dumps(data))

    def subscribe(self, handler):
        redis = self.redis.pubsub()
        redis.subscribe(self.channel)

        for data_raw in redis.listen():
            if data_raw['type'] != "message":
                continue

            data = simplejson.loads(data_raw["data"])
            handler(data)


from socketio.namespace import BaseNamespace
from socketio.sdjango import namespace
from supremo.utils import redis_client, PubSub
from gevent import Greenlet

@namespace('/chat')
class ChatNamespace(BaseNamespace):
    nicknames = []
    r = redis_client()
    q = PubSub(r, "channel")

    def initialize(self):
        # Setup redis listener
        def handler(data):
            self.emit('receive_message',data)

        greenlet = Greenlet.spawn(self.q.subscribe, handler)

    def on_submit_message(self,msg):
        self.q.publish(msg)

1 个答案:

答案 0 :(得分:2)

我使用了来自https://github.com/fcurella/django-push-demo和gevent-socketio 0.3.5rc1而不是rc2的部分代码,它现在正在使用多个工作人员和负载平衡。