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

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

本文探讨了django应用中支付后投票计数出现双重增加的常见问题,深入分析了其背后的并发竞争条件。我们将详细介绍如何利用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覆盖,导致数据丢失。虽然这里描述的是数据丢失,但如果每次更新都被“重复”执行(例如,由于支付回调被触发两次),则可能导致双倍增加。

解决方案:使用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

© 版权声明
THE END
喜欢就支持一下吧
点赞14 分享