Django外键关联模型基于订单项的库存更新实现咨询

现有代码的核心疏漏 #

• 仅在内存中修改item.product的stock属性，从未调用Product的save方法，修改后的库存值不会同步到数据库，扣减逻辑实际完全不生效
• get_or_create方法返回(实例对象, 是否创建标记)的二元组，若直接将返回值赋值给order，后续查询订单项时会传入元组作为过滤条件，直接触发查询类型错误
• 循环内每次访问item.product会触发独立的数据库查询，存在N+1性能问题
• 在Python内存层计算库存差值再写回数据库，高并发场景下会出现并发覆盖导致的库存超卖
• 未处理OrderItem.product的空值逻辑（字段本身设置了允许为null），遇到关联商品被删除的情况会触发空对象属性报错
• 未做库存边界校验，扣减后库存为负时会触发PositiveIntegerField的数据库写入异常
• 订单项更新、库存扣减操作未包裹事务，中途报错会出现数据不一致的脏状态

可直接运行的优化代码 #

from django.db.models import F
from django.db import transaction

# 正确解包get_or_create的返回值
order, is_created = Order.objects.get_or_create(customer=customer, complete=False)
# 提前关联查询Product对象，消除循环内的重复查询
order_items = OrderItem.objects.filter(order=order).select_related("product")

# 开启原子事务，保证所有操作要么全部成功，要么全部回滚
with transaction.atomic():
    for item in order_items:
        # 跳过无关联商品的订单项，规避空指针报错
        if not item.product:
            continue
        # 数据库层面执行库存扣减，同时校验库存是否充足
        update_count = Product.objects.filter(
            pk=item.product_id,
            stock__gte=item.quantity
        ).update(
            stock=F("stock") - item.quantity
        )
        # 库存不足时抛出业务异常，触发事务回滚
        if update_count == 0:
            raise ValueError(f"商品{item.product.title}库存不足")
        # 更新订单项的交易号
        item.transaction_id = transaction_id
        item.save()

关键设计说明 #

• 使用F表达式直接在数据库层完成库存减法，无需将库存值读到内存计算，从根源规避并发场景下的超卖问题
• 在更新查询中增加stock__gte=item.quantity的条件，仅当剩余库存大于等于购买数量时才执行扣减，天然避免库存写入负数
• 对订单、订单项、商品库存的全流程操作用事务包裹，不会出现“库存扣减成功但订单项交易号未更新”的脏数据
• 对允许为空的product字段做前置判断，兼容外键SET_NULL的删除逻辑
• 高并发场景下可在事务内对关联的Product行增加行锁（使用select_for_update()），进一步降低极端并发下的更新冲突概率

内容的提问来源于stack exchange，提问作者newDev2000