文章目录

• • 上下文管理器（Context Manager）
• 在Python中
• • contextmanager使用
  • 锁的acquire_timeout
  • sql事务的auto_commit()
• 在C++中
• • RAII
  • 另外的解法
  • 参考

上下文管理器（Context Manager）

上下文管理器是指

在一段代码执行之前执行一段代码，用于一些预处理工作。

在一段代码执行之后执行一段代码，用于做一些清理工作。

常用于对于一些需要申请/释放的资源来说，减少异常处理的代码

在Python中

一般与with搭配使用：

with EXPR as res:BLOCK

步骤：

• 执行EXPR语句，获取上下文管理器（Context Manager）
• 调用上下文管理器中的__enter__方法
• 将__enter__方法的结果返回给res
• 执行代码BLOCK
• 调用上下文管理器中的__exit__方法

这里注意一下：

__exit__方法有三个参数：exc_type, exc_val, exc_tb。

如果代码块BLOCK发生异常并退出，那么分别对应异常的type、value 和 traceback。否则三个参数全为None

contextmanager使用

在某func前加上@contextmanager 那么会将该func中yield语句之前的代码当做__enter__方法执行，yield语句之后的代码当做__exit__方法执行。同时yield返回值赋值给as后的变量

下面给出两个例子：

锁的acquire_timeout

对于锁的使用我们通常是这样的：

import threading
lock = threading.Lock()lock.acquire()
// do sth
lock.release()

如果要对try-lock设置timeout，并且在处理逻辑中有多出提前return的话，此时代码逻辑变得复杂起来，这样容易漏写release。

import threading
lock = threading.Lock()res = lock.acquire(blocking, timeout)
if res:try:// do sthlock.release()return 'yes'except BaseException as err:// do sth lock.release()return 'no'
else:// do other sthreturn 'no'

这可以将释放的操作放到__exit__()中操作，如下：

import threading
from contextlib import contextmanager@contextmanager
def acquire_timeout(lock, timeout):# __enter__方法result = lock.acquire(blocking=True, timeout=timeout)yield result# __exit__方法if result:lock.release()lock = threading.Lock()
with acquire_timeout(lock, 3) as result:if result:print("get lock")try:// do sthreturn 'yes'except BaseException as err:// do sth return 'no'else:print("not get lock")return 'no'

sql事务的auto_commit()

在我们使用 flask-sqlalchemy 的时候， 在提交事务的时候， 都需要 commit：

try:user = User()....db.session.commit()
except:db.session.rollback()

我们可以使用上下文管理器管理session的commit与rollback，这里将try-catch迁移出代码主逻辑，使得代码更加优雅。

from contextlib import contextmanager@contextmanager
def auto_commit(session):# __enter__方法try:yield# __exit__方法self.session.commit()  # 事务except BaseException as err:self.session.rollback() # 回滚raise errwith auto_commit(db.session):u = User()....db.session.add(u)

在C++中

RAII

C++中控制作用域内的资源的手法常见的有RAII，在构造函数中获取资源，在析构函数中释放资源。

以下是一个简单的RAII模板类的例子：

#include template
class RAII
{
public:RAII(T* t):m_t(t) {std::cout << "init" << std::endl;}~RAII() {std::cout << "del" << std::endl;delete m_t;}
private:T* m_t;
};int main()
{RAII raii(new int);// codereturn 0;
}

C:\Users\hanhandi\CLionProjects\untitled\cmake-build-debug\untitled.exe
init
del

另外的解法

https://www.zhihu.com/question/570807479/answer/2830724300

知乎上的这个回答更加贴近python中实现，在资源类中添加__enter 以及 __exit 函数。

借助于with模板函数在callback前后分别执行。

参考

https://zhuanlan.zhihu.com/p/24709718

https://blog.csdn.net/fenglepeng/article/details/114592795