Python Redis 订阅|极客教程

Python Redis 订阅

Redis 是一个开源的内存数据库系统，它支持多种数据结构，如字符串、列表、集合、哈希表等。除了用于存储数据，Redis 还可以用于消息队列和发布/订阅模式。

在本文中，我们将介绍如何使用 Python 连接 Redis 并进行订阅操作。订阅操作是指客户端向 Redis 服务器订阅某个频道，一旦有消息发布到该频道，订阅的客户端就会收到消息。

安装 Redis-py

要在 Python 中使用 Redis，我们需要安装 redis-py 包。可以使用 pip 来安装：

pip install redis

连接到 Redis 服务器

在进行订阅操作之前，我们需要先连接到 Redis 服务器。首先，导入 redis 模块并创建 Redis 客户端：

import redis

r = redis.Redis(host='localhost', port=6379)

上面的代码中，我们连接到了本地的 Redis 服务器，默认端口为 6379。如果 Redis 服务器在不同的主机或端口上运行，需要相应地修改参数。

订阅频道

使用 redis-py 可以很容易地订阅频道。首先，我们需要创建一个 PubSub 对象，并调用 subscribe 方法来订阅频道：

pubsub = r.pubsub()
pubsub.subscribe('mychannel')

上面的代码中，我们创建了一个 PubSub 对象，并订阅了名为 mychannel 的频道。如果我们希望订阅多个频道，可以多次调用 subscribe 方法。

接收消息

一旦订阅了频道，我们可以调用 get_message() 方法来接收消息。下面是一个简单的循环，不断接收消息并输出到控制台：

for message in pubsub.listen():
    if message['type'] == 'message':
        data = message['data'].decode('utf-8')
        print(f"Received message: {data}")

在上面的代码中，我们通过监听 PubSub 对象来接收消息，当消息类型为 message 时，我们将消息数据解码为 UTF-8 编码并输出到控制台。

发布消息

要向指定频道发布消息，可以使用 publish 方法：

r.publish('mychannel', 'Hello, Redis!')

上面的代码将字符串 'Hello, Redis!' 发布到名为 mychannel 的频道上。订阅了该频道的客户端将会接收到这条消息。

完整示例

下面是一个完整的示例，演示了如何订阅频道并发布消息：

import redis

r = redis.Redis(host='localhost', port=6379)
pubsub = r.pubsub()
pubsub.subscribe('mychannel')

def handle_message(message):
    data = message['data'].decode('utf-8')
    print(f"Received message: {data}")

while True:
    message = pubsub.get_message()
    if message:
        handle_message(message)

    # Publish a message
    msg = input("Enter a message to publish: ")
    r.publish('mychannel', msg)

在上面的示例中，我们通过调用 get_message() 方法来获取消息，并通过 handle_message 函数处理消息。一旦有消息发布到 mychannel 频道，我们就会收到消息并打印出来。同时，我们还可以输入消息并用 publish 方法发布到频道中。