如何在 [asyncio] 中完成所有任务之前立即获得某人任务的返回？

Question

如何在 [asyncio] 中的所有任务完成之前立即获得某人任务的返回？

import asyncio
import time

print(f"now: {time.strftime('%X')}")

async def test1():
    print(f"test1 started at {time.strftime('%X')}")
    await asyncio.sleep(5)
    with open('test.txt', 'w') as f:
        f.write('...')
    return 'end....'

async def test2(num):
    print(f"test2 started at {time.strftime('%X')}")
    return num * num

async def main(num):
    res = await asyncio.gather(test1(), test2(num))
    return res

def my(num):
    return asyncio.run(main(num))[1]

print(my(5))
print(f"all end at {time.strftime('%X')}")

从上面的代码（python 3.7+），我只能在test1和test2都完成后才能得到test2return。

如何让main函数在test2完成后返回test2，而不是等到test1完成？，因为test2执行得更快。 并且必须执行 test1（生成 test.txt 文件。）

Than 表示 test1 和 test2 异步时尽快返回（test2 的返回）到main 函数或my 函数。

Answer 1

要运行一组 awaitables/coroutines 直到任何 Future/Task 完成或被取消 - 您需要 asyncio.wait coroutine：

...
async def main(num):
    done, pending = await asyncio.wait([test1(), test2(num)], 
                                       return_when=asyncio.FIRST_COMPLETED)
    for coro in done:
        return await coro


def my(num):
    return asyncio.run(main(num))


print(my(5))
print(f"all end at {time.strftime('%X')}")

return_when表示该函数应该何时返回

---

输出：

now: 18:50:16
test1 started at 18:50:16
test2 started at 18:50:16
25
all end at 18:50:16

---

但是由于您需要完成所有协程 - 使用 asyncio.as_completed 方法或打印来自 done 集的结果，然后 - 从 pending 集等待并将 print(my(5)) 更改为 my(5)：

...
async def main(num):
    done, pending = await asyncio.wait([test1(), test2(num)],
                                       return_when=asyncio.FIRST_COMPLETED)
    for coro in done:
        print(await coro)
    for coro in pending:
        await coro


def my(num):
    return asyncio.run(main(num))


my(5)
print(f"all end at {time.strftime('%X')}")