关闭/修改来自 FASTAPI 端点的现有 aio_pika 连接

Question

我使用 aio_pika 作为 rabbitmq 的健壮接口。 FASTAPI 启动事件创建一个 aio_pika 通道并开始使用一个队列，如下所示。 我需要做的是有一个端点来关闭/修改这个已经在运行的连接。 这允许另一个远程应用程序关闭/修改此使用者。

我想要这个端点，但我不知道如何将已经由 FASTAPI 启动事件实例化的 swarm_connection 传递给端点。

#The endpoint I need, but how to pass in swarm_connection?
@app.get("/close")
async def close_pika():
    swarm_connection.close()

到目前为止，我正在努力启动的代码

@app.on_event("startup")
def startup():
    loop = asyncio.get_event_loop()
    asyncio.ensure_future(main(loop))

async def main(loop):
    #RabbitMQ
    swarm_connection = await aio_pika.connect_robust(host=host,
                                               port=5672,
                                               login=login,
                                               password=pass,
                                               loop=loop
                                               )
    # Creating channel
    swarm_channel = await swarm_connection.channel()
    # Maximum message count which will be processing at the same time.
    await swarm_channel.set_qos(prefetch_count = 1)

    org1_queue = await swarm_channel.declare_queue('org1', auto_delete=False, durable=True, arguments={'x-max-priority':1})
    await org1_queue.consume(solve_problem_test)

Answer 1

asyncio.wait方法可以等待几个异步任务中的第一个完成（如果使用参数return_when=asyncio.FIRST_COMPLETED运行）。 所以一个想法是使用asyncio.Event作为信号。 修改main方法中的最后一行以等待队列或设置的事件，然后在/close端点中设置事件。 类似于以下内容：

closing_connection = asyncio.Event()

@app.get("/close")
async def close_pika():
    closing_connection.set()

# ...
async def main(loop):
    # ...
    _, pending = await asyncio.wait(
        [org1_queue.consume(solve_problem_test), closing_connection.wait()],
        return_when=asyncio.FIRST_COMPLETED
    )
    for task in pending:
        task.cancel()

    # Here you can call swarm_connection.close() or whatever you need to do

最后一部分是取消任何未等待的任务。