python实现Redis订阅发布

Redis 发布订阅

Redis 发布订阅可以用在像消息通知，群聊，定向推送，参数刷新加载等业务场景

发布订阅模型有三个角色：

发布者（Publisher）
订阅者(Subscriber)
频道(channel)

每个订阅者可以订阅多个频道，发布者可以在某个频道里发布消息，订阅者会接受到自己订阅频道里发布的消息。

1.相关命令 (参考)

publish channel message         发布消息
subscribe [channel]             订阅频道
unsubscribe [channel]           取消订阅
psubscribe [pattern...]         订阅指定模式的频道
punsubscribe [pattern...]       退订指定模式的频道
pubsub channels                 列出至少有一个订阅者的频道
pubsub numsub [channel...]      列表给定频道的订阅者数量
pubsub numpat                   列表被订阅模式的数量

在终端使用示例

# 在 终端1 订阅cctv1
127.0.0.1:8100> subscribe cctv1
Reading messages... (press Ctrl-C to quit)
1) "subscribe"
2) "cctv1"
3) (integer) 1

# 在 终端2 向cctv1 发布消息
127.0.0.1:8100> publish cctv1 "this is cctv1"
(integer) 1

# 终端1 接受到终端2发的消息
127.0.0.1:8100> subscribe cctv1
Reading messages... (press Ctrl-C to quit)
1) "subscribe"
2) "cctv1"
3) (integer) 1
1) "message"
2) "cctv1"
3) "this is cctv1"

2. python 实现

订阅者类，只写了订阅功能，退订原理是一样的，偷了懒，把订阅频道和接收消息写到了一个方法里面，如果严格按面向对象的思想应该分开。

from PublishAndSubscribe.Channel import Channel
from PublishAndSubscribe.RedisTool import RedisToolclass Subscriber:def __init__(self, conn):self._conn = conndef subscribe(self, channel: Channel):# 获取发布/订阅对象pub = self._conn.pubsub()# 选择要订阅的频道pub.subscribe(channel.name)while True:# 接收消息msg = pub.parse_response()print(msg)if __name__ == '__main__':client = RedisTool.redis_connection("0.0.0.0", 8100, "password")cctv1 = Channel("CCTV1")Subscriber(client).subscribe(cctv1)

发布者类

from PublishAndSubscribe.Channel import Channel
from PublishAndSubscribe.RedisTool import RedisToolclass Publisher:def __init__(self, conn):self._conn = conndef publish(self, channel: Channel, mess: str):# 向特定频道发布消息self._conn.publish(channel.name, mess)if __name__ == '__main__':cctv1 = Channel("CCTV1")client = RedisTool.redis_connection("0.0.0.0", 8100, "password")publisher = Publisher(client)while True:publisher.publish(cctv1, input("请输入要发送的消息："))

频道类，Redis中频道的概念只是一个字符串

class Channel:def __init__(self, name: str):self.name = name

一个工具类，用来连接redis

import redisclass RedisTool:@staticmethoddef redis_connection(address: str, port: int, password: str):"""用来连接RedisArgs:address: Redis 服务端IP地址port: [int] Redis 服务端口password: Redis client 登录凭证Return:type[Redis]: 返回一个redis对象"""return redis.StrictRedis(address, port, password=password)