如何提前关闭未完全迭代的嵌套异步生成器？

这个问题我之前也碰到过，本质是嵌套异步生成器的关闭流程没处理好导致的状态异常，我来给你拆解下怎么解决：

问题根源 #

当你调用stream.aclose()时，系统会尝试向gen2抛出GeneratorExit异常，而gen2内部的async for会把这个异常传递给gen1。但此时gen1只生成了第一个值，它的协程状态还没完全就绪，直接传递异常就会触发can't send non-None value to a just-started coroutine这个错误——简单说就是，内部生成器还没准备好接收关闭信号，就被强制发送了异常。

修复方案 #

我们需要在外部生成器gen2里手动处理关闭逻辑，确保内部生成器gen1能被正确关闭，而不是让异常自动传递。修改后的代码如下：

import asyncio

async def gen1():
    try:
        yield 1
        yield 2
    finally:
        # 这里可以添加gen1的资源清理逻辑，比如关闭文件、释放连接等
        print("gen1已完成清理")

async def gen2():
    # 手动持有内部生成器的引用
    inner_gen = gen1()
    try:
        async for nr in inner_gen:
            yield nr
    except GeneratorExit:
        # 先手动关闭内部生成器，确保其协程状态正确处理
        await inner_gen.aclose()
        # 重新抛出异常，保证外部的关闭流程正常完成
        raise

async def main():
    stream = gen2()
    print(await stream.__anext__()) # 输出1
    await stream.aclose()

asyncio.run(main())

关键修改点 #

• 在gen2中手动保存gen1的引用，而不是直接在async for中调用，这样我们能在捕获GeneratorExit时主动关闭内部生成器。
• 给gen1加上try...finally块，这是异步生成器中做资源清理的标准方式，确保即使提前关闭，也能释放占用的资源。
• 捕获GeneratorExit后先关闭内部生成器，再重新抛出异常，既保证了内部生成器的正确关闭，也不打断外部的关闭流程。

这样修改后，你再运行代码就不会出现那个RuntimeError了，而且生成器的关闭流程也会更稳健。

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