Tidb + 分布式锁实现幂等--golang实现奖品发放系统

问题背景

最近业务上遇到这样的场景,觉得很有代表性,所以拿来说一说。我们有一个奖品发放系统,当用户申请奖品的时候,首先需要判断用户有没有申请过奖品,如果没有申请过,则去奖品总量扣除一个,然后再把用户申请记录写回数据库。

流程如下:

时序 事件
t1 检测用户申请过奖品
t2 奖品扣除
t3 插入用户申请记录
t4 返回申请成功

如果是正常的但线程执行完全没有问题,但是我们是并发的。所以很有可能会有好多事件都到t2导致多次申请,或者奖品多次扣除等情况。

解决方案

方案1:事务

最开始想到的就是数据库的事务了,在t1开始前启动事务,t1查询使用for update加锁,其他事务继续执行相同条件的for update的时候则会block,直到上个事务执行完成。这种方案很完美,但是只限于InnoDB,在tidb上行不通。tidb的事务是乐观锁,所以在t1的时候不会阻塞,那么可以在t3提交的时候conflict再回滚也可以啊。抱歉,tidb并不会产生插入冲突,因为tidb的锁不支持gap lock和next-key lock,所以如果我们的奖品申请表没有唯一索引冲突的话,完全可以插入(因为我某些设计,所以这里我们不能使用唯一索引)。

这个方案是行不通了,那么换种思路,既然tidb不支持for update block,那么我们是不是可以使用分布式锁来解决。

方案2:分布式锁

在检测用户有没有申请过奖品之前,我们可以以用户id为key申请分布式锁(可以使用redis实现),申请成功则进行下一步,其他用户阻塞等待锁释放。
具体实现如下:

        package logic

import (
	"errors"
	"fmt"
	"github.com/LucasGao67/blog001/dao"
	"github.com/LucasGao67/blog001/util"
	"sync"
)

var lock sync.Locker

func AppleAward(userId int64, remark string) error {
   

	// 1. 去申请锁
	key := fmt.Sprintf("test:{%d}", userId)
	//lock.()
	util.LockBlock(key)
	defer util.UnLodk(key)

	// 2. 去查询

	info, err := dao.Award.FindOne(userId)
	if err != nil {
   
		msg := "查询奖品申请信息失败"
		fmt.Printf("%s :%s\n", msg, err.Error()
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值