前文回顾:用乐观锁修复红包超发bug及CAS原理介绍
前文回顾
接下来我们使用乐观锁的方式来修复红包超发的bug
乐观锁
乐观锁是一种不会阻塞其他线程并发的机制,它不会使用数据库的锁进行实现,它的设计里面由于不阻塞其他线程,所以并不会引发线程频繁挂起和恢复,这样便能够提高并发能力,也称之为为非阻塞锁。 乐观锁使用的是 CAS原理。
CAS 原理
在 CAS 原理中,对于多个线程共同的资源,先保存一个旧(Old Value),比如进入线程后,查询当前存量为 100 个红包,那么先把旧值保存为 100,然后经过一定的逻辑处理。
当需要扣减红包的时候,先比较数据库当前的值和旧值是否一致,如果一致则进行扣减红包的操作,否则就认为它已经被其他线程修改过了,不再进行操作。
CAS 原理流程如下:
CAS 原理并不排斥并发,也不独占资源,只是在线程开始阶段就读入线程共享数据,保存为旧值。当处理完逻辑,需要更新数据的时候,会进行一次 比较,即比较各个线程当前共享的数据是否和旧值保持一致。如果一致,就开始更新数据;如果不一致,则认为该前共享的数据是否和旧值保持一致。如果一致,就开始更新数据;如果不一致,则认为该重试,这样就是一个可重入锁,但是 CAS 原理会有一个问题,那就是 ABA 问题,我们先来看下ABA问题。
ABA问题
在处理复杂运算的时候,被线程 2 修改的 X 的值有可能导致线程1的运算出错,而最后线程 2 将 X 的值修改为原来的旧值 A,那么到了线程 1运算结束的时间顺序 T6,它将j检测 X 的值是否发生变化,就会拿旧值 A 和 当前的 X 的值 A 比对 , 结果是一致的, 于是提交事务,然后在复杂计算的过程中 X 被线程 2 修改过了,这会导致线程1的运算出错。
在这个过程中,对于线程 2 而言 , X 的值的变化为 A->B->A,所以 CAS 原理的这个设计缺陷被形象地称为“ABA 问题”。
ABA 问题的发生 , 是因为业务逻辑存在回退的可能性 。 如果加入一个非业务逻辑的属性,比如在一个数据中加入版本号( version ),对于版本号有一个约定,就是只要修改 X变量的数据,强制版本号( version )只能递增,而不会回退,即使是其他业务数据回退,它也会递增,那么 ABA 问题就解决了。
只是这个 version 变量并不存在什么业务逻辑,只是为了记录更新次数,只能递增,帮助我们克服 ABA 问题罢了,有了这些理论,我们就可以开始使用乐观锁来完成抢红包业务了 。
库表改造
为了顺利使用乐观锁,需要先在红包表 C T RED PACKET ) 加入一个新的列版本号(version),这个字段在建表的时候已经建了,只是我们还没有使用 。 这是第一步~
代码改造
既然库表加上了Version字段,那么应用中肯定要用到,自然而言的落到了Dao层上。
RedPacketDao新增接口方法及Mapper映射文件
RedPacketDao.java
/**
* @Description: 扣减抢红包数. 乐观锁的实现方式
* @param id
* -- 红包id
* @param version
* -- 版本标记
* @return: 更新记录条数
*/
publicintdecreaseRedPacketForVersion(@Param("id") Long id,@Param("version") Integer version);
RedPacket.xml
update
T_RED_PACKET
set stock = stock - 1 ,
version = version + 1
where id = #{id}
and version = #{version}
在扣减红包的时候 , 增加了对版本号的判断,其次每次扣减都会对版本号加一,这样保证每次更新在版本号上有记录 , 从而避免 ABA 问题
对于查询也不使用 for update 语句,避免锁的发生,这样就没有线程阻塞的问题了。然后就可以在类 UserRedPacketServic接口中新增方法 grapRedPacketForVersion,然后在其实现类中完成对应的逻辑即可。
UserRedPacketServic接口及实现类的改造
/**
* 保存抢红包信息. 乐观锁的方式
*@paramredPacketId
* 红包编号
*@paramuserId
* 抢红包用户编号
*@return影响记录数.
*/
publicintgrapRedPacketForVersion(Long redPacketId, Long userId);
实现类
/**
* 乐观锁,无重入
* */
@Override
@Transactional(isolation = Isolation.READ_COMMITTED, propagation = Propagation.REQUIRED)
publicintgrapRedPacketForVersion(Long redPacketId, Long userId){
// 获取红包信息
RedPacket redPacket = redPacketDao.getRedPacket(redPacketId);
// 当前小红包库存大于0
if(redPacket.getStock() >0) {
// 再次传入线程保存的version旧值给SQL判断,是否有其他线程修改过数据
intupdate = redPacketDao.decreaseRedPacketForVersion(redPacketId, redPacket.getVersion());
// 如果没有数据更新,则说明其他线程已经修改过数据,则重新抢夺
if(update ==0) {
returnFAILED;
// 生成抢红包信息
UserRedPacket userRedPacket =newUserRedPacket();
userRedPacket.setRedPacketId(redPacketId);
userRedPacket.setUserId(userId);
userRedPacket.setAmount(redPacket.getUnitAmount());
userRedPacket.setNote("redpacket- "+ redPacketId);
// 插入抢红包信息
intresult = userRedPacketDao.grapRedPacket(userRedPacket);
returnresult;
// 失败返回
returnFAILED;
version 值一开始就保存到了对象中,当扣减的时候,再次传递给 SQL ,让 SQL 对数据库的 version 和当前线程的旧值 version 进行比较。如果一致则插入抢红包的数据,否则就不进行操作。
Controller层新增路由方法
为了方便区分测试,在控制器 UserRedPacketController 内新建映射
@RequestMapping(value ="/grapRedPacketForVersion")
@ResponseBody
publicMap grapRedPacketForVersion(Long redPacketId, Long userId) {
// 抢红包
int result = userRedPacketService.grapRedPacketForVersion(redPacketId, userId);
Map retMap =newHashMap();
booleanflag = result >0;
retMap.put("success", flag);
retMap.put("message", flag ?"抢红包成功":"抢红包失败");
returnretMap;
View层
为了区分,新建个jsp吧 , 注意POST 请求地址和红包id 。
grapForVersion.jsp
$(document).ready(function(){
//模拟30000个异步请求,进行并发
varmax =30000;
for(vari =1; i 100) {
returnFAILED;
// 获取红包信息,注意version值
RedPacket redPacket = redPacketDao.getRedPacket(redPacketId);
// 当前小红包库存大于0
if(redPacket.getStock() >0) {
// 再次传入线程保存的version旧值给SQL判断,是否有其他线程修改过数据
intupdate = redPacketDao.decreaseRedPacketForVersion(redPacketId, redPacket.getVersion());
// 如果没有数据更新,则说明其他线程已经修改过数据,则重新抢夺
if(update ==0) {
continue;
// 生成抢红包信息
UserRedPacket userRedPacket =newUserRedPacket();
userRedPacket.setRedPacketId(redPacketId);
userRedPacket.setUserId(userId);
userRedPacket.setAmount(redPacket.getUnitAmount());
userRedPacket.setNote("抢红包 "+ redPacketId);
// 插入抢红包信息
intresult = userRedPacketDao.grapRedPacket(userRedPacket);
returnresult;
}else{
// 一旦没有库存,则马上返回
returnFAILED;
当因为版本号原因更新失败后,会重新尝试抢夺红包,但是会实现判断时间戳,如果时间戳在 100 毫秒内,就继续,否则就不再重新尝试,而判定失败,这样可以避免过多的SQL 执行,维持系统稳定。
初始化数据后,进行测试
从结果来看,之前大量失败的场景消失了,也没有超发现象,3 万次尝试抢光了所有的红包,避免了总是失败的结果,但是有时候时间戳并不是那么稳定,也会随着系统的空闲或者繁忙导致重试次数不一。有时候我们也会考虑、限制重试次数,比如 3 次,如下所示:
乐观锁重入机制-按次数重入
/**
*@Title: grapRedPacketForVersion
*@Description: 乐观锁,按次数重入
*@paramredPacketId
*@paramuserId
*@return: int
*/
@Override
@Transactional(isolation = Isolation.READ_COMMITTED, propagation = Propagation.REQUIRED)
publicintgrapRedPacketForVersion(Long redPacketId, Long userId){
for(inti =0; i
// 获取红包信息,注意version值
RedPacket redPacket = redPacketDao.getRedPacket(redPacketId);
// 当前小红包库存大于0
if(redPacket.getStock() >0) {
// 再次传入线程保存的version旧值给SQL判断,是否有其他线程修改过数据
intupdate = redPacketDao.decreaseRedPacketForVersion(redPacketId, redPacket.getVersion());
// 如果没有数据更新,则说明其他线程已经修改过数据,则重新抢夺
if(update ==0) {
continue;
// 生成抢红包信息
UserRedPacket userRedPacket =newUserRedPacket();
userRedPacket.setRedPacketId(redPacketId);
userRedPacket.setUserId(userId);
userRedPacket.setAmount(redPacket.getUnitAmount());
userRedPacket.setNote("抢红包 "+ redPacketId);
// 插入抢红包信息
intresult = userRedPacketDao.grapRedPacket(userRedPacket);
returnresult;
}else{
// 一旦没有库存,则马上返回
returnFAILED;
returnFAILED;
通过 for 循环限定重试 3 次,3 次过后无论成败都会判定为失败而退出,这样就能避免过多的重试导致过多 SQL 被执行的问题,从而保证数据库的性能。
同样的测试步骤,来看下统计结果
3 万次请求,所有红包都被抢到了,也没有发生超发现象,这样就可以消除大量的请求失败,避免非重入的时候大量请求失败的场景。
还能更好?
现在是使用数据库的情况,有时候并不想使用数据库作为抢红包时刻的数据保存载体,而是选择性能优于数据库的 Redis。之前接触过了Redis的事务,结合lua来实现抢红包的功能。
Redis-09Redis的基础事务:https://blog.csdn.net/yangshangwei/article/details/82863772
Redis-10Redis的事务回滚:https://blog.csdn.net/yangshangwei/article/details/82866216
Redis-11使用 watch 命令监控事务:https://blog.csdn.net/yangshangwei/article/details/82867200
先看下理论知识,下篇博文一起来探讨使用Redis + lua 实现抢红包的功能吧。
代码
https://github.com/yangshangwei/ssm_redpacket