Django `transaction.atomic()` 完整使用指南

目录

  1. #概述
  2. #基本用法
  3. #事务一致性保障机制
  4. #破坏一致性的常见场景
  5. #高级用法
  6. #最佳实践
  7. #诊断与调试
  8. #附录

概述

transaction.atomic() 是 Django 提供的数据库事务管理工具,用于确保一系列数据库操作要么全部成功提交,要么全部回滚,维护数据的一致性。

基本用法

1. 作为上下文管理器

python 复制代码
from django.db import transaction

def view_func(request):
    with transaction.atomic():
        # 事务内的操作
        obj1.save()
        obj2.save()

2. 作为装饰器

python 复制代码
@transaction.atomic
def view_func(request):
    # 整个函数都在事务中执行
    obj1.save()
    obj2.save()

事务一致性保障机制

  • 进入块时:创建保存点或开始新事务
  • 成功退出:提交所有更改
  • 异常退出:自动回滚所有更改
  • 嵌套处理:内层块作为保存点处理

破坏一致性的常见场景

1. 异常处理不当

python 复制代码
with transaction.atomic():
    try:
        risky_operation()
    except Exception:
        logger.error("Error occurred")  # 不重新抛出异常 → 破坏一致性

修复方案

python 复制代码
with transaction.atomic():
    try:
        risky_operation()
    except Exception as e:
        logger.error("Error occurred")
        raise  # 重新抛出异常

2. 长时间运行事务

python 复制代码
with transaction.atomic():
    process_data()  # 耗时操作
    update_database()  # 可能因锁超时失败

3. 混合非数据库操作

python 复制代码
with transaction.atomic():
    save_to_db()
    call_external_api()  # 网络操作
    save_to_db_again()   # 外部API失败可能导致不一致

4. 多数据库操作

python 复制代码
with transaction.atomic():  # 只针对default数据库
    ModelA.objects.create(...)  # default
    ModelB.objects.using('other_db').create(...)  # 不受保护

5. 隔离级别冲突

python 复制代码
# 事务1
with transaction.atomic():
    item = Item.objects.select_for_update().get(pk=1)
    time.sleep(10)

# 事务2 (并发执行)
with transaction.atomic():
    item = Item.objects.get(pk=1)  # 可能被阻塞或读取过期数据

高级用法

1. 手动保存点控制

python 复制代码
with transaction.atomic():
    a.save()
    sid = transaction.savepoint()  # 创建保存点
    
    try:
        b.save()
    except Exception:
        transaction.savepoint_rollback(sid)  # 回滚到保存点
        # 可继续其他操作
    
    transaction.savepoint_commit(sid)  # 释放保存点

2. 指定数据库

python 复制代码
with transaction.atomic(using='replica_db'):
    ReplicaModel.objects.create(...)

3. 嵌套事务控制

python 复制代码
with transaction.atomic():  # 外层事务
    a.save()
    
    try:
        with transaction.atomic():  # 内层保存点
            b.save()
            raise ValueError
    except ValueError:
        pass  # 仅回滚内层
    
    c.save()  # 仍会提交

最佳实践

  1. 保持事务简短:只包含必要的数据库操作

  2. 明确异常处理

    • 要么完全处理并确保一致性
    • 要么重新抛出异常
  3. 避免混合操作

    • 不在事务中执行网络/文件IO
    • 不在事务中执行耗时计算
  4. 多数据库处理

    python 复制代码
    with transaction.atomic(using='default'):
        with transaction.atomic(using='other_db'):
            # 跨数据库操作
  5. 测试隔离级别:了解数据库的默认隔离级别

  6. 监控长事务:设置数据库长事务警报

诊断与调试

1. 检查事务状态

python 复制代码
from django.db import connection

print(f"In atomic block: {connection.in_atomic_block}")
print(f"Needs rollback: {connection.needs_rollback}")

2. 检查隔离级别(PostgreSQL)

python 复制代码
with connection.cursor() as cursor:
    cursor.execute("SHOW transaction_isolation")
    print(cursor.fetchone())

3. 日志记录

python 复制代码
LOGGING = {
    'loggers': {
        'django.db.backends': {
            'level': 'DEBUG',
            'handlers': ['console'],
        }
    }
}

附录

支持的数据库后端

数据库 支持级别 注意事项
PostgreSQL 完全支持 建议使用
MySQL/MariaDB 支持 需使用InnoDB
SQLite 支持 实际是文件锁
Oracle 完全支持 -
SQL Server 支持 需验证隔离级别

事务隔离级别参考

级别 脏读 不可重复读 幻读 说明
读未提交 可能 可能 可能 最低隔离
读已提交 不可能 可能 可能 多数数据库默认
可重复读 不可能 不可能 可能 MySQL默认
串行化 不可能 不可能 不可能 最高隔离
相关推荐
熊文豪20 分钟前
MySQL数据库迁移到KingbaseES完整指南
数据库·mysql·kingbasees·金仓数据库·kingbasees迁移指南
CodeCraft Studio28 分钟前
国产化Excel开发组件Spire.XLS教程:Python 将 CSV 转换为 Excel(.XLSX)
windows·python·excel·csv转excel·spire·excel开发
大神薯条老师28 分钟前
Python从入门到高手9.4节-基于字典树的敏感词识别算法
爬虫·python·深度学习·机器学习·数据分析
一袋米扛几楼9833 分钟前
【物联网】BLE 系统架构全景图
python·物联网·aws
TDengine (老段)38 分钟前
工业数据消费迎来“抖音式”革命:TDengine IDMP 让数据自己开口说话
大数据·数据库·物联网·ai·时序数据库·iot·tdengine
hvinsion40 分钟前
【开源工具】基于Flask与Socket.IO的跨平台屏幕监控系统实战(附完整源码)
javascript·python·flask·开源·socket.io·屏幕监控·web实战
李昊哲小课1 小时前
「从 0 到 1」的 Python-requests 爬虫完整教程
爬虫·python
IDOlaoluo1 小时前
PLSQL Developer 12.0.1 x64 安装步骤详解(附Oracle连接设置|附安装包下载)
数据库·oracle
追逐时光者2 小时前
一款为程序员和运维人员量身打造的一站式开发运维利器
数据库·docker·ssh
往日情怀酿做酒 V17639296382 小时前
SQL注入6----(其他注入手法)
数据库·sql