python redis 实现简单的消息订阅

Posted Ivy丶

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了python redis 实现简单的消息订阅相关的知识,希望对你有一定的参考价值。

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模块序列化。

以上是关于python redis 实现简单的消息订阅的主要内容,如果未能解决你的问题,请参考以下文章

利用redis简单实现消息订阅和发布

python实现 redis订阅与发布

利用Redis作消息队列,实现生产消费和发布订阅

python mitmdump抓包与redis订阅消息

Redis Pub/Sub 发布订阅模式的深度解析与实现消息队列

基于redis(订阅发布)实现python和java进程间通信