Django中如何结合atomic原子事务使用post_save信号？

基础结论 #

• Django信号本身不具备原子事务管理能力，不存在使用信号就必须配置原子事务的强制要求，是否配置完全取决于你的业务一致性诉求
• post_save信号的触发时机是：对应模型实例的save操作数据库执行完成、但当前外层事务尚未提交的阶段，这时候的所有变更都还没有持久化到数据库

原生默认逻辑的缺陷 #

你当前直接调用User.objects.create()的写法，Django默认走自动提交模式，每一步数据库操作执行完就立刻提交，会存在数据不一致风险：

1. User创建提交成功，但信号内的数据库操作执行失败，二者状态不匹配
2. 信号内操作执行成功，但后续接口其他逻辑出错，User创建已提交无法回滚，关联数据残留

对应实现方案 #

方案1：外层统一包裹事务（推荐，适合90%以上场景） #

如果要求User创建和信号内操作必须同时成功/失败，直接在调用User创建的逻辑外层加transaction.atomic()块即可，二者会被纳入同一个事务，任意环节出错都会整体回滚：

from django.db import transaction

# 接口调用逻辑修改如下
with transaction.atomic():
    User.objects.create()

该方案不需要修改原有信号代码，侵入性最低。

方案2：信号内部独立配置事务 #

如果你希望信号内的操作和外层业务逻辑事务隔离，可以直接在信号函数内声明事务块，注意要传入using参数匹配模型使用的数据库连接，适配多库场景：

from django.db import transaction
from django.dispatch import receiver
from django.db.models.signals import post_save

@receiver(post_save, sender=User)
def some_signal(sender, instance, using, **kwargs):
    with transaction.atomic(using=using):
        # 原有数据库操作逻辑
        if instance:
            ...

方案3：事务提交后再执行信号逻辑（适用于非核心依赖场景） #

如果信号内的操作不需要和User创建强一致（比如发送通知、生成统计日志等），希望User确认创建成功后再执行信号逻辑，避免事务回滚导致的无效操作，可以结合transaction.on_commit使用：

@receiver(post_save, sender=User)
def some_signal(sender, instance, using, **kwargs):
    # 只有当前事务全部提交成功后，才会执行传入的函数
    transaction.on_commit(
        lambda: your_custom_db_operation(instance),
        using=using
    )

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