Django：如何用独立事务/连接保存异常数据避免回滚

你的痛点我完全理解——那种外层嵌套两层try的写法不仅臃肿，还破坏了代码的可读性。其实我们可以自己封装一个类似你设想的transaction.separate_connection()的上下文管理器，让异常数据的保存操作脱离主事务的控制，就算主事务回滚也不会丢失。

问题回顾 #

你原有的代码因为异常保存操作在主事务内，所以抛出异常后主事务回滚会把异常数据也一起回滚：

def some_function():
    with transaction.atomic():
        # some code which writes to DB
        try:
            call_method_which_may_raise_exception()
        except KnownException as e:
            save_some_data_about_exception_to_database(e)
            raise
        # other code which writes to DB

而你想要的优雅写法核心是让异常保存操作在独立事务中执行，不受主事务影响。

实现独立事务上下文管理器 #

我们可以用Django的数据库连接API封装一个自定义上下文管理器：

from contextlib import contextmanager
from django.db import connections, transaction

@contextmanager
def separate_transaction(using='default'):
    # 获取独立的数据库连接（连接池会自动管理空闲连接）
    conn = connections[using]
    # 在新连接上开启原子事务
    with conn.cursor():
        with transaction.atomic(using=using):
            yield

优雅使用你的目标写法 #

现在你就可以像设想的那样调用了，代码简洁且逻辑清晰：

def some_function():
    with transaction.atomic():
        # some code which writes to DB
        try:
            call_method_which_may_raise_exception()
        except KnownException as e:
            with separate_transaction():
                save_some_data_about_exception_to_database(e)
            raise
        # other code which writes to DB

原理说明 #

这个上下文管理器的核心是使用独立的数据库连接执行异常保存操作：

• Django默认会为每个线程维护一个数据库连接，主事务的所有操作都在这个连接上执行
• 我们通过connections[using]获取新的连接（或复用连接池中的空闲连接），并在这个连接上开启独立事务
• 主事务的回滚只会影响自身连接内的操作，独立连接中的事务提交后数据会被永久保存，不会被回滚

注意事项 #

• 如果你的项目使用多数据库，记得在separate_transaction中指定对应的using参数（比如separate_transaction(using='logs')）
• 注意数据库连接池的配置，避免因频繁创建连接导致连接耗尽
• 确保save_some_data_about_exception_to_database函数没有绑定到特定连接，上下文管理器会自动切换当前连接

内容的提问来源于stack exchange，提问作者Alexey Ruzin