【Golang】基于内存的简单发布订阅系统

RumIV

已于 2024-10-08 10:38:57 修改

阅读量14

点赞数 1

分类专栏： Golang 文章标签： golang python 开发语言

于 2024-10-08 10:22:16 首次发布

原文链接：https://mp.weixin.qq.com/s?__biz=MzIxMzgyNTQyOA==&mid=2247488613&idx=1&sn=8725d95216126b3dea666cc6cf0cc17b&chksm=962b244066046de17e0b08af3797cc5a194454b870e30a97b4df4af0f82916940325c9c0db1e&mpshare=1&scene=1&srcid=1003f9v6gDfNCFLKMNO8j7Fx&sharer_shareinfo

版权

Golang 专栏收录该内容

23 篇文章 0 订阅

订阅专栏

本文将深入讲解一个基于内存的简单发布订阅系统，它可以用于处理各种任务。我们将从基本概念开始，逐步介绍其设计、实现和应用。

发布订阅模式

发布订阅模式是一种消息传递模式，它允许程序组件之间进行异步通信。在这个模式中，发布者（Publisher）将消息发布到一个主题（Topic），而订阅者（Subscriber）则订阅特定的主题，并接收来自发布者的消息。

发布订阅模式的主要优势在于：

解耦：发布者和订阅者之间没有直接的依赖关系，它们通过主题进行通信。
异步：发布者可以立即发布消息，而无需等待订阅者处理消息。
可扩展性：可以轻松地添加新的发布者或订阅者，而不会影响其他组件。

基于内存的发布订阅系统

基于内存的发布订阅系统将主题和消息存储在内存中，这使得它具有以下特点：

速度快：由于消息存储在内存中，因此可以快速发布和接收消息。
简单易用：实现起来相对简单，不需要额外的数据库或消息队列。
适合小型应用：对于少量发布者和订阅者，以及不需要持久化消息的应用来说，它是一个不错的选择。

系统设计

以下是一个基于内存的简单发布订阅系统的基本设计：

主题管理器（TopicManager）：负责管理所有主题，包括创建、删除和获取主题。
消息队列（MessageQueue）：用于存储每个主题的消息，并提供发布和订阅消息的接口。
发布者（Publisher）：将消息发布到指定的主题。
订阅者（Subscriber）：订阅特定的主题，并接收来自发布者的消息。

代码实现

以下是一个使用 Python 实现的基于内存的简单发布订阅系统：

import threading

class TopicManager:
    def __init__(self):
        self.topics = {}

    def create_topic(self, topic_name):
        if topic_name not in self.topics:
            self.topics[topic_name] = MessageQueue()
            return True
        return False

    def get_topic(self, topic_name):
        if topic_name in self.topics:
            return self.topics[topic_name]
        return None

    def delete_topic(self, topic_name):
        if topic_name in self.topics:
            del self.topics[topic_name]
            return True
        return False

class MessageQueue:
    def __init__(self):
        self.messages = []
        self.subscribers = []
        self.lock = threading.Lock()

    def publish(self, message):
        with self.lock:
            self.messages.append(message)
            for subscriber in self.subscribers:
                subscriber.notify(message)

    def subscribe(self, subscriber):
        with self.lock:
            self.subscribers.append(subscriber)

    def unsubscribe(self, subscriber):
        with self.lock:
            self.subscribers.remove(subscriber)

class Subscriber:
    def __init__(self, topic_name, handler):
        self.topic_name = topic_name
        self.handler = handler
        self.topic = topic_manager.get_topic(topic_name)
        if self.topic:
            self.topic.subscribe(self)

    def notify(self, message):
        self.handler(message)

class Publisher:
    def __init__(self):
        pass

    def publish(self, topic_name, message):
        topic = topic_manager.get_topic(topic_name)
        if topic:
            topic.publish(message)

# 初始化主题管理器
topic_manager = TopicManager()

# 创建一个主题
topic_manager.create_topic("my_topic")

# 定义一个订阅者，处理来自"my_topic"主题的消息
def handle_message(message):
    print(f"收到消息：{message}")

subscriber = Subscriber("my_topic", handle_message)

# 定义一个发布者，发布消息到"my_topic"主题
publisher = Publisher()
publisher.publish("my_topic", "Hello world!")

# 运行程序，订阅者将收到发布者的消息