Python如何连接消息队列系统_MQ消息处理步骤详解【教程】

Python连接MQ核心是选对客户端库、建立可靠连接、正确收发消息并做好异常与确认处理；主流MQ对应库包括RabbitMQ用pika、Kafka用kafka-python、Redis用redis-py、RocketMQ用rocketmq-client-python。

Python连接消息队列（MQ）系统，核心是选对客户端库、建立可靠连接、正确收发消息，并做好异常与确认处理。不同MQ系统协议和API略有差异，但通用逻辑一致。

一、选择MQ系统与对应Python客户端

主流MQ及推荐库：

RabbitMQ → 使用 pika（官方推荐，支持AMQP协议）
Kafka → 使用 kafka-python（纯Python实现，兼容Kafka 0.10+）
Redis（作为轻量级MQ）→ 使用 redis-py（通过List或Pub/Sub模式）
Apache RocketMQ → 使用 rocketmq-client-python（官方Python SDK）

安装示例（以RabbitMQ为例）：
pip install pika

二、RabbitMQ基础连接与消息发送（AMQP流程）

典型步骤：建立连接 → 创建信道 → 声明交换机/队列 → 发布消息

import pika
1. 连接RabbitMQ服务器
connection = pika.BlockingConnection(pika.ConnectionParameters('localhost'))
channel = connection.channel()
2. 确保队列存在（不存在则自动创建）
channel.queue_declare(queue='task_queue', durable=True)
3. 发送消息（持久化 + 消息确认）
channel.basic_publish(
exchange='',
routing_key='task_queue',
body='Hello World!',
properties=pika.BasicProperties(
delivery_mode=2,  # 消息持久化，重启后不丢失
)
)
print(" [x] Sent 'Hello World!'")
connection.close()

三、消费端：可靠接收与手动确认

避免消息丢失的关键是关闭自动确认（auto_ack=False），并显式调用 basic_ack。

def callback(ch, method, properties, body):
    print(f" [x] Received {body.decode()}")
    # 模拟处理耗时任务
    import time
    time.sleep(2)
    # 手动确认消息已处理完成
    ch.basic_ack(delivery_tag=method.delivery_tag)
关闭自动确认
channel.basic_consume(queue='task_queue', on_message_callback=callback)
print(' [*] Waiting for messages. To exit press CTRL+C')
channel.start_consuming()