redis-pubsub.py

import random
import signal

import gevent
import redis


REDIS_URL = 'redis://127.0.0.1:6379'
REDIS_CHANNEL = 'pubsub_demo'

redis = redis.from_url(REDIS_URL)


def publisher(pid):
    gevent.sleep(random.randint(0, 5) * 0.001)
    redis.publish(REDIS_CHANNEL, 'message from publisher %d' % pid)


def subscriber():
    pubsub = redis.pubsub()
    pubsub.subscribe(REDIS_CHANNEL)

    for message in pubsub.listen():
        gevent.sleep(0.1)
        if message['type'] == 'message':
            print message['data']


def main():
    tasks = []
    tasks.append(gevent.spawn(subscriber))
    tasks.extend([gevent.spawn(publisher, i) for i in xrange(20)])
    gevent.joinall(tasks)


if __name__ == '__main__':
    main()