redis-pub_sub

发表于 | 阅读次数:
字数统计: 188 | 阅读时长 ≈ 1 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44

 # -*- coding: utf-8 -*-

"""
发布和订阅 publish and subscribe
订阅者负责订阅频道， 发送者负责向频道发送二进制字符串消息
"""

import threading
import time
import redis


conn = redis.Redis()


def publisher(n):
    time.sleep(1)
    for i in xrange(n):
        conn.publish("channel", i)
        time.sleep(1)


def run_pubsub():
    """
    启动发送者线程，并让它发送三条消息
    """
    threading.Thread(target=publisher, args=(3,)).start()
    pubsub = conn.pubsub()
    pubsub.subscribe(["channel"])
    count = 0
    for item in pubsub.listen():
        print item
        count += 1
        # 在接收到一条订阅反馈消息和三条 发布者发送的消息之后，执行退订操作 停止监听消息
        if count == 4:
            pubsub.unsubscribe()
        # 收到退订反馈消息之后不再接收消息
        if count == 5:
            break


if __name__ == "__main__":
    run_pubsub()