python 收集后动态添加新任务(*tasks)

yyyllmsg 于 2022-12-10 发布在 Python

关注(0)|答案(1)|浏览(156)

例如以下代码：

async def f1(num):
    while True:
        print(num)
        await asyncio.sleep(2)

class ExampleClass:
  def __init__():
     self.tasks = []

  async def main():
    for i in range(10):
        tasks.append(asyncio.create_task(f1(i)))
    await asyncio.gather(*tasks)

  def add_new_task(task):
      self.tasks.append(task)

然后我在外面的某个地方呼唤

ExampleClass.add_new_task(task)

我需要的是添加新任务并与现有任务异步执行。
也许我应该使用任何其他的结构来实现我想要的东西？重要的是我的任务可能需要永远执行（永远轮询）

python

来源：https://stackoverflow.com/questions/74716923/dynamically-add-new-tasks-after-gathertasks

1条答案

按热度按时间

vjrehmav1#

asyncio.gather函数对于这样的任务并不方便，但是你可以看看Python3.11发布的asyncio.TaskGroup，它允许更容易地动态添加新任务。

import asyncio
from concurrent.futures import wait, FIRST_COMPLETED

async def main():
    async with asyncio.TaskGroup() as group:
        # Create some tasks
        tasks = [
            group.create_task(asyncio.sleep(1.0))
            for _ in range(10)
        ]
        
        # Wait for some tasks to finish
        done, tasks = await wait(tasks, return_when=FIRST_COMPLETED)

        # Add more tasks (/!\ `tasks` is now a set)
        tasks.add(group.create_task(asyncio.sleep(2.0)))
       
       # Wait the rest to complete 
        await asyncio.gather(*tasks)

if __name__ == "__main__":
    asyncio.run(main())

赞(0）回复(0）举报 2022-12-10

我来回答

python 收集后动态添加新任务(*tasks)

1条答案

相关问题

热门标签

最新问答