使用lua+redis解決發(fā)多張券的并發(fā)問題

更新時間：2021年01月11日 09:06:02 作者：gistmap

這篇文章主要介紹了使用lua+redis解決發(fā)多張券的并發(fā)問題,本文給大家介紹的非常詳細(xì)，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友可以參考下

前言

公司有一個發(fā)券的接口有并發(fā)安全問題，下面列出這個問題和解決這個問題的方式。

業(yè)務(wù)描述

這個接口的作用是給會員發(fā)多張券碼。涉及到4張主體，分別是：用戶，券，券碼，用戶領(lǐng)取記錄。
下面是改造前的偽代碼。
主要是因?yàn)椴槌鋈a那行存在并發(fā)安全問題，多個線程拿到同幾個券碼。以下都是基于如何讓取券碼變成原子的去展開。

public boolean sendCoupons(Long userId, Long couponId) {
 // 一堆校驗(yàn)
 // ...
 // 查出券碼
 List<CouponCode> couponCodes = couponCodeService.findByCouponId(couponId, num);
 // batchUpdateStatus是一個被@Transactional(propagation = Propagation.REQUIRES_NEW)修飾的方法
 // 批量更新為已被領(lǐng)取狀態(tài)
 couponCodeService.batchUpdateStatus(couponCods);
 // 發(fā)券
 // 發(fā)權(quán)益
 // 新增用戶券碼領(lǐng)取記錄
}

改造過程

因?yàn)槿a是多張，想用lua+redis的list結(jié)構(gòu)去做彈出。為什么用這種方案是因?yàn)閒or update直接被否了。

這是寫的lua腳本。。

local result = {}
for i=1,ARGV[1],1 do
 result[i] = redis.call("lpop", KEYS[1])
end
return table.contact(result , "|")

這是寫的執(zhí)行l(wèi)ua腳本的client。。其實(shí)主要的解決方法就是在redis的list里rpush（存）,lpop（?。┤?shù)據(jù)

@Slf4j
@Component
public class CouponCodeRedisQueueClient implements InitializingBean {

 /**
  * redis lua腳本文件路徑
  */
 public static final String POP_COUPON_CODE_LUA_PATH = "lua/pop-coupon-code.lua";
 public static final String SEPARATOR = "|";

 private static final String COUPON_CODE_KEY_PATTERN = "PROMOTION:COUPON_CODE_{0}";
 private String LUA_COUPON_CODE_SCRIPT;

 private String LUA_COUPON_CODE_SCRIPT_SHA;

 @Autowired
 private JedisTemplate jedisTemplate;

 @Override
 public void afterPropertiesSet() throws Exception {

  LUA_COUPON_CODE_SCRIPT = Resources.toString(Resources.getResource(POP_COUPON_CODE_LUA_PATH), Charsets.UTF_8);
  if (StringUtils.isNotBlank(LUA_COUPON_CODE_SCRIPT)) {

   LUA_COUPON_CODE_SCRIPT_SHA = jedisTemplate.execute(jedis -> {
    return jedis.scriptLoad(LUA_COUPON_CODE_SCRIPT);
   });
   log.info("redis lock script sha:{}", LUA_COUPON_CODE_SCRIPT_SHA);
  }

 }

 /**
  * 獲取Code
  *
  * @param activityId
  * @param num
  * @return
  */
 public List<String> popCouponCode(Long activityId, String num , int retryNum) {
  if(retryNum == 0){
   log.error("reload lua script error , try limit times ,activityId:{}", activityId);
   return Collections.emptyList();
  }
  List<String> keys = Lists.newArrayList();
  String key = buildKey(String.valueOf(activityId));
  keys.add(key);
  List<String> args = Lists.newArrayList();
  args.add(num);

  try {
   Object result = jedisTemplate.execute(jedis -> {
    if (StringUtils.isNotBlank(LUA_COUPON_CODE_SCRIPT_SHA)) {
     return jedis.evalsha(LUA_COUPON_CODE_SCRIPT_SHA, keys, args);
    } else {
     return jedis.eval(LUA_COUPON_CODE_SCRIPT, keys, args);
    }
   });
   log.info("pop coupon code by lua script.result:{}", result);
   if (Objects.isNull(result)) {
    return Collections.emptyList();
   }
   return Splitter.on(SEPARATOR).splitToList(result.toString());
  } catch (JedisNoScriptException jnse) {
   log.error("no lua lock script found.try to reload it", jnse);
   reloadLuaScript();
   //加載后重新執(zhí)行
   popCouponCode(activityId, num, --retryNum);
  } catch (Exception e) {
   log.error("failed to get a redis lock.key:{}", key, e);
  }
  return Collections.emptyList();
 }

 /**
  * 重新加載LUA腳本
  *
  * @throws Exception
  */
 public void reloadLuaScript() {
  synchronized (CouponCodeRedisQueueClient.class) {
   try {
    afterPropertiesSet();
   } catch (Exception e) {
    log.error("failed to reload redis lock lua script.retry load it.");
    reloadLuaScript();
   }
  }
 }

 /**
  * 構(gòu)建Key
  *
  * @param activityId
  * @return
  */
 public String buildKey(String activityId) {
  return MessageFormat.format(COUPON_CODE_KEY_PATTERN, activityId);
 }

}

當(dāng)然這種操作需要去提前把所有券的券碼丟到redis里去，這里我們也碰到了一些問題（券碼量比較大的情況下）。比如開始直接粗暴的用@PostConstruct去放入redis，導(dǎo)致項(xiàng)目啟動需要很久很久。。這里就不展開了，說一下我們嘗試的幾種方法