Python with提前退出：坑与处理方案

发布时间：2022-07-16 15:36:24 所属栏目：云计算来源：互联网

导读：问题的起源早些时候使用with实现了一版全局进程锁，希望实现以下效果：全局进程锁本身不用多说，大部分都依靠外部的缓存来实现的，redis上用的是setnx，有时候根据需要加上缓存击穿问题、随机延后以防止对缓存本身造成压力。当时同样写了单元测试来测试这

　　问题的起源
　　早些时候使用with实现了一版全局进程锁，希望实现以下效果：

　　全局进程锁本身不用多说，大部分都依靠外部的缓存来实现的，redis上用的是setnx，有时候根据需要加上缓存击穿问题、随机延后以防止对缓存本身造成压力。

　　当时同样写了单元测试来测试这段代码的有效性：

　　看起来非常完美地通过了。

　　这样的一个全局进程锁是通过__enter__方法抛出异常, __exit__方法中捕获异常来实现的:

　　看起来还不错，毕竟单元测试都过了。

　　但是，这样的实现是有问题的：

　　原因在于__exit__ 的执行不是包在__enter__ 之外的，因此__enter__抛出的异常，不会被__exit__捕获。

　　上面的单元测试恰好通过，是因为其中有两个with语句，外面的with 捕获的其实是里面的__enter__ 抛出的异常

　　使用改进后的单元测试：

　　第一种解决方案
　　既然想明白了with的执行顺序，那么第一种解决方案就呼之欲出了：既然__exit__捕获的异常在__enter__执行完成之后，那么我们提供一个函数确认一下就可以了，把ABContext实现改成这样：

　　使用的时候：

　　但这样的解决方法并不优雅，万一使用这个ABContext的时候忘记用ensure方法了，那么就等于完全没用这个Context方法，太容易失误了，而且代码也失去了Pythonic的性质。

　　第二种解决方法
　　翻了一下contextlib的标准库文档，发现有一个已经废弃的函数：contextlib.nested

　　可以执行多个上下文：

　　这个废弃的特性在Python2.7之后，可以直接由with关键字执行，形如：

　　这个特性还不错，根据__enter__的执行顺序的话，那么我们可以实现一个由第一个 context的__exit__来捕获，第二个context的__enter__来抛出异常，

　　如同这样：

　　结合前面我们实现的ABContext的使用是这样的：

　　good，单元测试就这样过了！

　　能不能再给力点？

　　确实，在with里要写俩context有点蛋疼，并不是特别优雅，能不能还是回到最初的那种用法：我们只用写一条context，这一个context做到了两个context的事情？

　　TIL
　　总之学到了contextlib里的一些有用的函数和装饰器，也第一次发现with可以放个context。

　　虽然放多个context的动态构造还有待研究，with 后面的代码块也不能填一个元组或者列表。。惆怅。。

（编辑：PHP编程网 - 黄冈站长网）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!

PaaS市场高速增长如何	一文搞懂基础设施即代
云计算核心技术Docker	云计算核心技术Docker