怎样在Python中为asyncio事件循环设置超时取消?

小静小哥_7745

小静小哥_7745

2026-10-09

800人浏览

原创

asyncio.wait_for 是首选方案,因其原生支持超时取消、自动抛出 timeouterror 且无需手动管理任务状态;错误用法包括混用同步 sleep 或误信 create_task 自带超时。

怎样在python中为asyncio事件循环设置超时取消?

asyncio.wait_for 为什么是首选方案

asyncio.wait_for 是 asyncio 原生支持超时取消的最直接方式,它不依赖手动管理任务状态或信号,也不需要自己写 timeout 逻辑。它在超时发生时会自动 cancel() 被包装的协程,并抛出 asyncio.TimeoutError。

常见错误是试图用 time.sleep() 或同步计时器套在协程外,这会阻塞整个事件循环;或者误以为 asyncio.create_task() 自带超时能力 —— 它没有。

使用场景包括:等待网络响应、调用第三方异步 API、等待某个条件变量就绪。

import asyncio
<p>async def fetch_data():
await asyncio.sleep(3)
return "done"</p><p>try:
result = await asyncio.wait_for(fetch_data(), timeout=2.0)
except asyncio.TimeoutError:
print("fetch_data timed out")</p>

注意:timeout 参数单位是秒(float),不是毫秒;超时后原协程会被取消,但已执行的 await 点(如底层 socket read)是否真正中断,取决于被 await 对象是否响应取消 —— 比如 aiohttp.ClientSession.get() 支持,而某些自定义协程若没检查 asyncio.current_task().cancelled() 就可能“假取消”。

什么时候该用 asyncio.shield + wait_for 组合

当你想防止超时取消波及到某些必须完成的清理逻辑时,asyncio.shield() 就派上用场了。比如你启动了一个协程做资源释放(如关闭连接、写日志),又希望主逻辑有超时,但不希望清理动作被连带取消。

常见错误是把整个业务逻辑包进 shield() —— 这会让超时失效;正确做法是只 shield 清理部分,主流程仍由 wait_for 控制。

async def main_work():
    # 主工作,可能耗时
    await asyncio.sleep(5)
<p>async def cleanup():</p><div class="aritcle_card flexRow artxards">
											<div class="artcardd flexRow">
												<a class="aritcle_card_img" rel="nofollow" href="/xiazai/skill7351" title="testing-python"><img
														src="https://img.php.cn/upload/skill/000/000/081/179143938488980.jpg" alt="testing-python" onerror="this.onerror='';this.src='/static/lhimages/moren/morentu.png'" ></a>
												<div class="aritcle_card_info flexColumn">
													<a rel="nofollow" href="/xiazai/skill7351" title="testing-python" class="overflowclass">testing-python</a>
													<p class="overflowclass">使用pytest编写和评估有效的Python测试。适用于编写测试、审查测试代码、调试测试失败或提高测试覆盖率。</p>
												</div>
												<a rel="nofollow" href="/xiazai/skill7351" title="testing-python" class="aritcle_card_btn flexRow flexcenter"><b></b><span>下载</span>
												</a>
											</div>
										</div><h1>必须运行完的清理</h1><pre class="brush:python;toolbar:false;">await asyncio.sleep(1)
print("cleanup done")

async def run_with_timeout(): task = asyncio.create_task(main_work()) try: await asyncio.wait_for(task, timeout=2.0) except asyncio.TimeoutError: print("main_work cancelled")

cleanup 不受 task 取消影响

    await asyncio.shield(cleanup())

性能影响很小,shield() 本质只是让被包裹的协程忽略外部取消请求,直到它自己结束或抛出异常。

loop.call_later + cancel() 手动实现的风险点

有人会想到用 loop.call_later(timeout, task.cancel) 配合 asyncio.create_task() 来模拟超时。这种方式可行,但容易踩坑:

  • loop.call_later 的回调在事件循环线程中执行,如果任务已在运行中并处于不可取消点(如等待一个不响应取消的 await),cancel() 只会标记状态,不会立即退出;
  • 没有统一异常类型,需手动判断 task.cancelled() 并处理;
  • 如果任务提前完成,忘记取消定时器,会导致 task.cancel() 在已完成任务上调用 —— 这不会报错,但属于冗余操作,且可能掩盖真实问题;
  • 多个并发任务时,每个都要维护自己的 timer handle,容易漏清理。
# 不推荐:手动管理取消定时器
loop = asyncio.get_running_loop()
task = asyncio.create_task(some_coro())
handle = loop.call_later(2.0, task.cancel)
<p>try:
result = await task
except asyncio.CancelledError:
print("cancelled by timer")
finally:
handle.cancel()  # 必须手动取消定时器,否则可能触发已结束任务的 cancel()</p>

除非你在写底层库或调试事件循环行为,否则没必要绕过 wait_for。

超时嵌套与子协程取消传播要注意什么

如果被 wait_for 包裹的协程内部又 await 了另一个协程,而那个协程本身也用了 wait_for,超时取消会逐层向内传播。但前提是:各层都未用 shield 或忽略取消。

容易被忽略的是:某些异步库(如早期版本的 aiomysql 或自定义协议解析器)在 await 底层 socket 时未适配 CancelledError,导致即使外层超时,内层仍在等待,最终整个任务卡住。

验证方法是在协程开头加一句:

if asyncio.current_task().cancelled():
    raise asyncio.CancelledError
并在每个 await 后检查返回值或异常。

另外,wait_for 的 timeout 是针对整个协程执行时间,不是某一次 await 的耗时。如果你需要“每次网络请求单独超时”,得在每次 await 前套一层 wait_for,而不是只包最外层。

Python免费学习笔记(深入):立即使用
在学习笔记中,你将探索 Python 的核心概念和高级技巧!

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

python

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

2023.07.20

1671

4

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

2023.07.25

4204

7

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2023.07.31

1669

3

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

2023.08.03

24457

23

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.04

2987

5

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.04

3007

5

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.11

1163

5

python合并两个列表
python合并两个列表

Python是一种强大的编程语言,具有许多方便的功能和工具。在Python中,有多种方法可以合并两个列表。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.10

596

4

python是前端还是后端
python是前端还是后端

Python属于前端也属于后端,其灵活性和丰富的生态系统使得开发人员能够在不同的领域中灵活运用。本专题为大家提供python相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.11

2343

5

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程