Redis高并發(fā)防止秒殺超賣實(shí)戰(zhàn)源碼解決方案

更新時(shí)間：2021年10月27日 11:44:50 作者：不要迷戀發(fā)哥

本文主要介紹了Redis高并發(fā)防止秒殺超賣實(shí)戰(zhàn)源碼解決方案，文中通過示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下

1：解決思路

將活動(dòng)寫入 redis 中，通過 redis 自減指令扣除庫存。

2：添加 redis 常量

commons/constant/RedisKeyConstant.java

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

3：添加 redis 配置類

4：修改業(yè)務(wù)層

廢話不多說，直接上源碼

1：秒殺業(yè)務(wù)邏輯層

@Service
public class SeckillService {
@Resource
private SeckillVouchersMapper seckillVouchersMapper;
@Resource
2private VoucherOrdersMapper voucherOrdersMapper;
@Value("${service.name.ms-oauth-server}")
private String oauthServerName;
@Resource
private RestTemplate restTemplate;
@Resource
private RedisTemplate redisTemplate;

2：添加需要搶購的代金券

@Transactional(rollbackFor = Exception.class)
public void addSeckillVouchers(SeckillVouchers seckillVouchers) {
// 非空校驗(yàn)
AssertUtil.isTrue(seckillVouchers.getFkVoucherId()== null,"請選擇需要搶購的代金券");
AssertUtil.isTrue(seckillVouchers.getAmount()== 0,"請輸入搶購總數(shù)量");
Date now = new Date();
AssertUtil.isNotNull(seckillVouchers.getStartTime(),"請輸入開始時(shí)間");
 
// 生產(chǎn)環(huán)境下面一行代碼需放行，這里注釋方便測試
// AssertUtil.isTrue(now.after(seckillVouchers.getStartTime()),"開始時(shí)間不能早于當(dāng)前時(shí)間");
AssertUtil.isNotNull(seckillVouchers.getEndTime(),"請輸入結(jié)束時(shí)間");
AssertUtil.isTrue(now.after(seckillVouchers.getEndTime()),"結(jié)束時(shí)間不能早于當(dāng)前時(shí)間");
AssertUtil.isTrue(seckillVouchers.getStartTime().after(seckillVouchers.getEndTime()),"開始時(shí)間不能晚于結(jié)束時(shí)間");
 
// 采用 Redis 實(shí)現(xiàn)
String key= RedisKeyConstant.seckill_vouchers.getKey() +seckillVouchers.getFkVoucherId();
// 驗(yàn)證 Redis 中是否已經(jīng)存在該券的秒殺活動(dòng),hash 不會(huì)做序列化和反序列化，
有利于性能的提高。entries(key），取到 key
Map<String, Object> map= redisTemplate.opsForHash().entries(key);
//如果不為空或 amount 庫存>0,該券已經(jīng)擁有了搶購活動(dòng),就不要再創(chuàng)建。
AssertUtil.isTrue(!map.isEmpty() && (int) map.get("amount") > 0,"該券已經(jīng)擁有了搶購活動(dòng)");
 
// 搶購活動(dòng)數(shù)據(jù)插入 Redis
seckillVouchers.setIsValid(1);
seckillVouchers.setCreateDate(now);
seckillVouchers.setUpdateDate(now);
//key 對應(yīng)的是 map,使用工具集將 seckillVouchers 轉(zhuǎn)成 map
redisTemplate.opsForHash().putAll(key,BeanUtil.beanToMap(seckillVouchers));
}

3：搶購代金券

@Transactional(rollbackFor = Exception.class)
public ResultInfo doSeckill(Integer voucherId, String accessToken, String path)
{
// 基本參數(shù)校驗(yàn)
AssertUtil.isTrue(voucherId == null || voucherId < 0,"請選擇需要搶購的代金券");
AssertUtil.isNotEmpty(accessToken,"請登錄");
 
// 采用 Redis
String key= RedisKeyConstant.seckill_vouchers.getKey() + voucherId;//根據(jù) key 獲取 map
Map<String, Object> map= redisTemplate.opsForHash().entries(key);
//map 轉(zhuǎn)對象
SeckillVouchers seckillVouchers = BeanUtil.mapToBean(map,SeckillVouchers.class, true, null);
 
// 判斷是否開始、結(jié)束
Date now = new Date();
AssertUtil.isTrue(now.before(seckillVouchers.getStartTime()),"該搶購還未開始");
AssertUtil.isTrue(now.after(seckillVouchers.getEndTime()),"該搶購已結(jié)束");
 
// 判斷是否賣完
AssertUtil.isTrue(seckillVouchers.getAmount() < 1,"該券已經(jīng)賣完了");
 
// 獲取登錄用戶信息
String url = oauthServerName +"user/me?access_token={accessToken}";
ResultInfo resultInfo = restTemplate.getForObject(url, ResultInfo.class,accessToken);
if (resultInfo.getCode() != ApiConstant.SUCCESS_CODE) {
resultInfo.setPath(path);
return resultInfo;
}
 
// 這里的 data 是一個(gè) LinkedHashMap，SignInDinerInfo
SignInDinerInfo dinerInfo = BeanUtil.fillBeanWithMap((LinkedHashMap)resultInfo.getData(), new SignInDinerInfo(), false);
 
// 判斷登錄用戶是否已搶到(一個(gè)用戶針對這次活動(dòng)只能買一次)
VoucherOrders order =voucherOrdersMapper.findDinerOrder(dinerInfo.getId(),seckillVouchers.getFkVoucherId());
AssertUtil.isTrue(order != null,"該用戶已搶到該代金券，無需再搶");
 
//扣庫存，采用 redis,redis 沒有設(shè)置自減，所以要自減，將步長設(shè)置為-1
long count = redisTemplate.opsForHash().increment(key,"amount",-1);
AssertUtil.isTrue(count < 0,"該券已經(jīng)賣完了");
 
// 下單存儲(chǔ)到數(shù)據(jù)庫
VoucherOrders voucherOrders = new VoucherOrders();
voucherOrders.setFkDinerId(dinerInfo.getId());
// Redis 中不需要維護(hù)外鍵信息
//voucherOrders.setFkSeckillId(seckillVouchers.getId());
voucherOrders.setFkVoucherId(seckillVouchers.getFkVoucherId());
String orderNo = IdUtil.getSnowflake(1, 1).nextIdStr();
voucherOrders.setOrderNo(orderNo);
voucherOrders.setOrderType(1);
voucherOrders.setStatus(0);
count = voucherOrdersMapper.save(voucherOrders);
AssertUtil.isTrue(count == 0,"用戶搶購失敗");
return ResultInfoUtil.buildSuccess(path,"搶購成功");
}
}

5：postman 測試

http://localhost:8083/add

{
"fkVoucherId":1,
"amount":100,
"startTime":"2020-02-04 11:12:00",
"endTime":"2021-02-06 11:12:00"
}

查看 redis

再次運(yùn)行 http://localhost:8083/add

6：壓力測試

查看 redis 中的庫存出現(xiàn)負(fù)值

在 redis 中修改庫存要分兩部進(jìn)行，先要獲取庫存的值，再扣減庫存。所以在高并發(fā)情況下，會(huì)導(dǎo)致 redis 扣減庫存出問題?？梢允褂?redis 的弱事務(wù)或 lua 腳本解決。 7：安裝Lua resources/stock.lua

if (redis.call('hexists', KEYS[1], KEYS[2])== 1) then
  local stock = tonumber(redis.call('hget', KEYS[1], KEYS[2]));
  if (stock > 0) then
    redis.call('hincrby', KEYS[1], KEYS[2],-1);
    return stock;
  end;
    return 0;
end;

hexists', KEYS[1], KEYS[2]) == 1
hexists 是判斷 redis 中 key 是否存在。
KEYS[1] 是 seckill_vouchers:1 KEYS[2] 是 amount
hget 是獲取 amount 賦給 stock
hincrby 是自增，當(dāng)為-1 是為自減。
因?yàn)樵?redis 中沒有自減指令，所以當(dāng)步長為 -1 表示自減。
現(xiàn)在使用 lua 腳本，將 redis 中查詢庫存和扣減庫存當(dāng)成原子性操作在一個(gè)線程內(nèi).

8：配置Lua

config/RedisTemplateConfiguration.java

@Bean
public DefaultRedisScript<Long> stockScript() {
  DefaultRedisScript<Long> redisScript = new DefaultRedisScript<>();
  //放在和 application.yml 同層目錄下
  redisScript.setLocation(new ClassPathResource("stock.lua"));
  redisScript.setResultType(Long.class);
  return redisScript;
}

9：修改業(yè)務(wù)層

ms-seckill/service/SeckilService.java

1：搶購代金券

@Transactional(rollbackFor = Exception.class)
public ResultInfo doSeckill(Integer voucherId, String accessToken, String path)
{
// 基本參數(shù)校驗(yàn)
AssertUtil.isTrue(voucherId == null || voucherId < 0,"請選擇需要搶購的代金券");
AssertUtil.isNotEmpty(accessToken,"請登錄");
// 采用 Redis
String key= RedisKeyConstant.seckill_vouchers.getKey() + voucherId;
//根據(jù) key 獲取 map
Map<String, Object> map= redisTemplate.opsForHash().entries(key);
//map 轉(zhuǎn)對象
SeckillVouchers seckillVouchers = BeanUtil.mapToBean(map,SeckillVouchers.class, true, null);
// 判斷是否開始、結(jié)束
Date now = new Date();AssertUtil.isTrue(now.before(seckillVouchers.getStartTime()),"該搶購還未開始");
AssertUtil.isTrue(now.after(seckillVouchers.getEndTime()),"該搶購已結(jié)束");
// 判斷是否賣完
AssertUtil.isTrue(seckillVouchers.getAmount() < 1,"該券已經(jīng)賣完了");
// 獲取登錄用戶信息
String url = oauthServerName +"user/me?access_token={accessToken}";
ResultInfo resultInfo = restTemplate.getForObject(url, ResultInfo.class,
accessToken);
if (resultInfo.getCode() != ApiConstant.SUCCESS_CODE) {
resultInfo.setPath(path);
return resultInfo;
}
// 這里的 data 是一個(gè) LinkedHashMap，SignInDinerInfo
SignInDinerInfo dinerInfo = BeanUtil.fillBeanWithMap((LinkedHashMap)
resultInfo.getData(), new SignInDinerInfo(), false);
// 判斷登錄用戶是否已搶到(一個(gè)用戶針對這次活動(dòng)只能買一次)
VoucherOrders order =voucherOrdersMapper.findDinerOrder(dinerInfo.getId(),
seckillVouchers.getFkVoucherId());
AssertUtil.isTrue(order != null,"該用戶已搶到該代金券，無需再搶");
 
//扣庫存，采用 redis,redis 沒有設(shè)置自減，所以要自減，將步長設(shè)置為-1
// long count = redisTemplate.opsForHash().increment(key,"amount",-1);
// AssertUtil.isTrue(count < 0,"該券已經(jīng)賣完了");
// 下單存儲(chǔ)到數(shù)據(jù)庫
VoucherOrders voucherOrders = new VoucherOrders();
voucherOrders.setFkDinerId(dinerInfo.getId());
// Redis 中不需要維護(hù)外鍵信息
//voucherOrders.setFkSeckillId(seckillVouchers.getId());
voucherOrders.setFkVoucherId(seckillVouchers.getFkVoucherId());
String orderNo = IdUtil.getSnowflake(1, 1).nextIdStr();
voucherOrders.setOrderNo(orderNo);
voucherOrders.setOrderType(1);
voucherOrders.setStatus(0);
long count = voucherOrdersMapper.save(voucherOrders);
AssertUtil.isTrue(count == 0,"用戶搶購失敗");
// 采用 Redis + Lua 解決問題
// 扣庫存
List<String> keys = new ArrayList<>();
//將 redis 的 key 放進(jìn)去keys.add(key);
keys.add("amount");
Long amount =(Long) redisTemplate.execute(defaultRedisScript, keys);
AssertUtil.isTrue(amount == null || amount < 1,"該券已經(jīng)賣完了");
return ResultInfoUtil.buildSuccess(path,"搶購成功");
}