python_redis.py

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

"""
Python操作Redis实现消息的发布与订阅
"""

import sys
import time
import redis

# 全局变量
conn_pool = redis.ConnectionPool(host="localhost", port=6379, db=1)
conn_inst = redis.Redis(connection_pool=conn_pool)
channel_name = "fm-101.1"


def public_test():
    while True:
        # 发布消息
        conn_inst.publish(channel_name, "hello " + str(time.time()))
        if int(time.time()) % 10 == 1:
            conn_inst.publish(channel_name, "over")
        time.sleep(1)


def subscribe_test(_type=0):
    pub = conn_inst.pubsub()
    pub.subscribe(channel_name)

    if _type == 0:
        # 订阅消息
        for item in pub.listen():
            print("Listen on channel: %s" % item)
            if item["type"] == "message" and item["data"].decode() == "over":
                print(item["channel"].decode(), "已停止发布")
                break
    else:
        # 另一种订阅模式
        while True:
            item = pub.parse_response()
            print("Listen on channel: %s" % item)
            if item[0].decode() == "message" and item[2].decode() == "over":
                print(item[1].decode(), "已停止发布")
                break

    # 取消订阅
    pub.unsubscribe()
    return


if __name__ == '__main__':
    if sys.argv[1] == "public":
        public_test()
    else:
        subscribe_test(int(sys.argv[2]))