当前位置：首页 > news >正文

Django transaction.atomic()事务处理

news 来源：原创 2024/9/20 5:33:09

在Django中，transaction.atomic()是一个上下文管理器，它会自动开始一个事务，并在代码块执行完毕后提交事务。如果在代码块中抛出异常，事务将被自动回滚，确保数据库的一致性和完整性。

在实际应用中，你可能需要根据具体的业务逻辑来决定何时抛出异常，以触发事务回滚。例如，如果某个操作依赖于另一个操作的成功，那么在第一个操作失败时，你可能需要抛出异常，以防止后续操作被执行，从而保持数据的一致性。

1，添加模型

Test/app11/models.py

from django.db import modelsclass Post(models.Model):title = models.CharField(max_length=200)content = models.TextField()pub_date = models.DateTimeField('date published')class Book(models.Model):title = models.CharField(max_length=100)author = models.CharField(max_length=100)publication_date = models.DateField()price = models.DecimalField(max_digits=5, decimal_places=2)def __str__(self):return self.title

2，添加视图函数

Test/app11/views.py

2.1 装饰器写法


from django.http import HttpResponse
from django.db import transaction
from .models import Post, Book
from django.utils import timezonedef atomic_transaction(func):def wrapper(*args, **kwargs):with transaction.atomic():try:return func(*args, **kwargs)except Exception as e:# 在这里可以添加错误日志记录或其他错误处理逻辑print(f"An error occurred: {e}")# 由于在transaction.atomic()上下文中，异常将自动触发事务回滚return wrapper@atomic_transaction
def create_post_and_book(request):post_title = "New Post4"book_title = "New Book4"post = Post(title=post_title, content="This is the content of the new post.", pub_date=timezone.now())post.save()book = Book(title=book_title, author="Author Name", publication_date=timezone.now().date(), price=29.99)book.save()# 故意引发异常以测试回滚# if True:#     raise Exception("An error occurred")print("Post and Book created successfully.")return HttpResponse("Post and Book created successfully.")

2.2 with写法

from django.http import HttpResponse
from django.db import transaction
from .models import Post, Book
from django.utils import timezonedef create_post_and_book(request):post_title = "New Post3"book_title = "New Book3"with transaction.atomic():post = Post(title=post_title, content="This is the content of the new post.", pub_date=timezone.now())post.save()# 故意引发异常以测试回滚if True:raise Exception("An error occurred")book = Book(title=book_title, author="Author Name", publication_date=timezone.now().date(), price=29.99)book.save()return HttpResponse("Post and Book created successfully.")

3，添加路由应用

Test/app11/urls.py

from django.urls import path
from . import viewsurlpatterns = [path('create_post_and_book/', views.create_post_and_book, name='create_post_and_book'),
]

4，访问页面

http://127.0.0.1:8000/app11/create_post_and_book/

当你的代码中抛出异常，如raise Exception("An error occurred")，这会中断程序的正常执行流程。在Django的事务处理中，如果在transaction.atomic()的上下文中抛出异常，Django会捕获这个异常并自动回滚事务，确保任何在事务开始后所做的数据库更改不会被永久保存。

可以看到数据并没有被保存到

在这个例子中，由于if True:条件总是为真，因此raise Exception("An error occurred")这一行代码将总是抛出异常。结果，transaction.atomic()会捕获这个异常并回滚事务，这意味着Post和Book对象都不会被保存到数据库中，因为事务没有成功提交。这是Django事务处理的一个关键特性，它确保了数据的一致性和完整性，避免了部分完成的事务可能导致的数据库状态不一致问题。