Python 异步: 等待有时间限制的协程(12)

简介: 我们可以使用 asyncio.wait_for() 函数等待 asyncio 任务或协程超时完成。如果在任务完成之前超时已过,任务将被取消。

1. 什么是 Asyncio wait_for()

asyncio.wait_for() 函数允许调用者等待 asyncio 任务或协程超时完成。如果没有指定超时,wait_for() 函数将等待直到任务完成。如果在任务完成之前指定了超时并超时,那么任务将被取消。

这允许调用者既可以设置他们愿意等待任务完成的时间,又可以通过在超时结束时取消任务来强制执行超时。

现在我们知道了 asyncio.wait_for() 函数是什么,让我们看看如何使用它。

2. 如何使用 Asyncio wait_for()

asyncio.wait_for() 函数接受一个等待和超时。等待对象可能是协程或任务。必须指定超时,并且可以是无超时、整数或浮点秒数。wait_for() 函数返回一个协程,该协程在明确等待或作为任务调度之前不会执行。

...
# wait for a task to complete
await asyncio.wait_for(coro, timeout=10)

如果提供协程,则在执行 wait_for() 协程时将其转换为任务。如果在任务完成之前超时已过,任务将被取消,并引发 asyncio.TimeoutError,这可能需要处理。

...
# execute a task with a timeout
try:
    # wait for a task to complete
    await asyncio.wait_for(coro, timeout=1)
except asyncio.TimeoutError:
    # ...

如果等待的任务因未处理的异常而失败,则该异常将传播回等待 wait_for() 协程的调用者,在这种情况下可能需要处理它。

...
# execute a task that may fail
try:
    # wait for a task to complete
    await asyncio.wait_for(coro, timeout=1)
except asyncio.TimeoutError:
    # ...
except Exception:
    # ...

接下来,让我们看看如何在超时时调用 wait_for()。

3. 带有超时的 Asyncio wait_for() 示例

我们可以探索如何在任务完成之前等待具有超时的协程。在此示例中,我们执行上述协程,但调用方等待 0.2 秒或 200 毫秒的固定超时。回想一下,一秒等于 1,000 毫秒。

任务协程被修改,使其休眠一秒以上,确保超时总是在任务完成之前到期。

# SuperFastPython.com
# example of waiting for a coroutine with a timeout
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 = 1 + random()
    # report message
    print(f'>task got {value}')
    # block for a moment
    await asyncio.sleep(value)
    # report all done
    print('>task done')
 
# main coroutine
async def main():
    # create a task
    task = task_coro(1)
    # execute and wait for the task without a timeout
    try:
        await asyncio.wait_for(task, timeout=0.2)
    except asyncio.TimeoutError:
        print('Gave up waiting, task canceled')
 
# start the asyncio program
asyncio.run(main())

运行示例首先创建 main() 协程并将其用作 asyncio 程序的入口点。main() 协程创建任务协程。然后它调用 wait_for() 并传递任务协程并将超时设置为 0.2 秒。

main()协程被挂起,执行task_coro()。它报告一条消息并休眠片刻。main() 协程在超时结束后恢复。 wait_for()协程取消task_coro()协程,main()协程挂起。

task_coro() 再次运行并响应要终止的请求。它引发 TimeoutError 异常并终止。main() 协程恢复并处理由 task_coro() 引发的 TimeoutError。

这突出显示了我们如何调用带超时的 wait_for() 函数,并在任务未在超时内完成时取消任务。

由于使用了随机数,程序每次运行时的输出都会不同。

>task got 0.685375224799321
Gave up waiting, task canceled
相关文章
|
13天前
|
安全 调度 Python
探索Python中的并发编程:协程与多线程的比较
本文将深入探讨Python中的并发编程技术,重点比较协程与多线程的特点和应用场景。通过对协程和多线程的原理解析,以及在实际项目中的应用案例分析,读者将能够更好地理解两种并发编程模型的异同,并在实践中选择合适的方案来提升Python程序的性能和效率。
|
4天前
|
消息中间件 安全 调度
基于Python的性能优化(线程、协程、进程)
一、多线程 在CPU不密集、IO密集的任务下,多线程可以一定程度的提升运行效率。
|
13天前
|
JavaScript 测试技术 Python
【如何学习Python自动化测试】—— 时间等待
【如何学习Python自动化测试】—— 时间等待
|
13天前
|
调度 Python
探索Python中的异步编程:从回调到协程
本文将介绍Python中的异步编程技术,从最初的回调函数到现代的协程模型。通过对比传统的同步编程方式和异步编程的优劣势,我们深入探讨了Python中异步编程的实现原理,以及如何利用asyncio库和async/await关键字来构建高效的异步应用程序。最后,我们还将讨论一些异步编程的最佳实践和常见问题的解决方法。
|
13天前
|
API UED Python
使用Python进行异步HTTP请求的实践指南
使用Python进行异步HTTP请求的实践指南
24 4
|
13天前
|
Python
Python中的协程:异步编程的利器
Python中的协程:异步编程的利器
17 1
|
13天前
|
缓存 安全 Linux
深入探索Python中的协程
深入探索Python中的协程
|
13天前
|
并行计算 数据处理 开发者
Python并发编程:解析异步IO与多线程
本文探讨了Python中的并发编程技术,着重比较了异步IO和多线程两种常见的并发模型。通过详细分析它们的特点、优劣势以及适用场景,帮助读者更好地理解并选择适合自己项目需求的并发编程方式。
|
13天前
|
网络协议 调度 开发者
python中gevent基于协程的并发编程模型详细介绍
`gevent`是Python的第三方库,提供基于协程的并发模型,适用于I/O密集型任务的高效异步编程。其核心是协程调度器,在单线程中轮流执行多个协程,通过非阻塞I/O实现高并发。主要特点包括协程调度、事件循环的I/O模型、同步/异步编程支持及易用性。示例代码展示了一个使用`gevent`实现的异步TCP服务器,当客户端连接时,服务器以协程方式处理请求,实现非阻塞通信。
17 0
|
13天前
|
数据采集 数据库 C++
python并发编程:并发编程中是选择多线程呢?还是多进程呢?还是多协程呢?
python并发编程:并发编程中是选择多线程呢?还是多进程呢?还是多协程呢?
24 0