关于transaction.atomic

Django 的transaction管理提供了一个强大的机制来处理数据库事务。在本文中，我们将探讨transaction.atomic特性，以及它如何确保 Django 应用程序中的数据一致性和完整性。

什么是transaction

在深入研究transaction.atomic之前，让我们先了解一下transaction的概念。在数据库上下文中，transaction表示一个逻辑工作单元，该工作单元要么整体成功，要么完全失败，从而确保数据一致性。transaction通常由多个数据库操作组成，例如插入、更新和删除。

什么是atomic

Django 提供了装饰器，它确保数据库操作块作为transaction.atomic执行。transaction.atomic保证区块内的所有操作都被视为一个单元。如果任何操作失败，则整个transaction将回滚，撤消区块内所做的所有更改。

例子：

下面是一个示例，说明如何使用 Django 的transaction.atomic功能对模型执行批量插入：Product

from django.db import transaction, models

def transfer(self, request):
    try:
        user_a = request.POST.get("user_a")
        user_b = request.POST.get("user_b")
        amount = request.POST.get("amount")

        with transaction.atomic():
            user_a_obj = Account.objects.select_for_update().get(user=user_a)
            user_a_obj.balance -= int(amount)
            user_a_obj.save()

            user_b_obj = Account.objects.select_for_update().get(user=user_b)
            user_b_obj.balance += int(amount)
            user_b_obj.save()

            return Response(
                {"status": "success", "message": "Your amount is transfered."}
            )

    except Exception as e:
        print(e)
        return Response({"status": "failed", "message": "Something went wrong."})
The method is called on the querysets for and . This method locks the selected rows in the database, preventing other transactions from modifying them until the current transaction is completed.select_for_update()user_one_objuser_two_obj

Bulk insert using Django’s transaction atomic
Here’s an example of how to perform a bulk insert using Django’s transaction atomic feature with the model:Product

from django.db import transaction

# Assume you have a list of products to insert
products_data = [
    {'name': 'Product 1', 'sku': 'SKU1', 'price': 10.99},
    {'name': 'Product 2', 'sku': 'SKU2', 'price': 19.99},
    {'name': 'Product 3', 'sku': 'SKU3', 'price': 14.99},
    # Add more products as needed
]

@transaction.atomic
def create_products(products_data):
    # Create a list to hold the Product objects
    products = []

    try:
        # Iterate over the products_data list
        for data in products_data:
            product = Product(name=data['name'], sku=data['sku'], price=data['price'])
            products.append(product)

        # Use the bulk_create method to insert the products in a single query
        Product.objects.bulk_create(products)

        # Transaction will be committed automatically if no exceptions occur
    except Exception as e:
        # Handle any exceptions that occur during the bulk creation process
        print(f"Error occurred: {e}")
        # Raise an exception to trigger a rollback

在这里，我们用装饰器装饰函数，以确保将整个批量插入操作视为transaction.atomic

在遍历所有数据后，我们使用 manager 的方法将所有对象插入到单个查询中，与单个调用相比，bulk_create性能有所提高。

如果在批量创建过程中发生任何异常，我们会捕获它并相应地处理它。通过引发异常，transaction将被回滚，并且数据库中不会保留任何更改。如果没有异常发生，transaction将自动提交，所有产品都将插入到数据库中。