《Python 上下文管理器实战：用优雅守护你的数据库事务》

《Python 上下文管理器实战：用优雅守护你的数据库事务》

一、引言：那些被遗忘的“关闭”

在我职业生涯的早期，有一次凌晨排查线上数据库死锁问题，发现罪魁祸首竟是一个未关闭的事务连接。那段代码没有处理异常，也没有释放资源，导致连接池耗尽，整个服务雪崩。

这类问题并不罕见。数据库连接、文件句柄、网络套接字……它们都需要被“善后”。而 Python 提供的上下文管理器机制，正是我们实现资源安全释放的利器。

本文将带你深入理解上下文管理器的设计理念，并用两种方式实现一个自动处理数据库事务的上下文管理器：一种是通过类实现，另一种是使用 @contextmanager 装饰器。我们将探讨它们的优劣，并结合实战案例，帮助你在项目中优雅地管理资源。

---

二、背景介绍：Python 的“承诺机制”

上下文管理器（Context Manager）是 Python 中用于管理资源生命周期的协议。它的核心目标是：

• 在资源使用前执行初始化逻辑（如打开文件、建立连接）。
• 在资源使用后自动清理（如关闭文件、释放连接），即使发生异常也能保证清理。

语法形式：

with resource as obj:
    # 使用 obj
# 自动清理

常见应用场景：

• 文件操作：with open(...) as f
• 数据库连接：with conn.cursor() as cur
• 网络连接：with socket.create_connection(...) as s
• 线程锁：with threading.Lock()

---

三、基础原理：上下文管理器的协议

类方式实现：

实现 __enter__() 和 __exit__() 方法即可：

class MyContext:
    def __enter__(self):
        print("资源初始化")
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        print("资源清理")

装饰器方式实现：

使用 contextlib.contextmanager 装饰器，将一个生成器函数变成上下文管理器：

from contextlib import contextmanager

@contextmanager
def my_context():
    print("资源初始化")
    yield
    print("资源清理")

---

四、实战：数据库事务上下文管理器

我们现在来实现一个自动处理事务的上下文管理器，目标如下：

• 成功执行则自动 commit
• 发生异常则自动 rollback
• 无需手动关闭连接

✅ 假设数据库连接对象：