[Django学习]如何在Django中声明“事务”(包括事务回滚)

零叁三2024-06-23 17:22

一、事务概念

事务(transaction)是访问并可能操作各种数据项的一个数据库操作序列，这些操作要么全部执行,要么全部不执行，是一个不可分割的工作单位。比如某软件的转账功能：现在用户A要给B转账300块，其流程为，先检查A的余额，若余额大于300，则将A的余额减去300，之后将B的余额增加300。

在上述例子中，A余额减300，和B余额增加300这两个操作要么都执行，要么一个也不执行，否则可能会出现A的余额减300后，B的余额没有变化，或是B的余额凭空增加300，但A的余额还是不变。转账这一系列操作为一个事务。

但若一开始发现A的余额不足300块，则对事务进行回滚，取消之后的一系列操作。

二、Django声明事务的两种方法

1.装饰器方法：

使用@transaction.atomic装饰器来装饰一个视图函数或方法。在这个函数或方法内部的所有数据库操作都将在一个事务中执行。如下面这个代码:

python 复制代码

from django.db import transaction  
from .models import Account  
  
@transaction.atomic  
def transfer_funds(from_account_id, to_account_id, amount):  
    try:  
        # 从A账户中减去金额  
        from_account = Account.objects.get(pk=from_account_id)  
        if from_account.balance < amount:  # 余额不足,回滚
            raise ValueError("账户余额不足")  
        from_account.balance -= amount  
        from_account.save()  
  
        # 向账户B中增加金额  
        to_account = Account.objects.get(pk=to_account_id)  
        to_account.balance += amount  
        to_account.save()  
  
    except Exception as e:  
        # 如果在此过程中发生任何异常，Django会自动回滚事务  
        print(f"An error occurred: {e}")    

    # 代码块成功执行完毕，Django会自动提交事务

2.上下文管理器：

也可以不把整个视图作为事务处理，而是在一个代码块里使用transaction.atomic()上下文管理器。

python 复制代码

from django.db import transaction  
from .models import Account  
  
def transfer_funds(from_account_id, to_account_id, amount):  
    try:  
        with transaction.atomic():  
            #  这里放刚刚转账操作的代码  
    except Exception as e:  
        print(f"操作发生错误: {e}")