通过redis的脚本lua如何实现抢红包功能

redis 脚本介绍

Redis从2.6版本开始，通过内嵌支持Lua环境

好处

• 减少网络开销。可以将多个请求通过脚本的形式一次发送，减少网络延迟
• 原子操作。redis将整个脚本当作一个整体去执行，中间不会被其他命令插入，无需担心脚本执行过程中会出现竞态条件
• 复用。客户端发送的脚本会永久保存在redis中，可以复用这一脚本

数据库表设计

简单两张表，一个红包表，一个红包领取记录表

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 '数量(分割成几分)',
 `create_time` datetime DEFAULT NULL COMMENT '创建时间',
 `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 '领取到奖励',
 `red_envelope_id` bigint(20) DEFAULT NULL COMMENT '红包id',
 `create_time` datetime DEFAULT NULL COMMENT '创建时间',
 `update_time` datetime DEFAULT NULL COMMENT '更新时间',
 PRIMARY KEY (`id`)
) ENGINE=InnoDB AUTO_INCREMENT=11 DEFAULT CHARSET=utf8mb4 COMMENT='红包领取记录'

代码编写

首先，生成一个红包，将其分成指定数量的随机小红包，以list结构（envelope:redEnvelopeId:红包id作为key）存储在reids中（以便抢红包弹出数据）

 public Long divideRedEnvelope(int amount, int num) {
  
  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;
  
  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;
   }
  }
  
  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();
 }

抢红包时，根据用户userId和红包id,生成KEYS[1]、KEYS[2]、KEYS[3] (存储小红包的key、领取红包记录的key、用户userId的key)传入脚本中。

​     1、先判断该用户是否抢过红包，有则返回-1，没有则从红包列表取出一个小红包

​     2、步骤1的小红包如果为空，则表明红包已经没抢光，返回 -2

​     3、否则返回取出的小红包金额

 public String grabRedEnvelope(Long userId, Long redEnvelopeId) {

  DefaultRedisscript redisscript = new DefaultRedisscript<>();
  redisscript.setResultType(String.class);
  redisscript.setscriptText(Luascript.redLua);
  List keyList = new ArrayList();
  
  keyList.add("envelope:redEnvelopeId:" + redEnvelopeId);
  
  keyList.add("envelope:record:" + redEnvelopeId);
  keyList.add("" + userId);
  keyList.add(String.valueOf(userId));
  
  String result = stringRedisTemplate.execute(redisscript, keyList);
  return result;
 }

实现抢红包的Lua脚本

public class Luascript {

 
 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 thenn" +
   "redis.call('hset',KEYS[2],KEYS[3],1);n" +
   "return re;n" +
   "elsen" +
   "return '-2';n" +
   "endn" +
   "end";
}

测试

首先通过接口分配红包生成一个100块、份额为10份的红包，并将其mysql数据库和redis

通过jmeter进行压测抢红包

结果

github代码链接

链接

总结

到此这篇关于通过redis的脚本lua如何实现抢红包功能的文章就介绍到这了,更多相关redis的脚本lua实现抢红包内容请搜索考高分网以前的文章或继续浏览下面的相关文章希望大家以后多多支持考高分网！