feat(payment): 在订单处理流程中新增Redis分布式锁机制防止并发问题

- 在定时任务同步订单状态时使用Redis锁,避免并发导致重复处理
- 同步回调接口中添加分布式锁,确保订单状态更新的原子性
- 主动查询订单支付状态接口增加Redis锁,防止重复发放积分
- 异步通知处理逻辑引入锁机制,处理锁定失败时适当重试或记录日志
- 捕获并区分锁定异常,避免错误日志泛滥,提升系统稳定性
- 保留数据库行级锁作为数据库层并发控制的保障措施
This commit is contained in:
公司git 2026-01-23 17:55:33 +08:00
parent 93d5c503b2
commit 05eba467b4
2 changed files with 134 additions and 99 deletions

10
app.py
View File

@ -39,11 +39,15 @@ def sync_pending_orders(app):
for order in pending_orders:
try:
# 使用防止并发的锁
lock_key = f"lock:order:{order.out_trade_no}"
# 尝试获取锁等待3秒。如果获取不到说明正在处理本次跳过
with redis_client.lock(lock_key, timeout=10, blocking_timeout=3):
# 查询订单状态
alipay_result = alipay_service.query_order_status(order.out_trade_no)
if alipay_result and alipay_result.get('trade_status') in ['TRADE_SUCCESS', 'TRADE_FINISHED']:
# 使用行级锁重新查询,防止并发问题
# 使用行级锁重新查询,防止并发问题 (数据库层面的双重保障)
order_locked = Order.query.filter_by(out_trade_no=order.out_trade_no).with_for_update().first()
# 二次校验状态,防止异步回调已经处理
@ -66,6 +70,10 @@ def sync_pending_orders(app):
# 订单已经被处理,跳过
logging.info(f"定时任务-订单 {order.out_trade_no} 已被处理,跳过")
except Exception as e:
# Redis lock error or other errors
if "LockError" in str(e) or "BlockingIOError" in str(e):
logging.info(f"定时任务-订单 {order.out_trade_no} 锁定失败或正在处理中")
else:
db.session.rollback()
logging.error(f"定时任务处理订单 {order.out_trade_no} 失败: {str(e)}")

View File

@ -1,5 +1,5 @@
from flask import Blueprint, request, redirect, url_for, session, jsonify, render_template
from extensions import db
from extensions import db, redis_client
from models import Order, User, to_bj_time, get_bj_now
from services.alipay_service import AlipayService
from services.logger import system_logger
@ -81,6 +81,9 @@ def payment_return():
trade_no = data.get('trade_no')
try:
# 使用分布式锁防止并发
lock_key = f"lock:order:{out_trade_no}"
with redis_client.lock(lock_key, timeout=10, blocking_timeout=3):
# 查询订单 (加锁防止并发导致双重发放)
order = Order.query.filter_by(out_trade_no=out_trade_no).with_for_update().first()
@ -105,6 +108,8 @@ def payment_return():
except Exception as e:
db.session.rollback()
# 忽略锁错误,说明可能已经在处理
if "LockError" not in str(e) and "BlockingIOError" not in str(e):
system_logger.error(f"同步回调-订单状态更新失败: {str(e)}")
return redirect(url_for('auth.buy_page', success='true', out_trade_no=out_trade_no))
@ -204,6 +209,9 @@ def api_sync_order():
# 如果支付宝显示已支付,更新本地订单状态
if trade_status in ['TRADE_SUCCESS', 'TRADE_FINISHED']:
try:
lock_key = f"lock:order:{out_trade_no}"
with redis_client.lock(lock_key, timeout=10, blocking_timeout=3):
# 使用行级锁重新查询,防止并发问题
order_locked = Order.query.filter_by(out_trade_no=out_trade_no).with_for_update().first()
@ -238,6 +246,13 @@ def api_sync_order():
'points': order_locked.points,
'paid_at': order_locked.paid_at_bj.strftime('%Y-%m-%d %H:%M:%S') if order_locked.paid_at else None
})
except Exception as e:
if "LockError" in str(e) or "BlockingIOError" in str(e):
# 如果获取锁失败,可能是因为正在处理中,返回成功状态(前端会重试或刷新)
return jsonify({'code': 200, 'msg': '处理中', 'status': 'PAID'})
else:
db.session.rollback()
raise e
# 支付宝显示未支付
elif trade_status in ['TRADE_CLOSED', 'WAIT_BUYER_PAY']:
@ -282,6 +297,9 @@ def payment_notify():
trade_no = data.get('trade_no')
# 加锁查询,确保并发安全
try:
lock_key = f"lock:order:{out_trade_no}"
with redis_client.lock(lock_key, timeout=10, blocking_timeout=3):
order = Order.query.filter_by(out_trade_no=out_trade_no).with_for_update().first()
if order and order.status == 'PENDING':
order.status = 'PAID'
@ -299,6 +317,15 @@ def payment_notify():
return "success"
else:
return "fail"
except Exception as e:
# 如果锁获取失败暂时返回fail让支付宝重试或者返回success如果确信正在处理
# 返回fail让支付宝重试比较稳妥因为可能正在处理但还没提交
system_logger.warning(f"处理异步通知锁定失败: {str(e)}")
if "LockError" in str(e) or "BlockingIOError" in str(e):
# 正在处理中,告诉支付宝由于并发我们正在处理,稍后重试(或视为成功?)
# 如果返回fail支付宝会重试。
return "fail"
raise e
else:
return "fail"