python 消费mq

原创

mob64ca12f2c96c 2024-10-13 04:37:50 ©著作权

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

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

使用 Python 消费消息队列 (MQ)

在现代软件开发中，消息队列（MQ）是实现系统间异步通信的一种重要方式。作为一名刚入行的小白，学习如何在 Python 中使用消息队列是一个非常重要的技能。本篇文章将带你了解整个过程，具体实现消费 MQ 的方法，并逐步讲解每一步需要做的事情和相关的代码。

整体流程概览

以下是实现 Python 消费 MQ 的整体流程：

步骤	操作	描述
1	安装所需库	安装相关MQ库，如pika等
2	建立连接	与MQ服务器建立连接
3	创建通道	在连接上创建一个通道
4	声明队列	声明要消费的消息队列
5	注册回调函数	定义处理消息的回调函数
6	开始消费	调用消费函数，接收消息
7	关闭连接	完成消费后关闭连接

详细步骤与代码实现

第一步：安装所需库

在使用 Python 消费 MQ 之前，你需要安装一些 Python 库，例如 pika，可以通过以下命令实现：

pip install pika

第二步：建立连接

使用 pika 库与消息队列建立连接，以下是代码示例：

import pika  # 导入 pika 库

# 建立一个连接到本地的 RabbitMQ 服务器
connection = pika.BlockingConnection(pika.ConnectionParameters('localhost'))

# 创建一个通道
channel = connection.channel()

第三步：创建通道

已经在上面的步骤中创建了一个通道，我们可以继续使用这个通道进行后续的操作。

第四步：声明队列

你需要指定要消费的队列，以下是代码示例：

# 声明一个队列（如果队列不存在则创建）
channel.queue_declare(queue='hello')  # 'hello' 是队列的名称

第五步：注册回调函数

你需要注册一个回调函数，当接收到消息时，将自动调用这个函数。以下是代码示例：

# 定义一个处理消息的函数
def callback(ch, method, properties, body):
    print(f" [x] Received {body}")  # 输出接收到的消息

第六步：开始消费

通过通道开始消费队列中的消息，以下是代码示例：

# 告诉 RabbitMQ 使用callback函数处理队列中的消息
channel.basic_consume(queue='hello', on_message_callback=callback, auto_ack=True)

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

# 启动消息消费
channel.start_consuming()

第七步：关闭连接

在实际应用中，通常在适当时机关闭连接。由于消费是一个永久的操作，因此通常会在程序退出时再关闭连接，以下是代码示例：

# 在适当的时候调用下面的代码来关闭连接
connection.close()

代码总结

将上述代码整合，得到一个完整的消费者示例：

import pika

# 建立连接
connection = pika.BlockingConnection(pika.ConnectionParameters('localhost'))
channel = connection.channel()

# 声明队列
channel.queue_declare(queue='hello')

# 定义回调函数
def callback(ch, method, properties, body):
    print(f" [x] Received {body}")

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

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

# 关闭连接（代码实际中可能不会触及此行，因为 start_consuming 是个阻塞操作）
connection.close()

类图

以下是该代码的类图，使用 mermaid 语法表示：

classDiagram
    class PikaConnection {
      +connect()
    }
    class Channel {
      +queue_declare(queue)
      +basic_consume(queue, callback)
      +start_consuming()
    }
    class Callback {
      +execute(message)
    }

    PikaConnection --> Channel
    Channel --> Callback

消费流程图

该消息消费者的流程图如下，使用 mermaid 语法表示：

journey
  title 消费 MQ 消息流程
  section 建立连接
    连接到 RabbitMQ : 5: Me
  section 声明队列
    声明队列 : 5: Me
  section 消费消息
    消费消息 : 5: Me
    处理消息 : 5: Me
  section 关闭连接
    关闭连接 : 5: Me