aio-pika 结合 fastapi

理解 aio-pika 和 fastapi 的基本概念与用途

• aio-pika 是一个异步的 RabbitMQ 客户端库，为 Python 提供了异步的 RabbitMQ 连接和消息处理功能。
• fastapi 是一个用于构建 API 的 Python 框架，它支持异步操作，并提供了许多实用的工具和功能来简化 API 的开发。

创建一个简单的 FastAPI 项目

首先，创建一个简单的 FastAPI 项目，例如使用 fastapi 创建一个简单的 API 服务：

from fastapi import FastAPI

app = FastAPI()

@app.get("/")
def read_root():
    return {"message": "Hello World"}

在 FastAPI 项目中集成 aio-pika，设置 RabbitMQ 连接

接下来，需要在 FastAPI 项目中集成 aio-pika 以连接 RabbitMQ，并设置相关的连接参数。

import asyncio
from aio_pika import connect

class RabbitMQClient:
    def __init__(self, host, port, user, password):
        self.host = host
        self.port = port
        self.user = user
        self.password = password
        self.connection: asyncio.Task = None

    async def connect(self):
        if not self.connection:
            self.connection = asyncio.create_task(
                connect(
                    host=self.host,
                    port=self.port,
                    username=self.user,
                    password=self.password,
                )
            )
            await self.connection

    async def close(self):
        if self.connection:
            self.connection.cancel()
            try:
                await self.connection
            except asyncio.CancelledError:
                pass

rabbitmq_client = RabbitMQClient("localhost", 5672, "guest", "guest")

编写一个使用 aio-pika 发送消息的 FastAPI 路由

在 FastAPI 中，可以编写一个路由来处理消息发送的请求。例如，我们可以创建一个发送消息的 API 端点：

from fastapi import FastAPI, HTTPStatus
from aio_pika import Message

app = FastAPI()

@app.post("/send_message")
async def send_message(message: str):
    await rabbitmq_client.connect()
    channel = await rabbitmq_client.connection.channel()
    exchange = await channel.declare_exchange("direct", "direct")
    await channel.basic_publish(
        exchange=exchange,
        routing_key="test_queue",
        body=Message(message.encode("utf-8"), delivery_mode=2),
    )
    await rabbitmq_client.close()
    return {"message": "Message sent successfully"}

编写一个使用 aio-pika 接收消息的 FastAPI 路由，并处理接收到的消息

同样，我们可以编写一个接收消息的路由，并使用 aio-pika 处理接收到的消息：

@app.get("/receive_message")
async def receive_message():
    await rabbitmq_client.connect()
    channel = await rabbitmq_client.connection.channel()
    queue = await channel.declare_queue("test_queue", auto_delete=True)
    await queue.consume(self.on_message_callback)

    async def on_message_callback(message: Message):
        print("Received message:", message.body.decode("utf-8"))
        await message.ack()

    return {"message": "Waiting for messages"}