python中Task封裝協(xié)程的知識點總結(jié)

import asyncio
 
async def func():
print(1)
await asyncio.sleep(2)
print(2)
return "返回值"
 
 
async def main():
print("main開始")
 
# 創(chuàng)建協(xié)程，將協(xié)程封裝到Task對象中并添加到事件循環(huán)的任務(wù)列表中，等待事件循環(huán)去執(zhí)行（默認(rèn)是就緒狀態(tài)）。
# 在調(diào)用
task_list = [
asyncio.create_task(func(), name="n1"),
asyncio.create_task(func(), name="n2")
]
 
print("main結(jié)束")
 
# 當(dāng)執(zhí)行某協(xié)程遇到IO操作時，會自動化切換執(zhí)行其他任務(wù)。
# 此處的await是等待所有協(xié)程執(zhí)行完畢，并將所有協(xié)程的返回值保存到done
# 如果設(shè)置了timeout值，則意味著此處最多等待的秒，完成的協(xié)程返回值寫入到done中，未完成則寫到pending中。
done, pending = await asyncio.wait(task_list, timeout=None)
print(done, pending)
 
 
asyncio.run(main())

import asyncio
import arrow


def current_time():
    '''
    獲取當(dāng)前時間
    :return:
    '''
    cur_time = arrow.now().to('Asia/Shanghai').format('YYYY-MM-DD HH:mm:ss')
    return cur_time


async def func(sleep_time):
    func_name_suffix = sleep_time        # 使用 sleep_time（函數(shù) I/O 等待時長）作為函數(shù)名后綴，以區(qū)分任務(wù)對象
    print(f"[{current_time()}] 執(zhí)行異步函數(shù) {func.__name__}-{func_name_suffix}")
    await asyncio.sleep(sleep_time)
    print(f"[{current_time()}] 函數(shù) {func.__name__}-{func_name_suffix} 執(zhí)行完畢")
    return f"【[{current_time()}] 得到函數(shù) {func.__name__}-{func_name_suffix} 執(zhí)行結(jié)果】"


async def run():
    task_list = []
    for i in range(5):
        task = asyncio.create_task(async_func(i))
        task_list.append(task)

    done, pending = await asyncio.wait(task_list, timeout=None)
    for done_task in done:
        print((f"[{current_time()}] 得到執(zhí)行結(jié)果 {done_task.result()}"))

def main():
    loop = asyncio.get_event_loop()
    loop.run_until_complete(run())


if __name__ == '__main__':
    main()