python 发布订阅的消息队列

原创

mob64ca12f18f13 2024-11-02 04:25:59 ©著作权

文章标签 消息队列服务器 python 文章分类 Python 后端开发

©著作权归作者所有：来自51CTO博客作者mob64ca12f18f13的原创作品，请联系作者获取转载授权，否则将追究法律责任

Python 发布/订阅的消息队列实现指南

在现代的软件架构中，消息队列是一种重要的通信方式。它可以有效地解耦系统组件，提升系统的可扩展性和可靠性。在本文中，我们将围绕如何用 Python 实现一个简单的发布/订阅的消息队列进行讲解。

流程概述

在实现发布/订阅消息队列之前，让我们先了解一下整个流程。以下是实现步骤的概述：

步骤	描述	代码位置
1	安装所需的库	N/A
2	创建消息队列的生产者	producer.py
3	创建消息队列的消费者	consumer.py
4	运行生产者和消费者	N/A
5	测试消息的发布和订阅	N/A

实现步骤详解

步骤1：安装所需的库

我们需要安装 pika 库，它是用于RabbitMQ的Python客户端。打开终端并运行以下命令：

pip install pika

步骤2：创建消息队列的生产者

生产者的主要功能是发布消息到消息队列中。以下是 producer.py 的代码：

# producer.py
import pika

# 连接到RabbitMQ服务器
connection = pika.BlockingConnection(pika.ConnectionParameters('localhost'))
channel = connection.channel()

# 声明一个交换机，类型为'fanout'，意味着它将广播消息到所有绑定的队列
channel.exchange_declare(exchange='logs', exchange_type='fanout')

# 定义要发送的消息内容
message = "Hello World!"
channel.basic_publish(exchange='logs', routing_key='', body=message)

print("[x] Sent %r" % message)

# 关闭连接
connection.close()

注释解释：

使用 pika.BlockingConnection 连接到本地 RabbitMQ 服务器。
channel.exchange_declare 声明一个名为 logs 的交换机，并指定其类型为 fanout，这表示发送到该交换机的消息将广播到所有绑定的队列。
使用 channel.basic_publish 方法发送消息。

步骤3：创建消息队列的消费者

消费者的角色是接收消息。以下是 consumer.py 的代码：

# consumer.py
import pika
import sys

# 连接到RabbitMQ服务器
connection = pika.BlockingConnection(pika.ConnectionParameters('localhost'))
channel = connection.channel()

# 声明一个交换机，类型为'fanout'，与生产者保持一致
channel.exchange_declare(exchange='logs', exchange_type='fanout')

# 创建一个临时队列
result = channel.queue_declare('', exclusive=True)
queue_name = result.method.queue

# 绑定队列到交换机
channel.queue_bind(exchange='logs', queue=queue_name)

print('[*] Waiting for logs. To exit press CTRL+C')

# 定义回调函数，处理消息
def callback(ch, method, properties, body):
    print("[x] Received %r" % body)

# 开始消费消息
channel.basic_consume(queue=queue_name, on_message_callback=callback, auto_ack=True)

try:
    channel.start_consuming()
except KeyboardInterrupt:
    print("Stopped consuming")

# 关闭连接
connection.close()

注释解释：

消费者同样连接到 RabbitMQ 服务器。
同样声明交换机 logs，绑定一个临时队列，使其能够接收广播的消息。
定义 callback 函数，处理接收到的消息并打印。

步骤4：运行生产者和消费者

在终端中先运行消费者，然后再运行生产者，这样消费者能马上接收到消息。

打开两个终端窗口：

在第一个窗口，运行以下命令启动消费者：
```
python consumer.py
```
在第二个窗口，运行以下命令启动生产者：
```
python producer.py
```

步骤5：测试消息的发布和订阅

确保消费者正在运行后，执行生产者代码，你会在消费者控制台上看到输出的消息。

项目路线图

下面是整个项目的甘特图，帮助你更直观地理解项目的进度：

gantt
    title 消息队列项目进度
    dateFormat  YYYY-MM-DD
    section 安装和准备
    安装依赖               :done,  des1, 2023-10-01, 1d
    section 生产者实现
    编写生产者代码        :done,  des2, 2023-10-02, 1d
    section 消费者实现
    编写消费者代码       :done,  des3, 2023-10-03, 1d
    section 测试和验证
    运行生产者和消费者  :done,  des4, 2023-10-04, 1d