Database django 中的比赛条件

发布于06月12日

下面是一个具有潜在竞争条件的Django视图的简单示例:

# myapp/views.py
from django.contrib.auth.models import User
from my_libs import calculate_points

def add_points(request):
    user = request.user
    user.points += calculate_points(user)
    user.save()

竞争条件应该相当明显:用户可以发出两次请求，应用程序可能同时执行user = request.user次，导致其中一个请求覆盖另一个请求.

假设函数calculate_points相对复杂，并且基于各种各样的奇怪的东西进行计算，这些东西不能放在一个update中，也很难放在存储过程中.

所以我的问题是:django可以使用什么样的锁定机制来处理类似的情况？

推荐答案

Django 1.4+支持select_for_update，在早期版本中，您可以执行原始SQL查询，例如select ... for update，这取决于底层数据库是否会锁定该行，不能进行任何更新，您可以对该行做任何您想做的事情，直到事务结束.例如:

from django.db import transaction

@transaction.commit_manually()
def add_points(request):
    user = User.objects.select_for_update().get(id=request.user.id)
    # you can go back at this point if something is not right 
    if user.points > 1000:
        # too many points
        return
    user.points += calculate_points(user)
    user.save()
    transaction.commit()