Django支付系统中的并发更新：如何使用F()表达式避免投票数双倍增加

问题背景：投票计数异常增长

在开发基于Django的投票或打赏应用时，一个常见且令人困扰的问题是，用户完成支付后，相应的投票或积分计数可能出现异常增长。例如，预期增加10票，实际却增加了20票。这种“双倍增加”现象通常发生在处理支付回调或异步验证时，尤其是在高并发环境下，极易导致数据不一致。

原始代码中，在verify_payment函数内，更新用户投票总数的逻辑通常如下：

# 假设 profile_obj 已经获取
profile_obj.totalVote += payment.vote
profile_obj.save()

乍一看，这段代码似乎直接明了。然而，在多用户同时操作或支付回调机制设计不当的情况下，这种看似简单的操作极易引发数据不一致，导致投票数错误。

深入理解并发问题：竞态条件

上述代码段的问题根源在于“读-改-写”操作的非原子性。当多个请求几乎同时执行profile_obj.totalVote += payment.vote时，可能发生以下竞态条件：

1. 请求A 读取 profile_obj.totalVote 的当前值（例如 100）。
2. 请求B 几乎同时读取 profile_obj.totalVote 的当前值（也是 100）。
3. 请求A 在内存中计算新值 100 + payment.vote_A (假设 payment.vote_A 为 10)，得到 110。
4. 请求B 在内存中计算新值 100 + payment.vote_B (假设 payment.vote_B 为 5)，得到 105。
5. 请求A 将 110 写入数据库，profile_obj.totalVote 更新为 110。
6. 请求B 将 105 写入数据库，profile_obj.totalVote 被覆盖为 105。

最终结果是 105，而不是期望的 100 + 10 + 5 = 115。请求A的更新被请求B覆盖，导致数据丢失。虽然这里描述的是数据丢失，但如果每次更新都被“重复”执行（例如，由于支付回调被触发两次），则可能导致双倍增加。

Buildt.ai

AI驱动的软件开发平台，可以自动生成代码片段、代码分析及其他自动化任务

下载

解决方案：使用Django F() 表达式进行原子更新

为了解决这种并发问题，Django提供了F()表达式。F()表达式允许我们直接在数据库层面引用模型字段的值，并在数据库中完成操作，而不是先从数据库读取到Python内存，修改后再写回。这样，数据库可以确保操作的原子性，有效避免竞态条件。

将投票更新逻辑修改为使用F()表达式：

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

# ... 其他导入 ...

@transaction.atomic
def verify_payment(request, ref):
    print(f"verify_payment called for reference: {ref}")
    try:
        payment = Payment.objects.get(ref=ref)
    except Payment.DoesNotExist:
        print("Invalid payment reference:", ref)
        return render(
            request,
            "competitions/error.html",
            {"error_message": "Invalid payment reference."},
        )

    # 关键：检查支付是否已经验证过并处理，防止重复更新
    if payment.verified:
        print(f"Payment with reference {ref} already verified and processed. Skipping vote update.")
        return render(request, "competitions/success.html")

    # 调用 Payment 模型的方法进行外部支付网关验证
    verified = payment.verify_payment()

    if verified:
        contestant = payment.contestant
        profile_obj = contestant.user.profile

        # 使用 F() 表达式进行原子更新，直接在数据库层面操作
        # 这将确保 totalVote 的更新是原子性的，避免竞态条件
        profile_obj.totalVote = F('totalVote') + payment.vote
        profile_obj.save()

        # 如果需要立即获取更新后的值用于日志或其他操作，需要从数据库重新加载
        profile_obj.refresh_from_db()

        print(f"Payment verified for with total votes of {profile_obj.totalVote}.")

        return render(request, "competitions/success.html")
    else:
        return render(
            request,
            "competitions/error.html",
            {"error_message": "Payment verification failed."},
        )

代码解析：

• `from django.db.models