python 的 Task 如何封装协程

在Python中，Task用于封装协程，其目的是使协程能够并发运行。Task的一个核心特点是它能够将协程的执行与其产生结果解耦，让开发者能夏洛特地调度和管理协程的执行。在asyncio模块中，Task对象通常由asyncio.create_task()工厂函数创建，它接受一个协程作为参数。create_task()将协程封装为一个Task，这样就可以让协程加入事件循环中去异步运行。

在详细描述前，理解协程（coroutine）的概念是重要的。协程是一种程序组件，可以在特定时刻暂停它的执行，并在稍后的某个时间点恢复。这种能力使得协程在处理I/O密集型和高并发的应用程序时异常有用。

创建Task来封装协程是进行并发编程的一种高效方法。通过创建Task，你不仅可以在事件循环中调度协程执行，还可以检索协程的状态和结果，处理异常，以及取消协程的执行。这一切都可以异步地完成，从而大大提高程序的性能和响应能力。

一、创建TASK

要创建Task封装协程，首先需要有一个运行中的事件循环。使用asyncio.get_event_loop()可以获取当前线程的事件循环。如果没有运行中的事件循环，首先需要创建一个新的事件循环。

import asyncio
loop = asyncio.get_event_loop()  # 获取当前线程的事件循环
coroutine = some_coroutine_function()  # 定义协程
task = loop.create_task(coroutine)  # 创建Task对象

然而，从Python 3.7开始，推荐的方式是使用asyncio.run()来运行顶层的协程，该函数会创建一个新的事件循环，并在运行完成后关闭事件循环。

二、运行TASK

创建Task之后，要开始运行，需要将Task提交至事件循环中。

import asyncio
async def my_coroutine():
    # 协程的内容
    ...
task = asyncio.create_task(my_coroutine())  # 封装协程
awAIt task  # 在协程中等待Task完成

此用法适用于已经在协程中的情况下。如果在同步代码中，可以使用如下方式启动协程和Task：

asyncio.run(my_coroutine())

三、TASK的状态和结果

Task在执行过程中，可以查询其状态。一个Task可以处于不同的状态：'PENDING'、'RUNNING'、'CANCELLED'或者'FINISHED'。

# 为Task状态和结果的查询提供示例代码
import asyncio
async def my_coroutine(i):
    print(f"Task {i} is running")
    await asyncio.sleep(1)
    return f"Result from Task {i}"
async def main():
    tasks = [asyncio.create_task(my_coroutine(i)) for i in range(5)]
    for task in tasks:
        print(f"Task {task.get_name()} is {task.get_state()}")
        result = await task
        print(f"Task {task.get_name()} result: {result}")
asyncio.run(main())

此段代码中，main()创建和启动了五个Task，等待它们完成，并打印出每个Task的状态和结果。

四、管理TASK

可以利用asyncio提供的一些工具函数来管理和取消Task。例如，await asyncio.gather(*tasks)可以等待一组Task全部完成，task.cancel()可以取消一个正在运行的Task。

import asyncio
async def cancel_me():
    print('cancel_me(): before sleep')
    try:
        # 在这里等待取消
        await asyncio.sleep(3600)
    except asyncio.CancelledError:
        print('cancel_me(): cancel sleep')
        raise
    finally:
        print('cancel_me(): after sleep')
async def main():
    # 创建一个任务(Task)
    task = asyncio.create_task(cancel_me())
    # 稍候取消这个Task
    await asyncio.sleep(1)
    task.cancel()
    try:
        await task
    except asyncio.CancelledError:
        print("main(): cancel_me is cancelled now")
asyncio.run(main())