Python async for循环提前break时async上下文管理器finally执行顺序异常

你观察到的行为属于Python异步生成器的默认设计逻辑，并非未修复的bug：

• 当你在async for循环中提前执行break时，异步生成器（即request_all函数返回的对象）会被直接挂起，不会自动触发后续的清理流程。
• 异步生成器内部async with上下文管理器的finally块，只有在生成器的aclose()方法被调用时才会执行。默认状态下aclose()会在异步生成器被垃圾回收时才被隐式调用，这就导致了你看到的done先于finally输出，以及程序退出时抛出RuntimeError: async generator ignored GeneratorExit异常的问题。
• 你提到的相关issue修复的是事件循环退出时异步生成器未被自动清理的问题，和当前提前中断async for的场景无关，因此即使是修复后的Python版本也会复现该行为。

有两种通用的方案可以解决该问题：

方案1：显式调用异步生成器的aclose()方法 #

在break之后主动等待异步生成器完成清理，修改main函数代码如下：

async def main():
    gen = request_all()
    try:
        async for r in gen:
            print(r)
            break
    finally:
        await gen.aclose()
    print('done')

方案2：使用aclosing上下文管理器自动处理清理 #

Python 3.10及以上版本可以直接使用contextlib.aclosing，它会在上下文退出时自动调用异步生成器的aclose()方法：

import asyncio
from contextlib import asynccontextmanager, aclosing

@asynccontextmanager
async def receiving():
    try:
        print('try')
        yield 123
    finally:
        print('finally')

async def request_all():
    async with receiving():
        yield 456

async def main():
    async with aclosing(request_all()) as gen:
        async for r in gen:
            print(r)
            break
    print('done')

asyncio.run(main())

如果需要兼容Python 3.10以下版本，可以自行实现aclosing逻辑：

from contextlib import asynccontextmanager

@asynccontextmanager
async def aclosing(agen):
    try:
        yield agen
    finally:
        await agen.aclose()

两种方案调整后，输出顺序都会符合你的预期，也不会再抛出清理相关的异常。

内容的提问来源于stack exchange，提问作者tzijnge