asyncio.wait() 函數(shù)可用于等待一組異步任務(wù)完成。回想一下，asyncio 任務(wù)是包裝協(xié)程的 asyncio.Task 類的一個(gè)實(shí)例。它允許獨(dú)立調(diào)度和執(zhí)行協(xié)程，Task 實(shí)例提供任務(wù)句柄以查詢狀態(tài)和獲取結(jié)果。

wait() 函數(shù)允許我們等待一組任務(wù)完成。等待調(diào)用可以配置為等待不同的條件，例如所有任務(wù)完成、第一個(gè)任務(wù)完成以及第一個(gè)任務(wù)因錯(cuò)誤而失敗。

接下來，讓我們看看如何使用 wait() 函數(shù)。

2. 如何使用 asyncio.wait()

asyncio.wait() 函數(shù)采用可等待對象的集合，通常是 Task 對象。

這可以是我們創(chuàng)建的列表、字典或任務(wù)對象集，例如通過在列表理解中調(diào)用 asyncio.create_task() 函數(shù)。

...
# create many tasks
tasks = [asyncio.create_task(task_coro(i)) for i in range(10)]

asyncio.wait() 在滿足任務(wù)集合的某些條件之前不會(huì)返回。默認(rèn)情況下，條件是所有任務(wù)都已完成。

wait() 函數(shù)返回兩個(gè)集合的元組。第一個(gè)集合包含所有滿足條件的任務(wù)對象，第二個(gè)集合包含所有其他尚未滿足條件的任務(wù)對象。

這些集被稱為“完成”集和“待定”集。

...
# wait for all tasks to complete
done, pending = await asyncio.wait(tasks)

從技術(shù)上講，asyncio.wait() 是一個(gè)返回協(xié)程的協(xié)程函數(shù)。然后我們可以等待這個(gè)協(xié)程，它將返回集合的元組。

...
# create the wait coroutine
wait_coro = asyncio.wait(tasks)
# await the wait coroutine
tuple = await wait_coro

等待的條件可以由默認(rèn)設(shè)置為 asyncio.ALL_COMPLETED 的“return_when”參數(shù)指定。

...
# wait for all tasks to complete
done, pending = await asyncio.wait(tasks, return_when=asyncio.ALL_COMPLETED)

我們可以通過將 return_when 設(shè)置為 FIRST_COMPLETED 來等待第一個(gè)任務(wù)完成

...
# wait for the first task to be completed
done, pending = await asyncio.wait(tasks, return_when=asyncio.FIRST_COMPLETED)

當(dāng)?shù)谝粋€(gè)任務(wù)完成并在完成集中返回時(shí)，其余任務(wù)不會(huì)被取消并繼續(xù)并發(fā)執(zhí)行。

我們可以通過將 return_when 設(shè)置為 FIRST_EXCEPTION 來等待第一個(gè)任務(wù)因異常而失敗。

...
# wait for the first task to fail
done, pending = await asyncio.wait(tasks, return_when=asyncio.FIRST_EXCEPTION)

在這種情況下，完成集將包含第一個(gè)因異常而失敗的任務(wù)。如果沒有任務(wù)因異常而失敗，則完成集將包含所有任務(wù)，只有在所有任務(wù)完成后 wait() 才會(huì)返回。

我們可以通過以秒為單位的“超時(shí)”參數(shù)指定我們愿意等待給定條件的時(shí)間。

如果在滿足條件之前超時(shí)到期，則返回任務(wù)元組以及當(dāng)時(shí)滿足條件的任何任務(wù)子集，例如如果等待所有任務(wù)完成，則完成的任務(wù)子集。

...
# wait for all tasks to complete with a timeout
done, pending = await asyncio.wait(tasks, timeout=3)

如果在滿足條件之前達(dá)到超時(shí)，則不會(huì)引發(fā)異常并且不會(huì)取消剩余任務(wù)。

現(xiàn)在我們知道如何使用 asyncio.wait() 函數(shù)，讓我們看一些有效的例子。

3. 等待所有任務(wù)的示例

我們可以探索如何使用 asyncio.wait() 等待所有任務(wù)。在這個(gè)例子中，我們將定義一個(gè)簡單的任務(wù)協(xié)程，它生成一個(gè)隨機(jī)值，休眠幾分之一秒，然后用生成的值報(bào)告一條消息。

然后，主協(xié)程將與協(xié)程一起在列表理解中創(chuàng)建許多任務(wù)，然后等待所有任務(wù)完成。

# SuperFastPython.com
# example of waiting for all tasks to complete
from random import random
import asyncio
# coroutine to execute in a new task
async def task_coro(arg):
    # generate a random value between 0 and 1
    value = random()
    # block for a moment
    await asyncio.sleep(value)
    # report the value
    print(f'>task {arg} done with {value}')
# main coroutine
async def main():
    # create many tasks
    tasks = [asyncio.create_task(task_coro(i)) for i in range(10)]
    # wait for all tasks to complete
    done,pending = await asyncio.wait(tasks)
    # report results
    print('All done')
# start the asyncio program
asyncio.run(main())

運(yùn)行示例首先創(chuàng)建 main() 協(xié)程并將其用作 asyncio 程序的入口點(diǎn)。

然后 main() 協(xié)程在列表理解中創(chuàng)建一個(gè)包含十個(gè)任務(wù)的列表，每個(gè)任務(wù)提供一個(gè)從 0 到 9 的唯一整數(shù)參數(shù)。

然后 main() 協(xié)程被掛起并等待所有任務(wù)完成。任務(wù)執(zhí)行。每個(gè)生成一個(gè)隨機(jī)值，休眠片刻，然后報(bào)告其生成的值。

所有任務(wù)完成后，main() 協(xié)程恢復(fù)并報(bào)告最終消息。這個(gè)例子強(qiáng)調(diào)了我們?nèi)绾问褂?wait() 函數(shù)來等待一組任務(wù)完成。

這可能是該函數(shù)最常見的用法。請注意，由于使用了隨機(jī)數(shù)，每次運(yùn)行程序時(shí)結(jié)果都會(huì)不同。

>task 5 done with 0.0591009105682192
>task 8 done with 0.10453715687017351
>task 0 done with 0.15462838864295925
>task 6 done with 0.4103492027393125
>task 9 done with 0.45567100006991623
>task 2 done with 0.6984682905809402
>task 7 done with 0.7785363531316224
>task 3 done with 0.827386088873161
>task 4 done with 0.9481344994700972
>task 1 done with 0.9577302665040541
All done

以上就是Python 異步等待任務(wù)集合的詳細(xì)內(nèi)容，更多關(guān)于Python 異步等待任務(wù)集合的資料請關(guān)注腳本之家其它相關(guān)文章！

您可能感興趣的文章: