Python asyncio异步生成器如何提前返回并fire and forget处理剩余值

问题根因 #

你的代码无法输出剩余值的核心原因有两个：

• asyncio.run() 会在其传入的顶层协程（这里就是run()）执行完毕后，直接终止并销毁关联的事件循环。你通过create_task提交的handle_remains属于后台任务，run()返回时该任务还没得到执行时间片，事件循环就被关闭了。
• 你在asyncio.run()之后加的time.sleep(10)是阻塞主线程的同步睡眠，此时事件循环已经不存在，后台异步任务根本没有运行的载体。

解决方法 #

方案1：手动管理事件循环（推荐实现fire and forget需求） #

放弃asyncio.run()的自动管理，手动启停事件循环，保证后台任务运行时事件循环存活：

import asyncio

async def async_iter(num):
    for i in range(num):
        await asyncio.sleep(0.5)
        yield i

async def handle_remains(it):
    async for i in it:
        print(i)

async def run():
    it = async_iter(10)
    async for i in it:
        print(i)
        break
    # 提交后台任务
    asyncio.create_task(handle_remains(it))
    # 直接返回第一个元素
    return i

if __name__ == '__main__':
    loop = asyncio.new_event_loop()
    asyncio.set_event_loop(loop)
    # 运行run协程拿到第一个返回值
    first_val = loop.run_until_complete(run())
    print(f"拿到的第一个元素是：{first_val}")
    # 事件循环继续运行，处理后台任务，可按需调整运行时长
    try:
        loop.run_forever()
    except KeyboardInterrupt:
        pass
    finally:
        loop.close()

方案2：调整asyncio.run的顶层协程逻辑 #

如果不想手动管理循环，你可以把后台任务的生命周期也纳入顶层协程管理，这种方式更符合asyncio的设计规范，但不算严格的即发即弃：

import asyncio

async def async_iter(num):
    for i in range(num):
        await asyncio.sleep(0.5)
        yield i

async def handle_remains(it):
    async for i in it:
        print(i)

async def run():
    it = async_iter(10)
    async for i in it:
        print(i)
        break
    task = asyncio.create_task(handle_remains(it))
    # 拿到第一个值后可先执行你需要的同步业务逻辑
    print(f"第一个元素已拿到：{i}")
    # 等待后台任务执行完成后再退出协程
    await task
    return i

if __name__ == '__main__':
    asyncio.run(run())

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