0

0

Python 异步上下文管理器的实现

舞夢輝影

舞夢輝影

发布时间:2026-01-29 17:33:09

|

429人浏览过

|

来源于php中文网

原创

async with 语句要求对象必须同时实现 __aenter__ 和 __aexit__ 两个异步方法,前者返回可 await 对象,后者接收四个参数并可选抑制异常,二者缺一不可。

python 异步上下文管理器的实现

async with 语句要求对象实现哪些方法

必须同时提供 __aenter____aexit__ 两个异步方法,缺一不可。Python 在执行 async with obj: 时会按顺序 await 这两个方法,而不是普通同步的 __enter__/__exit__

常见错误是只实现了其中一个,导致运行时报错:AttributeError: __aexit__ not foundSyntaxError: 'async with' requires an object with __aenter__ and __aexit__

  • __aenter__ 应返回一个可 await 的对象(通常用 return selfreturn await something()
  • __aexit__ 必须带四个参数:(self, exc_type, exc_value, traceback),且返回值应为 bool 类型(用于抑制异常),或直接不写 return(等价于 return None
  • 不能在 __aenter____aexit__ 中使用 yield —— 那是异步生成器的写法,不是上下文管理器所需

如何正确释放异步资源(比如网络连接)

典型场景是封装一个需要 await 关闭的客户端,例如基于 aiohttp 或自定义 TCP 连接。关键点在于:清理逻辑本身也得是异步的,不能塞进同步的 __exit__

错误示范:def __exit__(self, ...): self._conn.close() —— 这里 close() 若是协程,不 await 就等于没调用;若它是同步方法,则可能漏掉底层异步关闭步骤(如等待 FIN-ACK)。

立即学习Python免费学习笔记(深入)”;

迷你天猫商城
迷你天猫商城

迷你天猫商城是一个基于Spring Boot的综合性B2C电商平台,需求设计主要参考天猫商城的购物流程:用户从注册开始,到完成登录,浏览商品,加入购物车,进行下单,确认收货,评价等一系列操作。 作为迷你天猫商城的核心组成部分之一,天猫数据管理后台包含商品管理,订单管理,类别管理,用户管理和交易额统计等模块,实现了对整个商城的一站式管理和维护。所有页面均兼容IE10及以上现代浏览器。部署方式1、项目

下载
  • 所有资源释放操作必须放在 __aexit__ 内,并显式 await
  • 建议在 __aexit__ 开头加 if self._closed: 判断,避免重复关闭
  • __aenter__ 中初始化失败(比如 await connect() 抛异常),__aexit__ 仍会被调用,此时需检查资源是否已创建,否则可能触发 AttributeError

与同步上下文管理器共存时的陷阱

一个类如果既想支持 with 又想支持 async with,不能简单地让 __aenter__ 调用 __enter__,反之亦然。因为它们的调用协议和生命周期不同。

更常见的需求其实是「兼容两种写法」,但实际中应避免这种设计:它容易掩盖阻塞风险(比如在 async with 块里误用同步 I/O)。

  • 不要在 __aenter__ 中直接 return self.__enter__() —— 返回的是普通对象,不是协程,async with 会报 TypeError: unawaitable
  • 若真要复用逻辑,把初始化/清理抽成独立的 async 方法(如 async def open(self) / async def close(self)),再由 __aenter__/__aexit__ 调用它们
  • 同步版本保持原样,不依赖异步方法;二者逻辑应隔离,避免交叉 await

测试异步上下文管理器是否生效

最直接的方式是写个最小 async 函数,用 async with 包裹后观察行为,重点验证两点:进入时是否完成初始化、退出时是否真正 await 了清理。

别只测“不崩溃”,要确认副作用发生。例如打开文件描述符后检查 os.listdir('/proc/self/fd')(Linux),或打日志看 __aexit__ 是否被 await 执行。

  • pytest + asyncio.run()@pytest.mark.asyncio 运行测试函数
  • __aexit__ 里故意 raise 异常,观察是否被外层捕获 —— 可验证返回值是否影响异常传播
  • 若管理器内部用了 asyncio.Lock 或类似资源,记得在测试前后检查锁状态,避免假阴性
异步上下文管理器的难点不在语法,而在对资源生命周期的精确控制:每个 await 都是调度点,每个未 await 的协程调用都是潜在的泄漏源。最容易被忽略的是 __aexit__ 中对 exc_type 的处理 —— 很多人直接忽略它,结果异常吞没后调试困难。

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

通义千问
通义千问

阿里巴巴推出的全能AI助手

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
if什么意思
if什么意思

if的意思是“如果”的条件。它是一个用于引导条件语句的关键词,用于根据特定条件的真假情况来执行不同的代码块。本专题提供if什么意思的相关文章,供大家免费阅读。

778

2023.08.22

磁盘配额是什么
磁盘配额是什么

磁盘配额是计算机中指定磁盘的储存限制,就是管理员可以为用户所能使用的磁盘空间进行配额限制,每一用户只能使用最大配额范围内的磁盘空间。php中文网为大家提供各种磁盘配额相关的内容,教程,供大家免费下载安装。

1415

2023.06.21

如何安装LINUX
如何安装LINUX

本站专题提供如何安装LINUX的相关教程文章,还有相关的下载、课程,大家可以免费体验。

706

2023.06.29

linux find
linux find

find是linux命令,它将档案系统内符合 expression 的档案列出来。可以指要档案的名称、类别、时间、大小、权限等不同资讯的组合,只有完全相符的才会被列出来。find根据下列规则判断 path 和 expression,在命令列上第一个 - ( ) , ! 之前的部分为 path,之后的是 expression。还有指DOS 命令 find,Excel 函数 find等。本站专题提供linux find相关教程文章,还有相关

295

2023.06.30

linux修改文件名
linux修改文件名

本专题为大家提供linux修改文件名相关的文章,这些文章可以帮助用户快速轻松地完成文件名的修改工作,大家可以免费体验。

782

2023.07.05

linux系统安装教程
linux系统安装教程

linux系统是一种可以免费使用,自由传播,多用户、多任务、多线程、多CPU的操作系统。本专题提供linux系统安装教程相关的文章,大家可以免费体验。

575

2023.07.06

linux查看文件夹大小
linux查看文件夹大小

Linux是一种自由和开放源码的类Unix操作系统,存在着许多不同的Linux版本,但它们都使用了Linux内核。Linux可安装在各种计算机硬件设备中,比如手机、平板电脑、路由器、视频游戏控制台、台式计算机、大型机和超级计算机。linux怎么查看文件夹大小呢?php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

550

2023.07.20

linux查看ip命令
linux查看ip命令

本专题为大家提供linux查看ip命令相关文章内容,感兴趣的朋友可以免费下载体验试试。

300

2023.07.20

java入门学习合集
java入门学习合集

本专题整合了java入门学习指南、初学者项目实战、入门到精通等等内容,阅读专题下面的文章了解更多详细学习方法。

1

2026.01.29

热门下载

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

精品课程

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

共48课时 | 8万人学习

Git 教程
Git 教程

共21课时 | 3.1万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号