python – 异步迭代器定期滴答

2019年7月28日 194次阅读

我正在实现一个与异步一起使用的异步迭代器,它应该以(大部分)常规间隔返回一个新值.

我们可以用一个简单的时钟来说明这样的迭代器,它会每隔~n秒递增一个计数器：

import asyncio

class Clock(object):
    def __init__(self, interval=1):
        self.counter = 0
        self.interval = interval
        self.tick = asyncio.Event()
        asyncio.ensure_future(self.tick_tock())

    async def tick_tock(self):
        while True:
            self.tick.clear()
            await asyncio.sleep(self.interval)
            self.counter = self.__next__()
            self.tick.set()

    def __next__(self):
        self.counter += 1
        return self.counter

    def __aiter__(self):
        return self

    async def __anext__(self):
        await self.tick.wait()
        return self.counter

有没有比使用asyncio.Event更好或更清洁的方法？不止一个协程会在这个迭代器上异步.

最佳答案 在我看来,你的方法很好.注意,从python 3.6开始,你也可以使用
asynchronous generators：

async def clock(start=0, step=1, interval=1.):
    for i in count(start, step):
        yield i
        await asyncio.sleep(interval)

但是,您将无法在多个协同程序之间共享它们.您必须在任务中运行时钟并通过异步迭代接口使数据可用,这基本上就是您在代码中所做的.这是possible implementation.

点赞