Redis实现发布订阅功能实战篇（redis监听发布订阅）

Redis实现发布/订阅功能：实战篇

Redis是一种高性能的键值存储数据库，除了可以作为缓存数据库外，它还提供了非常有用的发布/订阅功能。这个功能可以用于实现消息队列、即时消息通知等实时消息处理场景。本文将带大家看一下如何使用Redis的发布/订阅功能。

1. 发布/订阅的基本概念

发布/订阅是一个消息通信模式，包括两个主要的角色：发布者和订阅者。发布者负责发送消息，订阅者则负责接收并处理消息。在Redis中，要实现发布/订阅，首先要建立一个消息通道，这个通道就是主题，可以简单理解为一个消息分类。通过将消息发布到不同的主题中，订阅者可以订阅自己感兴趣的主题，以便接收与自己相关的消息。

2. Redis的发布/订阅实现

在Redis中，发布/订阅是通过PUBLISH和SUBSCRIBE命令来完成的。我们通过以下代码来演示一个简单的发布/订阅示例：

import redis
# 建立Redis实例
r = redis.Redis(host='localhost', port=6379, db=0)
# 订阅者
def subscriber(channel):
    pubsub = r.pubsub()
    pubsub.subscribe(channel)
    # 开始监听消息
    for msg in pubsub.listen():
        print(msg)

# 发布者
def publisher(channel, message):
    r.publish(channel, message)

# 启动订阅者线程
subscribe_thread = threading.Thread(target=subscriber, args=('test_channel',))
subscribe_thread.start()

# 发送消息
publisher('test_channel', 'Hello world!')

上面的代码中，我们创建了一个Redis实例，然后定义了一个订阅者函数subscriber()和一个发布者函数publisher()。我们使用threading模块启动了一个新的线程来执行订阅者函数，并在主线程中调用发布者函数来发送消息。在subscriber()函数中，我们使用pubsub对象来订阅名为“test_channel”的主题，然后监听该主题的所有消息，并使用print()函数打印出来。在publisher()函数中，我们使用r.publish()方法向“test_channel”主题发送一个名为“Hello world！”的消息。运行该代码后，可以看到输出如下：

{‘type’: ‘subscribe’, ‘channel’: b’test_channel’, ‘pattern’: None, ‘data’: 1}

{‘type’: ‘message’, ‘pattern’: None, ‘channel’: b’test_channel’, ‘data’: b’Hello world!’}

{‘type’: ‘unsubscribe’, ‘channel’: b’test_channel’, ‘pattern’: None, ‘data’: 0}

第一条消息表示订阅成功，第二条消息表示收到了一个消息，第三条消息表示取消订阅。

3. Redis的发布/订阅高级用法

Redis的发布/订阅除了基本的PUBLISH和SUBSCRIBE命令外，还提供了一些高级用法。

3.1. 模式订阅

Redis支持使用通配符来订阅多个主题，这称为模式订阅。我们使用PSUBSCRIBE命令来进行模式订阅。例如，订阅名字以“news:”开头的所有主题可以使用如下代码：

pubsub = r.pubsub()
pubsub.psubscribe('news:*')

3.2. 跨服务器通信

Redis的发布/订阅功能不仅限于单个Redis实例，还可以在多个Redis服务器之间共享消息。订阅者可以连接到不同的Redis实例，并订阅相同的主题。这样，当一个Redis服务器接收到消息时，它会同时将消息推送到所有订阅该主题的Redis服务器。

为了实现跨服务器通信，我们需要在每个Redis实例上开启发布/订阅功能。例如，假设我们有两个Redis实例，分别位于主机A和主机B上。我们可以在每个实例上开启发布/订阅功能，并将它们连接到对方的实例：

# Redis实例A连接到Redis实例B
r_a = redis.Redis(host='host_b', port=6379, db=0)
r_a.publish('test_channel', 'Hello from A')
# Redis实例B连接到Redis实例A
r_b = redis.Redis(host='host_a', port=6379, db=0)
r_b.publish('test_channel', 'Hello from B')

在这个例子中，我们使用host_a和host_b来分别代表两台主机的IP地址。当我们在实例A上执行“r_a.publish()”方法时，消息将被发送到test_channel这个主题中；而当我们在实例B上执行“r_b.publish()”方法时，消息也将被发送到同一个主题中。订阅test_channel主题的所有订阅者都将收到这个消息。