python redis 实现简单的消息订阅

Posted 2021-03-05 Ivy丶

python + redis 实现简单的消息订阅

订阅端

import redis
from functools import wraps


class Subscribe:
    def __init__(self, channel: str, **kwargs):
        self.coon = redis.StrictRedis(**kwargs)
        self.channel = channel
        self.registerd = list()
        self.course = self.coon.pubsub()
        self.course.subscribe(self.channel)

    def register(self, func):
        self.registerd.append(func)

        @wraps(func)
        def wrapper(*args, **kwargs):
            return func(*args, **kwargs)

        return wrapper

    def listen(self):
        messages = self.course.listen()
        for message in messages:
            self.dispatch(message)

    def dispatch(self, message):
        for handler in self.registerd:
            handler(message)


DEFAULT_CHANNEL = 'channle'

subscribe = Subscribe(DEFAULT_CHANNEL)


@subscribe.register
def print_args(message):
    print(message)


subscribe.listen()

• 先创建一个订阅对象,利用register装饰器来注册监听到的消息的处理函数
• 最后执行listen方法来监听，当当前对象监听到了消息,会自动转交给所有的处理函数

发布端

import redis

coon = redis.StrictRedis()

coon.publish('channel', '发布的消息对象')

•   发布端只要向同一个数据库的里执行使用相同的key执行`publish`命令即可

• 可以利用pickle模块来对python对象进行序列化, 传入redis。如果是线程锁对象,可以使用dill模块序列化。