如何将aiopika与FastAPI结合使用？

Question

我想在异步工作时使用FastAPI和aio pika创建一个REST服务。对于其他异步数据库驱动程序，我可以在启动时在路由处理程序中获取客户端时创建它们。例如，对于motor，我将声明简单连接管理器：

from motor.motor_asyncio import AsyncIOMotorClient


class Database:
    client: AsyncIOMotorClient = None


db = Database()


async def connect_to_mongo():
    db.client = AsyncIOMotorClient("mongo:27017")


async def close_mongo_connection():
    db.client.close()


async def get_mongo_client() -> AsyncIOMotorClient:
    return db.client

然后添加两个处理程序：

app.add_event_handler("startup", connect_to_mongo)
app.add_event_handler("shutdown", close_mongo_connection)

然后只需使用get_mongo_client就可以得到一个给我的处理程序

这里的问题是aio-pika需要asyncio循环才能运行。以下是文档中的一个示例：

connection = await aio_pika.connect_robust(
        "amqp://guest:guest@127.0.0.1/", loop=loop
    )

使用FastAPI，我没有异步IO循环。是否有任何方法可以将其与示例中的接口一起使用？我可以使用asyncio.get_event_loop()创建新循环并将其传递给connect_robust而不在任何地方使用它吗？像这样：

connection = await aio_pika.connect_robust(
            "amqp://guest:guest@127.0.0.1/", loop=asyncio.get_event_loop()
        )

Answer 1

好的，根据docs，我可以用connect代替connect_robust：

connection = await aio_pika.connect(
        "amqp://guest:guest@127.0.0.1/"
    )