1. 为什么要引入令牌大闸?
场景1:分布式锁和限流都不能解决机器人刷票的问题,1000个请求抢票,900个限流快速失败,另外100个有可能是同一个在刷库。
引入令牌,令牌中记录用户信息,会进行校验用户是否拿过令牌,如果拿过令牌,那么几秒内不允许再获得令牌
场景2:没有余票时,需要查库存才能知道没票,会影响性能,不如查询令牌余票来的快
令牌的数量是和票数是相关的,令牌可以和票数相等,那么通过查询令牌就可以知道是否还有余票,会减少查询数据库,减少IO压力
2. 增加秒杀令牌表来维护令牌信息
增加一张表,表的创建SQL代码如下所示:
drop table if exists `sk_token`;
create table `sk_token` ( `id` bigint not null comment 'id', `date` date not null comment '日期', `train_code` varchar(20) not null comment '车次编号', `count` int not null comment '令牌余量', `create_time` datetime(3) comment '新增时间', `update_time` datetime(3) comment '修改时间', primary key (`id`), unique key `date_train_code_unique` (`date`, `train_code`)
) engine=innodb default charset=utf8mb4 comment='秒杀令牌';
利用代码生成器生成相应的文件
3. 初始化车次信息时初始化令牌信息
在SkTokenService中实现genDaily方法
/** * 初始化 */
public void genDaily(Date date, String trainCode) { LOG.info("删除日期【{}】车次【{}】的令牌记录", DateUtil.formatDate(date), trainCode); SkTokenExample skTokenExample = new SkTokenExample(); skTokenExample.createCriteria().andDateEqualTo(date).andTrainCodeEqualTo(trainCode); skTokenMapper.deleteByExample(skTokenExample); DateTime now = DateTime.now(); SkToken skToken = new SkToken(); skToken.setDate(date); skToken.setTrainCode(trainCode); skToken.setId(SnowUtil.getSnowflakeNextId()); skToken.setCreateTime(now); skToken.setUpdateTime(now); //计算该车次共有多少个座位int seatCount = dailyTrainSeatService.countSeat(date, trainCode); LOG.info("车次【{}】座位数:{}", trainCode, seatCount); //查询该车次共有多少个车站long stationCount = dailyTrainStationService.countByTrainCode(date, trainCode); LOG.info("车次【{}】到站数:{}", trainCode, stationCount); // 3/4需要根据实际卖票比例来定,一趟火车最多可以卖(seatCount * stationCount)张火车票 int count = (int) (seatCount * stationCount); // * 3/4); LOG.info("车次【{}】初始生成令牌数:{}", trainCode, count); skToken.setCount(count); skTokenMapper.insert(skToken);
}
然后在生成每日数据时加入该方法即可
//生成该车次的车站数据
dailyTrainStationService.genDaily(date,train.getCode());
//生成该车次的车厢数据
dailyTrainCarriageService.genDaily(date,train.getCode());
//生成该车次的座位数据
dailyTrainSeatService.genDaily(date,train.getCode());
//生成该车次的余票数据
dailyTrainTicketService.genDaily(dailyTrain,date,train.getCode());
LOG.info("生成日期【{}】车次【{}】的信息结束", DateUtil.formatDate(date), train.getCode());
//生成令牌余量数据
skTokenService.genDaily(date,train.getCode());
4. 增加校验秒杀令牌功能
在执行核心业务之前加上下面代码
//校验令牌容量
boolean validSkToken=skTokenService.validSkToken(req.getDate(),req.getTrainCode(), req.getMemberId());
if(validSkToken){ LOG.info("令牌校验通过");
}else{ LOG.info("令牌校验不通过"); throw new BusinessException(BusinessExceptionEnum.CONFIRM_ORDER_SK_TOKEN_FAIL);
}
其对应逻辑:先从redis缓存中查询令牌余量,如果存在缓存(60s过期),则直接从缓存中查询令牌余量,
如果余量大于0,则获取令牌,同时更新缓存中令牌余量
如果不存在缓存,则从数据库中查询
/** * 校验令牌 */
public boolean validSkToken(Date date, String trainCode, Long memberId) { LOG.info("会员【{}】获取日期【{}】车次【{}】的令牌开始", memberId, DateUtil.formatDate(date), trainCode); // 需要去掉这段,否则发布生产后,体验多人排队功能时,会因拿不到锁而返回:等待5秒,加入20人时,只有第1次循环能拿到锁 // if (!env.equals("dev")) { // // 先获取令牌锁,再校验令牌余量,防止机器人抢票,lockKey就是令牌,用来表示【谁能做什么】的一个凭证 // String lockKey = RedisKeyPreEnum.SK_TOKEN + "-" + DateUtil.formatDate(date) + "-" + trainCode + "-" + memberId; // Boolean setIfAbsent = redisTemplate.opsForValue().setIfAbsent(lockKey, lockKey, 5, TimeUnit.SECONDS); // if (Boolean.TRUE.equals(setIfAbsent)) { // LOG.info("恭喜,抢到令牌锁了!lockKey:{}", lockKey); // } else { // LOG.info("很遗憾,没抢到令牌锁!lockKey:{}", lockKey); // return false; // } // } String skTokenCountKey = RedisKeyPreEnum.SK_TOKEN_COUNT + "-" + DateUtil.formatDate(date) + "-" + trainCode; Object skTokenCount = redisTemplate.opsForValue().get(skTokenCountKey); if (skTokenCount != null) { LOG.info("缓存中有该车次令牌大闸的key:{}", skTokenCountKey); Long count = redisTemplate.opsForValue().decrement(skTokenCountKey, 1); if (count < 0L) { LOG.error("获取令牌失败:{}", skTokenCountKey); return false; } else { LOG.info("获取令牌后,令牌余数:{}", count); redisTemplate.expire(skTokenCountKey, 60, TimeUnit.SECONDS); // 每获取5个令牌更新一次数据库 if (count % 5 == 0) { skTokenMapperCust.decrease(date, trainCode, 5); } return true; } } else { LOG.info("缓存中没有该车次令牌大闸的key:{}", skTokenCountKey); // 检查是否还有令牌 SkTokenExample skTokenExample = new SkTokenExample(); skTokenExample.createCriteria().andDateEqualTo(date).andTrainCodeEqualTo(trainCode); List<SkToken> tokenCountList = skTokenMapper.selectByExample(skTokenExample); if (CollUtil.isEmpty(tokenCountList)) { LOG.info("找不到日期【{}】车次【{}】的令牌记录", DateUtil.formatDate(date), trainCode); return false; } SkToken skToken = tokenCountList.get(0); if (skToken.getCount() <= 0) { LOG.info("日期【{}】车次【{}】的令牌余量为0", DateUtil.formatDate(date), trainCode); return false; } // 令牌还有余量 // 令牌余数-1 Integer count = skToken.getCount() - 1; skToken.setCount(count); LOG.info("将该车次令牌大闸放入缓存中,key: {}, count: {}", skTokenCountKey, count); // 不需要更新数据库,只要放缓存即可 redisTemplate.opsForValue().set(skTokenCountKey, String.valueOf(count), 60, TimeUnit.SECONDS); skTokenMapper.updateByPrimaryKey(skToken); return true; } // 令牌约等于库存,令牌没有了,就不再卖票,不需要再进入购票主流程去判断库存,判断令牌肯定比判断库存效率高 // int updateCount = skTokenMapperCust.decrease(date, trainCode, 1); // if (updateCount > 0) { // return true; // } else { // return false; // }}