在开发涉及支付和投票功能的 django 应用时,开发者可能会遇到一个常见且棘手的问题:用户完成支付后,对应的投票计数却增加了两倍于预期值。例如,如果用户购买了10票,系统最终记录的票数却是20票。这种异常通常发生在支付回调或重定向处理不当,以及数据库更新操作缺乏原子性保障的情况下。
具体到提供的代码,verify_payment 视图函数负责处理支付验证和票数更新。当支付网关(如 Paystack)成功处理交易后,它可能会向应用发送一个回调请求,或者用户被重定向回应用的 verify_payment 页面。如果这些事件被触发多次,而 totalVote 的更新逻辑没有严格的幂等性检查,就可能导致票数被重复增加。
幂等性缺失导致的双重计数: verify_payment 函数中的 payment.verify_payment() 方法会在验证成功后将 payment.verified 字段设置为 True 并保存。然而,在 verify_payment 视图中,profile_obj.totalVote += payment.vote 这行代码是在 payment.verify_payment() 返回 True 之后执行的,但并没有在执行票数增加前再次检查 payment.verified 状态。这意味着,如果 verify_payment 视图被多次调用(例如,由于网络延迟用户多次刷新页面,或支付网关多次发送回调),即使 payment.verify_payment() 方法由于 if not self.verified: 检查只执行一次外部 API 调用,但视图中 verified 变量为 True 的逻辑路径依然可能被多次执行,从而导致 totalVote 字段被重复更新。
竞态条件 (Race Condition): 即使解决了双重触发的问题,直接使用 profile_obj.totalVote += payment.vote 这样的 Python 层面操作仍然存在竞态条件风险。这种操作通常分为三个步骤:
为了彻底解决上述问题,我们需要采取以下策略:
Django 的 F() 表达式允许我们在不实际将数据库值取出到 Python 内存的情况下,直接在数据库层面引用模型字段的值进行操作。这使得数据库能够以原子方式执行更新,从而避免竞态条件。
例如,将 profile_obj.totalVote += payment.vote 改为:
from django.db.models import F # ... profile_obj.totalVote = F("totalVote") + payment.vote profile_obj.save(update_fields=['totalVote'])
这告诉数据库直接将 totalVote 字段的当前值加上 payment.vote 的值,并将结果写回 totalVote 字段。整个操作在数据库内部完成,是原子性的。
结合幂等性检查和 F() 表达式,我们可以对 verify_payment 视图进行如下重构:
from django.db import transaction from django.db.models import F from django.shortcuts import render, get_object_or_404 from django.conf import settings # 假设 Paystack, Payment, Contestant 等模型和类已正确导入 @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 {ref} already verified and votes processed. Skipping duplicate update.") # 可以记录一个警告日志,表示收到重复的回调/请求 return render(request, "competitions/success.html") # 调用 Payment 模型的 verify_payment 方法进行外部支付网关验证 # 此方法成功时会设置 payment.verified = True 并保存 verified_with_paystack = payment.verify_payment() if verified_with_paystack: contestant = payment.contestant profile_obj = contestant.user.profile # 使用 F() 表达式进行原子性更新,避免竞态条件 # 注意:这里我们只更新 totalVote 字段,以避免不必要的副作用 profile_obj.totalVote = F("totalVote") + payment.vote profile_obj.save(update_fields=['totalVote']) # 刷新 profile_obj 以获取数据库中最新的 totalVote 值,用于日志输出 profile_obj.refresh_from_db() print(f"Payment verified for {contestant.user.first_name}. New total votes: {profile_obj.totalVote}.") return render(request, "competitions/success.html") else: print(f"Payment verification failed for reference: {ref}") return render( request, "competitions/error.html", {"error_message": "Payment verification failed."}, )
Payment 模型中的 verify_payment 方法设计良好,它内部已经包含了对 self.verified 的检查,确保不会重复调用外部支付网关 API。
class Payment(models.Model): # ... 其他字段 def save(self, *args, **kwargs): while not self.ref: ref = secrets.token_urlsafe(50) object_with_similar_ref = Payment.objects.filter(ref=ref) if not object_with_similar_ref: self.ref = ref super().save(*args, **kwargs) def amount_value(self): return int(self.amount) * 100 def verify_payment(self): # 确保只在未验证时才尝试与支付网关交互 if not self.verified: paystack = Paystack() # 假设 Paystack 类已正确定义和导入 status, result = paystack.verify_payment(self.ref, self.amount) if status: # 再次确认金额匹配,防止篡改 if result["amount"] / 100 == self.amount: self.verified = True self.save() # 保存验证状态 return self.verified
通过上述改进,您的 Django 投票系统将能够更安全、更准确地处理支付和投票计数,有效避免双重增加和竞态条件导致的数据不一致问题。
以上就是优化 Django 投票系统:避免支付后票数重复增加及竞态条件的详细内容,更多请关注php中文网其它相关文章!
每个人都需要一台速度更快、更稳定的 PC。随着时间的推移,垃圾文件、旧注册表数据和不必要的后台进程会占用资源并降低性能。幸运的是,许多工具可以让 Windows 保持平稳运行。
Copyright 2014-2025 //m.sbmmt.com/ All Rights Reserved | php.cn | 湘ICP备2023035733号