Django 和 Postgres 中为 select_for_update 生成的查询顺序的差异

使用Django和PostgreSQL时，我们经常需要进行并发控制，以防止多个用户同时修改同一行数据。在这种情况下，我们可以使用Django的select_for_update方法，该方法会在查询期间对所选行进行锁定，以确保其他事务无法修改这些行。然而，有一点需要注意的是，在Django和PostgreSQL中，生成的查询顺序可能会有所不同，这可能会对并发控制产生影响。

python
from django.db import models
class Product(models.Model):
    name = models.CharField(max_length=100)
    price = models.DecimalField(max_digits=10, decimal_places=2)
    quantity = models.IntegerField()

python
from django.db import transaction
@transaction.atomic
def purchase_product(user_id, product_id):
    product = Product.objects.select_for_update().get(id=product_id)
    if product.quantity > 0:
        product.quantity -= 1
        product.save()
        print(f"User {user_id} purchased the product successfully.")
    else:
        print(f"User {user_id} failed to purchase the product.")
purchase_product(1, 1)
purchase_product(2, 1)

python
from django.db import transaction, OperationalError
@transaction.atomic
def purchase_product(user_id, product_id):
    try:
        product = Product.objects.select_for_update(nowait=True).get(id=product_id)
        if product.quantity > 0:
            product.quantity -= 1
            product.save()
            print(f"User {user_id} purchased the product successfully.")
        else:
            print(f"User {user_id} failed to purchase the product.")
    except OperationalError:
        print(f"User {user_id} failed to purchase the product.")
purchase_product(1, 1)
purchase_product(2, 1)