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

程序员文章站 2022-07-06 17:54:30

redis 脚本介绍redis从2.6版本开始，通过内嵌支持lua环境好处减少网络开销。可以将多个请求通过脚本的形式一次发送，减少网络延迟原子操作。redis将整个脚本当作一个整体去执行，中间...

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) {
  /**
   * 每个人至少分到一分钱,如果有2000分，6人，随机得到五个小于1994（2000-6）的数
   * 比如 a1=4，a2=120，a3=324，a4=500，a5=700(随机拿到的五个数进行排序)，那么红包钱分别为： 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;
  /**
   * 随机数
   */
  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;
   }
  }
  /**
   * 产生的小红包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();
 }

抢红包时，根据用户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<string> redisscript = new defaultredisscript<>();
  redisscript.setresulttype(string.class);
  redisscript.setscripttext(luascript.redlua);
  list<string> keylist = new arraylist();
  /**
   * 产生的小红包key
   */
  keylist.add("envelope:redenvelopeid:" + redenvelopeid);
  /**
   * 红包领取记录key
   */
  keylist.add("envelope:record:" + redenvelopeid);
  keylist.add("" + userid);
  keylist.add(string.valueof(userid));
  /**
   * -1 已经抢到红包 -2 红包已经完了 ，其余是抢到红包并返回红包余额
   */
  string result = stringredistemplate.execute(redisscript, keylist);
  return result;
 }

实现抢红包的lua脚本

public class luascript {

 /**
  * -1 已经抢到红包 -2 红包被抢光 re 红包金额 ，keys[1]、keys[2]、keys[3]分别为存储小红包的key、红包领取记录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";
}

测试

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

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