通過redis的腳本lua實現(xiàn)搶紅包功能的方法

CREATE TABLE `t_red_envelope` (
 `id` bigint(20) NOT NULL AUTO_INCREMENT COMMENT 'ID',
 `amount` decimal(10,2) DEFAULT NULL COMMENT '金額',
 `num` int(11) DEFAULT NULL COMMENT '數(shù)量(分割成幾分)',
 `create_time` datetime DEFAULT NULL COMMENT '創(chuàng)建時間',
 `update_time` datetime DEFAULT NULL COMMENT '更新時間',
 PRIMARY KEY (`id`)
) ENGINE=InnoDB AUTO_INCREMENT=10 DEFAULT CHARSET=utf8mb4 COMMENT='紅包'

CREATE TABLE `t_red_envelope_record` (
 `id` bigint(20) NOT NULL AUTO_INCREMENT COMMENT 'id',
 `user_id` bigint(20) DEFAULT NULL COMMENT '用戶id',
 `reward` decimal(10,2) DEFAULT NULL COMMENT '領(lǐng)取到獎勵',
 `red_envelope_id` bigint(20) DEFAULT NULL COMMENT '紅包id',
 `create_time` datetime DEFAULT NULL COMMENT '創(chuàng)建時間',
 `update_time` datetime DEFAULT NULL COMMENT '更新時間',
 PRIMARY KEY (`id`)
) ENGINE=InnoDB AUTO_INCREMENT=11 DEFAULT CHARSET=utf8mb4 COMMENT='紅包領(lǐng)取記錄'

 public Long divideRedEnvelope(int amount, int num) {
  /**
   * 每個人至少分到一分錢,如果有2000分，6人，隨機得到五個小于1994（2000-6）的數(shù)
   * 比如 a1=4，a2=120，a3=324，a4=500，a5=700(隨機拿到的五個數(shù)進行排序)，那么紅包錢分別為： a1+1,a2-a1+1,a3-a2+1,a4-a3+1,a5-a4+1,1994-a5+1(總和剛好為2000)
   */
  RedEnvelope redEnvelope = new RedEnvelope();
  redEnvelope.setAmount(new BigDecimal(amount));
  redEnvelope.setNum(num);
  redEnvelope.setCreateTime(new Date());
  redEnvelope.setUpdateTime(new Date());
  redEnvelopeDao.insert(redEnvelope);
  /**
   * 拿來隨機分的，按分來算
   */
  int totalAmount = amount * 100 - num;
  /**
   * 隨機數(shù)
   */
  int[] randomNum = new int[num - 1];
  /**
   * 紅包金額
   */
  int[] redEnvelopeAmount = new int[num];

  for (int i = 0; i < num - 1; i++) {
   int rand = new Random().nextInt(totalAmount);
   randomNum[i] = rand;
  }
  Arrays.sort(randomNum);
  /**
   * 條件語句分別分配的第一個、最后一個、中間的紅包
   */
  for (int i = 0; i < num; i++) {
   if (i == 0) {
    redEnvelopeAmount[i] = randomNum[i] + 1;
   } else if (i == num - 1) {
    redEnvelopeAmount[i] = totalAmount - randomNum[i - 1] + 1;
   } else {
    redEnvelopeAmount[i] = randomNum[i] - randomNum[i - 1] + 1;
   }
  }
  /**
   * 產(chǎn)生的小紅包key,以list存儲在reids中
   */
  String key = "envelope:redEnvelopeId:" + redEnvelope.getId();
  Boolean flag = stringRedisTemplate.hasKey(key);
  if (!flag) {
   for (Integer i : redEnvelopeAmount) {
    stringRedisTemplate.opsForList().leftPush(key, i + "");
   }
  }
  return redEnvelope.getId();
 }

 public String grabRedEnvelope(Long userId, Long redEnvelopeId) {

  DefaultRedisScript<String> redisScript = new DefaultRedisScript<>();
  redisScript.setResultType(String.class);
  redisScript.setScriptText(LuaScript.redLua);
  List<String> keyList = new ArrayList();
  /**
   * 產(chǎn)生的小紅包key
   */
  keyList.add("envelope:redEnvelopeId:" + redEnvelopeId);
  /**
   * 紅包領(lǐng)取記錄key
   */
  keyList.add("envelope:record:" + redEnvelopeId);
  keyList.add("" + userId);
  keyList.add(String.valueOf(userId));
  /**
   * -1 已經(jīng)搶到紅包 -2 紅包已經(jīng)完了 ，其余是搶到紅包并返回紅包余額
   */
  String result = stringRedisTemplate.execute(redisScript, keyList);
  return result;
 }

public class LuaScript {

 /**
  * -1 已經(jīng)搶到紅包 -2 紅包被搶光 re 紅包金額 ，keys[1]、keys[2]、keys[3]分別為存儲小紅包的key、紅包領(lǐng)取記錄key、用戶id
  */
 public static String redLua = "if redis.call('hexists',KEYS[2],KEYS[3]) ~=0 then \n" +
   " return '-1';\n" +
   " else \n" +
   "local re=redis.call('rpop',KEYS[1]);\n" +
   "if re then\n" +
   "redis.call('hset',KEYS[2],KEYS[3],1);\n" +
   "return re;\n" +
   "else\n" +
   "return '-2';\n" +
   "end\n" +
   "end";
}