Redis配合Lua实现高并发防止秒杀超卖，实战源码解决方案

独行侠梦

于 2021-11-17 23:00:00 发布

阅读量1.7k

 收藏 4

点赞数 1

文章标签： 数据库 java redis mysql 分布式

版权

◆ 1：解决思路

将活动写入 redis 中，通过 redis 自减指令扣除库存。

◆ 2：添加 redis 常量

commons/constant/RedisKeyConstant.java 

seckill_vouchers("seckill_vouchers:","秒杀券的 key"),

◆ 3：添加 redis 配置类

◆ 4：修改业务层

废话不多说，直接上源码

1：秒杀业务逻辑层

1. @Service

2. public class SeckillService {

3. @Resource

4. private SeckillVouchersMapper seckillVouchersMapper;

5. @Resource

6. 2private VoucherOrdersMapper voucherOrdersMapper;

7. @Value("${service.name.ms-oauth-server}")

8. private String oauthServerName;

9. @Resource

10. private RestTemplate restTemplate;

11. @Resource

12. private RedisTemplate redisTemplate;

2：添加需要抢购的代金券

1. @Transactional(rollbackFor = Exception.class)

2. public void addSeckillVouchers(SeckillVouchers seckillVouchers) {

3. // 非空校验

4. AssertUtil.isTrue(seckillVouchers.getFkVoucherId()== null,"请选择需要抢购的代金券");

5. AssertUtil.isTrue(seckillVouchers.getAmount()== 0,"请输入抢购总数量");

6. Date now = new Date();

7. AssertUtil.isNotNull(seckillVouchers.getStartTime(),"请输入开始时间");

9. // 生产环境下面一行代码需放行，这里注释方便测试

10. // AssertUtil.isTrue(now.after(seckillVouchers.getStartTime()),"开始时间不能早于当前时间");

11. AssertUtil.isNotNull(seckillVouchers.getEndTime(),"请输入结束时间");

12. AssertUtil.isTrue(now.after(seckillVouchers.getEndTime()),"结束时间不能早于当前时间");

13. AssertUtil.isTrue(seckillVouchers.getStartTime().after(seckillVouchers.getEndTime()),"开始时间不能晚于结束时间");

15. // 采用 Redis 实现

16. String key= RedisKeyConstant.seckill_vouchers.getKey() +seckillVouchers.getFkVoucherId();

17. // 验证 Redis 中是否已经存在该券的秒杀活动,hash 不会做序列化和反序列化，

18. 有利于性能的提高。entries(key），取到 key

19. Map<String, Object> map= redisTemplate.opsForHash().entries(key);

20. //如果不为空或 amount 库存>0,该券已经拥有了抢购活动,就不要再创建。

21. AssertUtil.isTrue(!map.isEmpty() && (int) map.get("amount") > 0,"该券已经拥有了抢购活动");

23. // 抢购活动数据插入 Redis

24. seckillVouchers.setIsValid(1);

25. seckillVouchers.setCreateDate(now);

26. seckillVouchers.setUpdateDate(now);

27. //key 对应的是 map,使用工具集将 seckillVouchers 转成 map

28. redisTemplate.opsForHash().putAll(key,BeanUtil.beanToMap(seckillVouchers));

29. }

3：抢购代金券

1. @Transactional(rollbackFor = Exception.class)

2. public ResultInfo doSeckill(Integer voucherId, String accessToken, String path)

3. {

4. // 基本参数校验

5. AssertUtil.isTrue(voucherId == null || voucherId < 0,"请选择需要抢购的代金券");

6. AssertUtil.isNotEmpty(accessToken,"请登录");

8. // 采用 Redis

9. String key= RedisKeyConstant.seckill_vouchers.getKey() + voucherId;//根据 key 获取 map

10. Map<String, Object> map= redisTemplate.opsForHash().entries(key);

11. //map 转对象

12. SeckillVouchers seckillVouchers = BeanUtil.mapToBean(map,SeckillVouchers.class, true, null);

14. // 判断是否开始、结束

15. Date now = new Date();

16. AssertUtil.isTrue(now.before(seckillVouchers.getStartTime()),"该抢购还未开始");

17. AssertUtil.isTrue(now.after(seckillVouchers.getEndTime()),"该抢购已结束");

19. // 判断是否卖完

20. AssertUtil.isTrue(seckillVouchers.getAmount() < 1,"该券已经卖完了");

22. // 获取登录用户信息

23. String url = oauthServerName +"user/me?access_token={accessToken}";

24. ResultInfo resultInfo = restTemplate.getForObject(url, ResultInfo.class,accessToken);

25. if (resultInfo.getCode() != ApiConstant.SUCCESS_CODE) {

26. resultInfo.setPath(path);

27. return resultInfo;

28. }

30. // 这里的 data 是一个 LinkedHashMap，SignInDinerInfo

31. SignInDinerInfo dinerInfo = BeanUtil.fillBeanWithMap((LinkedHashMap)resultInfo.getData(), new SignInDinerInfo(), false);

33. // 判断登录用户是否已抢到(一个用户针对这次活动只能买一次)

34. VoucherOrders order =voucherOrdersMapper.findDinerOrder(dinerInfo.getId(),seckillVouchers.getFkVoucherId());

35. AssertUtil.isTrue(order != null,"该用户已抢到该代金券，无需再抢");

37. //扣库存，采用 redis,redis 没有设置自减，所以要自减，将步长设置为-1

38. long count = redisTemplate.opsForHash().increment(key,"amount",-1);

39. AssertUtil.isTrue(count < 0,"该券已经卖完了");

41. // 下单存储到数据库

42. VoucherOrders voucherOrders = new VoucherOrders();

43. voucherOrders.setFkDinerId(dinerInfo.getId());

44. // Redis 中不需要维护外键信息

45. //voucherOrders.setFkSeckillId(seckillVouchers.getId());

46. voucherOrders.setFkVoucherId(seckillVouchers.getFkVoucherId());

47. String orderNo = IdUtil.getSnowflake(1, 1).nextIdStr();

48. voucherOrders.setOrderNo(orderNo);

49. voucherOrders.setOrderType(1);

50. voucherOrders.setStatus(0);

51. count = voucherOrdersMapper.save(voucherOrders);

52. AssertUtil.isTrue(count == 0,"用户抢购失败");

53. return ResultInfoUtil.buildSuccess(path,"抢购成功");

54. }

55. }

◆ 5：postman 测试

http://localhost:8083/add

1. {

2. "fkVoucherId":1,

3. "amount":100,

4. "startTime":"2020-02-04 11:12:00",

5. "endTime":"2021-02-06 11:12:00"

6. }

 查看 redis

 再次运行 http://localhost:8083/add

◆  6：压力测试

 查看 redis 中的库存出现负值

在 redis 中修改库存要分两部进行，先要获取库存的值，再扣减库存。所以在高并

发情况下，会导致 redis 扣减库存出问题。可以使用 redis 的弱事务或 lua 脚本解决。

◆ 7：安装Lua

resources/stock.lua

1. if (redis.call('hexists', KEYS[1], KEYS[2])== 1) then

2. local stock = tonumber(redis.call('hget', KEYS[1], KEYS[2]));

3. if (stock > 0) then

4. redis.call('hincrby', KEYS[1], KEYS[2],-1);

5. return stock;

6. end;

7. return 0;

8. end;

hexists', KEYS[1], KEYS[2]) == 1

hexists 是判断 redis 中 key 是否存在。

KEYS[1] 是 seckill_vouchers:1 KEYS[2] 是 amount

hget 是获取 amount 赋给 stock

hincrby 是自增，当为-1 是为自减。

因为在 redis 中没有自减指令，所以当步长为 -1 表示自减。

现在使用 lua 脚本，将 redis 中查询库存和扣减库存当成原子性操作在一个线程内.

◆ 8：配置Lua

config/RedisTemplateConfiguration.java

1. @Bean

2. public DefaultRedisScript<Long> stockScript() {

3. DefaultRedisScript<Long> redisScript = new DefaultRedisScript<>();

4. //放在和 application.yml 同层目录下

5. redisScript.setLocation(new ClassPathResource("stock.lua"));

6. redisScript.setResultType(Long.class);

7. return redisScript;

8. }

◆ 9：修改业务层

ms-seckill/service/SeckilService.java

1：抢购代金券

1. @Transactional(rollbackFor = Exception.class)

2. public ResultInfo doSeckill(Integer voucherId, String accessToken, String path)

3. {

4. // 基本参数校验

5. AssertUtil.isTrue(voucherId == null || voucherId < 0,"请选择需要抢购的代金券");

6. AssertUtil.isNotEmpty(accessToken,"请登录");

7. // 采用 Redis

8. String key= RedisKeyConstant.seckill_vouchers.getKey() + voucherId;

9. //根据 key 获取 map

10. Map<String, Object> map= redisTemplate.opsForHash().entries(key);

11. //map 转对象

12. SeckillVouchers seckillVouchers = BeanUtil.mapToBean(map,SeckillVouchers.class, true, null);

13. // 判断是否开始、结束

14. Date now = new Date();AssertUtil.isTrue(now.before(seckillVouchers.getStartTime()),"该抢购还未开始");

15. AssertUtil.isTrue(now.after(seckillVouchers.getEndTime()),"该抢购已结束");

16. // 判断是否卖完

17. AssertUtil.isTrue(seckillVouchers.getAmount() < 1,"该券已经卖完了");

18. // 获取登录用户信息

19. String url = oauthServerName +"user/me?access_token={accessToken}";

20. ResultInfo resultInfo = restTemplate.getForObject(url, ResultInfo.class,

21. accessToken);

22. if (resultInfo.getCode() != ApiConstant.SUCCESS_CODE) {

23. resultInfo.setPath(path);

24. return resultInfo;

25. }

26. // 这里的 data 是一个 LinkedHashMap，SignInDinerInfo

27. SignInDinerInfo dinerInfo = BeanUtil.fillBeanWithMap((LinkedHashMap)

28. resultInfo.getData(), new SignInDinerInfo(), false);

29. // 判断登录用户是否已抢到(一个用户针对这次活动只能买一次)

30. VoucherOrders order =voucherOrdersMapper.findDinerOrder(dinerInfo.getId(),

31. seckillVouchers.getFkVoucherId());

32. AssertUtil.isTrue(order != null,"该用户已抢到该代金券，无需再抢");

34. //扣库存，采用 redis,redis 没有设置自减，所以要自减，将步长设置为-1

35. // long count = redisTemplate.opsForHash().increment(key,"amount",-1);

36. // AssertUtil.isTrue(count < 0,"该券已经卖完了");

37. // 下单存储到数据库

38. VoucherOrders voucherOrders = new VoucherOrders();

39. voucherOrders.setFkDinerId(dinerInfo.getId());

40. // Redis 中不需要维护外键信息

41. //voucherOrders.setFkSeckillId(seckillVouchers.getId());

42. voucherOrders.setFkVoucherId(seckillVouchers.getFkVoucherId());

43. String orderNo = IdUtil.getSnowflake(1, 1).nextIdStr();

44. voucherOrders.setOrderNo(orderNo);

45. voucherOrders.setOrderType(1);

46. voucherOrders.setStatus(0);

47. long count = voucherOrdersMapper.save(voucherOrders);

48. AssertUtil.isTrue(count == 0,"用户抢购失败");

49. // 采用 Redis + Lua 解决问题

50. // 扣库存

51. List<String> keys = new ArrayList<>();

52. //将 redis 的 key 放进去keys.add(key);

53. keys.add("amount");

54. Long amount =(Long) redisTemplate.execute(defaultRedisScript, keys);

55. AssertUtil.isTrue(amount == null || amount < 1,"该券已经卖完了");

56. return ResultInfoUtil.buildSuccess(path,"抢购成功");

57. }

◆ 10：压力测试

将 redis 中库存改回 100

压力测试

查看 redis 中 amount=0 ，不会变成负值

查看数据库下单表 t_voucher_orders ，共计下 100 个订单。

 

往期推荐

天秀！搞java的技术人写了本小说：《JavaScript百炼成仙》

55 个细节帮你全方位的完成Java 性能优化的 （珍藏）

Java 实现视频弹幕功能

Java的ClassLoader加载是怎么保证安全的？

Redis分布式锁需要考虑的这些事！

Redis 面试题！精华！收藏一波 !

解决kafka 消息堆积问题的排查及调优

MySQL 的隔离级别和事务需要知道的

回复【干货】获取精选干货视频教程

回复【加群】加入疑难问题攻坚交流群

回复【mat】获取内存溢出问题分析详细文档教程

回复【赚钱】获取用java写一个能赚钱的微信机器人

回复【副业】获取程序员副业攻略一份

戳这儿