Python 异步: 使用和查询任务(8)

2023-02-16 18:57:51     来源:腾讯云

任务是异步程序的货币。在本节中,我们将仔细研究如何在我们的程序中与它们交互。

1. 任务生命周期

异步任务具有生命周期。首先,任务是从协程创建的。然后安排在事件循环中独立执行。在某个时候,它会运行。

在运行时它可能会被挂起,例如等待另一个协程或任务。它可能正常完成并返回结果或因异常而失败。


【资料图】

另一个协程可能会介入并取消任务。最终,它将完成并且无法再次执行。

我们可以将这个生命周期总结如下:

创建预定取消运行暂停结果Exception取消完成

请注意,Suspended、Result、Exception 和 Canceled 本身并不是状态,它们是正在运行的任务的重要转换点。

下图总结了此生命周期,显示了每个阶段之间的转换。

现在我们已经从高层次上熟悉了任务的生命周期,让我们仔细看看每个阶段。

2. 如何检查任务状态

创建任务后,我们可以检查任务的状态。我们可能要检查两种状态,它们是:

任务是否完成任务是否取消

让我们依次仔细看看每一个。

2.1. 检查任务是否完成

我们可以通过 done() 方法检查任务是否完成。如果任务完成,该方法返回 True,否则返回 False。

# check if a task is doneif task.done():# ...

如果任务有机会运行但现在不再运行,则该任务已完成。已安排的任务未完成。同样,正在运行的任务未完成。

如果出现以下情况,则完成任务:

协程正常结束。协程显式返回。协程中出现意外错误或异常任务被取消。

2.2. 检查任务是否取消

我们可以通过 cancelled() 方法检查任务是否被取消。如果任务被取消,该方法返回 True,否则返回 False。

...# check if a task was canceledif task.cancelled():# ...

如果在任务上调用 cancel() 方法并成功完成,则任务被取消,例如 cancel() 返回 True。

如果未调用 cancel() 方法,或者调用了 cancel() 方法但未能取消任务,则不会取消任务。

3. 如何获取任务结果

我们可以通过 result() 方法获取任务的结果。这将返回由 Task 包装的协程的返回值,如果包装的协程没有显式返回值,则返回 None 。

...# get the return value from the wrapped coroutinevalue = task.result()

如果协程引发未处理的错误或异常,则在调用 result() 方法时会重新引发,并且可能需要处理。

...try:# get the return value from the wrapped coroutinevalue = task.result()except Exception:# task failed and there is no result

如果任务被取消,则在调用 result() 方法时会引发 CancelledError 异常,可能需要进行处理。

...try:# get the return value from the wrapped coroutinevalue = task.result()except asyncio.CancelledError:# task was canceled

因此,最好先检查任务是否已取消。

...# check if the task was not canceledif not task.cancelled():# get the return value from the wrapped coroutinevalue = task.result()else:# task was canceled

如果任务尚未完成,则在调用 result() 方法时会引发 InvalidStateError 异常,可能需要进行处理。

...try:# get the return value from the wrapped coroutinevalue = task.result()except asyncio.InvalidStateError:# task is not yet done

因此,最好先检查任务是否已完成。

...# check if the task is not doneif not task.done():await task# get the return value from the wrapped coroutinevalue = task.result()

4. 如何获取任务异常

任务包装的协程可能会引发未处理的异常。这实际上会取消任务。

我们可以通过 exception() 方法在任务包装的协程中检索未处理的异常。

...# get the exception raised by a taskexception = task.exception()

如果包装协程中未引发未处理的异常,则返回 None 值。

如果任务被取消,则在调用 exception() 方法时会引发 CancelledError 异常,可能需要对其进行处理。

...try:# get the exception raised by a taskexception = task.exception()except asyncio.CancelledError:# task was canceled

因此,最好先检查任务是否已取消。

...# check if the task was not canceledif not task.cancelled():# get the exception raised by a taskexception = task.exception()else:# task was canceled

如果任务尚未完成,则在调用 exception() 方法时会引发 InvalidStateError 异常,可能需要进行处理。

...try:# get the exception raised by a taskexception = task.exception()except asyncio.InvalidStateError:# task is not yet done

因此,最好先检查任务是否已完成。

...# check if the task is not doneif not task.done():await task# get the exception raised by a taskexception = task.exception()

5. 如何取消任务

我们可以通过 cancel() 方法取消计划任务。如果任务被取消,则 cancel 方法返回 True,否则返回 False。

...# cancel the taskwas_cancelled = task.cancel()

如果任务已经完成,则无法取消,cancel() 方法将返回 False,任务不会处于已取消状态。

下次任务有机会运行时,它将引发 CancelledError 异常。如果 CancelledError 异常未在包装协程内处理,任务将被取消。否则,如果在包装协程内处理了 CancelledError 异常,任务将不会被取消。

cancel() 方法还可以接受一个消息参数,该参数将在 CancelledError 的内容中使用。

6. 如何在任务中使用回调

我们可以通过 add_done_callback() 方法向任务添加完成回调函数。此方法采用任务完成时要调用的函数的名称。回调函数必须将 Task 实例作为参数。

# done callback functiondef handle(task):print(task) ...# register a done callback functiontask.add_done_callback(handle)

回想一下,当包装的协程返回时正常完成、引发未处理的异常或取消任务时,任务可能会完成。add_done_callback() 方法可用于添加或注册任意数量的 done 回调函数。

我们还可以通过 remove_done_callback() 函数删除或注销回调函数。

...# remove a done callback functiontask.remove_done_callback(handle)

7. 如何设置任务名称

一个任务可能有一个名字。如果多个任务是从同一个协程创建的,那么这个名称会很有用,我们需要一些方法以编程方式区分它们。当通过“名称”参数从协程创建任务时,可以设置名称。

...# create a task from a coroutinetask = asyncio.create_task(task_coroutine(), name="MyTask")

任务的名称也可以通过 set_name() 方法设置。

...# set the name of the tasktask.set_name("MyTask")

我们可以通过 get_name() 方法检索任务的名称。

...# get the name of a taskname = task.get_name()

标签: 编程算法

包装