6.异步迭代器

异步迭代器

实现了__aiter__（）和__anext__()方法的对象

__anext__（）必须返回一个awaitable对象

async for会处理异步迭代器的__anext__()方法返回的可等待对象，直到引发StopAsyncIteration异常

 

异步可迭代对象

可在async for语句中被使用的对应,必须通过它的__aiter__方法返回一个asynchronous_iterator

class Reader:
    # 自定义异步迭代器 同事也是异步可迭代对象
    def __init__(self):

        self.count = 0

    async def readline(self):
        self.count += 1
        if self.count == 100:
            return None
        return self.count

    def __aiter__(self):
        return self

    async def __anext__(self):
        v = await  self.readline()
        if v == None:
            return StopAsyncIteration
        return v


async def main():
    obj = Reader()
    # async for必须在一个协程对象内部
    async for item in obj:
        print(item)


asyncio.run(main())