如何利用Redis和Python开发分布式消息推送功能

import redis
import time

class MessagePublisher:
    def __init__(self, channel_name):
        self.redis_conn = redis.StrictRedis(host='localhost', port=6379, db=0)
        self.channel_name = channel_name

    def publish_message(self, message):
        self.redis_conn.publish(self.channel_name, message)

class MessageSubscriber:
    def __init__(self, channel_name):
        self.redis_conn = redis.StrictRedis(host='localhost', port=6379, db=0)
        self.channel_name = channel_name
        self.pubsub = self.redis_conn.pubsub()
        self.pubsub.subscribe(self.channel_name)

    def listen_messages(self):
        for message in self.pubsub.listen():
            if message['type'] == 'message':
                print(f"Received message: {message['data'].decode()}")

if __name__ == '__main__':
    publisher = MessagePublisher('messages')
    subscriber = MessageSubscriber('messages')

    # 发布消息
    publisher.publish_message('Hello, subscribers!')
    time.sleep(1)  # 等待订阅者接收消息

    # 订阅者监听消息
    subscriber.listen_messages()