asyncio队列的task_done方法用于通知队列,指示已完成一个任务。它会在从队列中获取一个项目并完成处理后调用。task_done方法通常与join方法一起使用,用于等待队列中的所有任务完成。
以下是一个代码示例,演示了如何使用asyncio队列的task_done方法:
import asyncio
async def worker(queue):
while True:
item = await queue.get()
# 模拟任务处理
await asyncio.sleep(1)
print(f'Processing item: {item}')
queue.task_done()
async def main():
queue = asyncio.Queue()
# 启动10个工作线程
workers = [asyncio.create_task(worker(queue)) for _ in range(10)]
# 添加任务到队列
for item in range(20):
await queue.put(item)
# 等待所有任务完成
await queue.join()
# 取消工作线程
for worker in workers:
worker.cancel()
# 等待所有工作线程退出
await asyncio.gather(*workers, return_exceptions=True)
asyncio.run(main())
在上述代码中,创建了一个名为worker的协程函数,用于处理队列中的每个项目。在每次处理完成后,调用了queue.task_done()方法来通知队列已完成一个任务。
然后,在main函数中,创建了一个asyncio队列,启动了10个工作线程,并添加了20个任务到队列中。然后,使用queue.join()方法等待队列中的所有任务完成。最后,取消工作线程并等待它们退出。
这个例子展示了如何使用asyncio队列的task_done方法来追踪和等待任务的完成。