Python 结合 Redis 实现高效消息发布/订阅机制

要使用 Python 结合 Redis 开发一个简单的消息订阅系统，你可以利用 Redis 的发布/订阅（Pub/Sub）功能。Redis 的 Pub/Sub 允许消息发布者将消息推送到一个频道，而订阅者可以接收这些频道的消息。

下面是一个简单的例子，展示如何使用 Python 和 Redis 实现一个基本的消息订阅系统。我们将使用 redis-py 库，这是 Python 的一个 Redis 客户端。

环境准备

安装 Redis：确保你的系统上已经安装并运行了 Redis 服务器。
安装 redis-py：你可以通过 pip 安装 redis-py 库。

pip install redis

示例代码

发布者（Publisher）

import redis
import time

def publisher():
    # 连接到 Redis
    r = redis.StrictRedis(host='localhost', port=6379, db=0)
    
    channel = 'my_channel'
    
    try:
        while True:
            message = input("Enter a message to publish (or 'exit' to quit): ")
            if message.lower() == 'exit':
                break
            r.publish(channel, message)
    except KeyboardInterrupt:
        print("Publisher stopped.")

if __name__ == '__main__':
    publisher()

订阅者（Subscriber）

import redis

def subscriber():
    # 连接到 Redis
    r = redis.StrictRedis(host='localhost', port=6379, db=0)
    
    channel = 'my_channel'
    
    # 定义消息处理函数
    def message_handler(message):
        print(f"Received message: {message['data'].decode('utf-8')}")
    
    # 订阅频道
    pubsub = r.pubsub()
    pubsub.subscribe(**{channel: message_handler})
    
    print(f"Subscribed to {channel}. Waiting for messages...")
    
    try:
        # 开始监听消息
        for message in pubsub.listen():
            # 可以在这里处理其他类型的消息，比如 'subscribe', 'unsubscribe' 等
            if message['type'] == 'message':
                # 消息处理已经在 subscribe 中通过回调完成
                pass
    except KeyboardInterrupt:
        print("Subscriber stopped.")
        pubsub.close()

if __name__ == '__main__':
    subscriber()