带有asyncio的惰性迭代器（生成器）

Question

我有一个像这样的阻塞，非异步代码：

def f():
    def inner():
        while True:
            yield read()
    return inner()

使用此代码，调用者可以选择何时停止函数以生成数据。 如何将此更改为异步？ 此解决方案不起作用：

async def f():
    async def inner():
        while True:
            yield await coroutine_read()
    return inner()

...因为yield不能用于async def函数。 如果我从inner()签名中删除async ，我就不await使用await了。

Answer 1

UPD：

从Python 3.6开始，我们有异步生成器，并且能够直接在协程内使用yield 。

---

如上所述，您不能在async funcs中使用yield 。 如果你想创建协 __aiter__ 器，你必须使用__aiter__和__anext__魔术方法手动完成：

import asyncio


# `coroutine_read()` generates some data:
i = 0
async def coroutine_read():
    global i
    i += 1
    await asyncio.sleep(i)
    return i


# `f()` is asynchronous iterator.
# Since we don't raise `StopAsyncIteration` 
# it works "like" `while True`, until we manually break.
class f:
    async def __aiter__(self):
        return self

    async def __anext__(self):
        return await coroutine_read()


# Use f() as asynchronous iterator with `async for`:
async def main():
    async for i in f():
        print(i)
        if i >= 3:
            break


if __name__ == "__main__":
    loop = asyncio.get_event_loop()
    loop.run_until_complete(main())

输出：

1
2
3
[Finished in 6.2s]

您可能还希望看到StopAsyncIteration使用的其他帖子 。