[已解决]请教一个协程相关问题

alwaysbot

使用协程异步获取func_1,func_2,func_3的数据，如果其中有一个返回，那么执行do_sth()
这里该怎么写呢，如果一直循环create_task会不会有什么影响呢
感谢解答

1. 
   
2. import asyncio
   
3. 
   
4. def main():
   
5.     while True:
   
6.         result_1 = asyncio.create_task(func_1())
   
7.         result_2 = asyncio.create_task(func_2())
   
8.         result_3 = asyncio.create_task(func_3())
   
9.         #
   
10.         #如果result_1 ,result_2 ,result_3 中有一个有返回值就执行do_sth(),然后继续循环
    
11.         #
    
12.         await asyncio.sleep(0)
    
13. 
    
14. if __name__ == '__main__':
    
15.         asyncio.run(main())
    

复制代码

最佳答案

月排行榜 / 总排行榜

kogawananari

2020-10-1 12:57:42

1. import asyncio
   
2. from asyncio import FIRST_COMPLETED
   
3. import random
   
4. 
   
5. arr = [3,5,7]
   
6. 
   
7. async def func1(n):
   
8.     await asyncio.sleep(n)
   
9.     return 'func1 done'
   
10. 
    
11. async def func2(n):
    
12.     await asyncio.sleep(n)
    
13.     return 'func2 done'
    
14. 
    
15. async def func3(n):
    
16.     await asyncio.sleep(n)
    
17.     return 'func3 done'
    
18. 
    
19. 
    
20. async def main():
    
21.     random.shuffle(arr)
    
22.     result_1 = asyncio.create_task(func1(arr[0]))
    
23.     result_2 = asyncio.create_task(func2(arr[1]))
    
24.     result_3 = asyncio.create_task(func3(arr[2]))
    
25.     done, pending = await asyncio.wait({result_1,result_2,result_3},return_when=FIRST_COMPLETED)
    
26.     #其实有bug 两个任务同时完成就会都完成 done集合就会有两个
    
27.     for i in done:
    
28.         print(i.result())
    
29. 
    
30. while 1:
    
31.     asyncio.run(main())

复制代码

loop.run_forever()代替死循环更加灵活

跳转到最佳答案楼层

kogawananari

1. import asyncio
   
2. from asyncio import FIRST_COMPLETED
   
3. import random
   
4. 
   
5. arr = [3,5,7]
   
6. 
   
7. async def func1(n):
   
8.     await asyncio.sleep(n)
   
9.     return 'func1 done'
   
10. 
    
11. async def func2(n):
    
12.     await asyncio.sleep(n)
    
13.     return 'func2 done'
    
14. 
    
15. async def func3(n):
    
16.     await asyncio.sleep(n)
    
17.     return 'func3 done'
    
18. 
    
19. 
    
20. async def main():
    
21.     random.shuffle(arr)
    
22.     result_1 = asyncio.create_task(func1(arr[0]))
    
23.     result_2 = asyncio.create_task(func2(arr[1]))
    
24.     result_3 = asyncio.create_task(func3(arr[2]))
    
25.     done, pending = await asyncio.wait({result_1,result_2,result_3},return_when=FIRST_COMPLETED)
    
26.     #其实有bug 两个任务同时完成就会都完成 done集合就会有两个
    
27.     for i in done:
    
28.         print(i.result())
    
29. 
    
30. while 1:
    
31.     asyncio.run(main())

复制代码

loop.run_forever()代替死循环更加灵活