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

alwaysbot

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

import asyncio

def main():
    while True:
        result_1 = asyncio.create_task(func_1())
        result_2 = asyncio.create_task(func_2())
        result_3 = asyncio.create_task(func_3())
        #
        #如果result_1 ,result_2 ,result_3 中有一个有返回值就执行do_sth(),然后继续循环
        #
        await asyncio.sleep(0)

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

最佳答案

月排行榜 / 总排行榜

kogawananari

2020-10-1 12:57:42

import asyncio
from asyncio import FIRST_COMPLETED
import random

arr = [3,5,7]

async def func1(n):
    await asyncio.sleep(n)
    return 'func1 done'

async def func2(n):
    await asyncio.sleep(n)
    return 'func2 done'

async def func3(n):
    await asyncio.sleep(n)
    return 'func3 done'


async def main():
    random.shuffle(arr)
    result_1 = asyncio.create_task(func1(arr[0]))
    result_2 = asyncio.create_task(func2(arr[1]))
    result_3 = asyncio.create_task(func3(arr[2]))
    done, pending = await asyncio.wait({result_1,result_2,result_3},return_when=FIRST_COMPLETED)
    #其实有bug 两个任务同时完成就会都完成 done集合就会有两个
    for i in done:
        print(i.result()) 

while 1:
    asyncio.run(main())

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

跳转到最佳答案楼层

kogawananari

import asyncio
from asyncio import FIRST_COMPLETED
import random

arr = [3,5,7]

async def func1(n):
    await asyncio.sleep(n)
    return 'func1 done'

async def func2(n):
    await asyncio.sleep(n)
    return 'func2 done'

async def func3(n):
    await asyncio.sleep(n)
    return 'func3 done'


async def main():
    random.shuffle(arr)
    result_1 = asyncio.create_task(func1(arr[0]))
    result_2 = asyncio.create_task(func2(arr[1]))
    result_3 = asyncio.create_task(func3(arr[2]))
    done, pending = await asyncio.wait({result_1,result_2,result_3},return_when=FIRST_COMPLETED)
    #其实有bug 两个任务同时完成就会都完成 done集合就会有两个
    for i in done:
        print(i.result()) 

while 1:
    asyncio.run(main())

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