gpt4 book ai didi

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

转载 作者:qq735679552 更新时间:2022-09-29 22:32:09 27 4
gpt4 key购买 nike

CFSDN坚持开源创造价值,我们致力于搭建一个资源共享平台,让每一个IT人在这里找到属于你的精彩世界.

这篇CFSDN的博客文章通过redis的脚本lua如何实现抢红包功能由作者收集整理,如果你对这篇文章有兴趣,记得点赞哟.

redis 脚本介绍 。

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

好处 。

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

数据库表设计 。

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

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
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中(以便抢红包弹出数据) 。

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
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、否则返回取出的小红包金额 。

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
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脚本 。

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
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如何实现抢红包功能

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

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

通过jmeter进行压测抢红包 。

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

结果 。

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

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

github代码链接 。

链接 。

总结 。

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

原文链接:https://juejin.im/post/5ebeab656fb9a0433567b36f 。

最后此篇关于通过redis的脚本lua如何实现抢红包功能的文章就讲到这里了,如果你想了解更多关于通过redis的脚本lua如何实现抢红包功能的内容请搜索CFSDN的文章或继续浏览相关文章,希望大家以后支持我的博客! 。

27 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com