服务器之家:专注于服务器技术及软件下载分享
分类导航

Mysql|Sql Server|Oracle|Redis|MongoDB|PostgreSQL|Sqlite|DB2|mariadb|Access|数据库技术|

服务器之家 - 数据库 - Redis - 通过redis的脚本lua如何实现抢红包功能

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

2020-07-05 17:00yuansmlk Redis

这篇文章主要给大家介绍了关于通过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

延伸 · 阅读

精彩推荐
  • RedisRedis的配置、启动、操作和关闭方法

    Redis的配置、启动、操作和关闭方法

    今天小编就为大家分享一篇Redis的配置、启动、操作和关闭方法,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧 ...

    大道化简5312019-11-14
  • Redisredis中如何使用lua脚本让你的灵活性提高5个逼格详解

    redis中如何使用lua脚本让你的灵活性提高5个逼格详解

    这篇文章主要给大家介绍了关于redis中如何使用lua脚本让你的灵活性提高5个逼格的相关资料,文中通过示例代码介绍的非常详细,对大家的学习或者工作具...

    一线码农5812019-11-18
  • RedisRedis如何实现数据库读写分离详解

    Redis如何实现数据库读写分离详解

    Redis的主从架构,能帮助我们实现读多,写少的情况,下面这篇文章主要给大家介绍了关于Redis如何实现数据库读写分离的相关资料,文中通过示例代码介绍...

    罗兵漂流记6092019-11-11
  • RedisRedis全量复制与部分复制示例详解

    Redis全量复制与部分复制示例详解

    这篇文章主要给大家介绍了关于Redis全量复制与部分复制的相关资料,文中通过示例代码介绍的非常详细,对大家学习或者使用Redis爬虫具有一定的参考学习...

    豆子先生5052019-11-27
  • Redisredis实现排行榜功能

    redis实现排行榜功能

    排行榜在很多地方都能使用到,redis的zset可以很方便地用来实现排行榜功能,本文就来简单的介绍一下如何使用,具有一定的参考价值,感兴趣的小伙伴们...

    乘月归5022021-08-05
  • Redisredis 交集、并集、差集的具体使用

    redis 交集、并集、差集的具体使用

    这篇文章主要介绍了redis 交集、并集、差集的具体使用,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友...

    xiaojin21cen10152021-07-27
  • RedisRedis 事务知识点相关总结

    Redis 事务知识点相关总结

    这篇文章主要介绍了Redis 事务相关总结,帮助大家更好的理解和学习使用Redis,感兴趣的朋友可以了解下...

    AsiaYe8232021-07-28
  • Redis详解Redis复制原理

    详解Redis复制原理

    与大多数db一样,Redis也提供了复制机制,以满足故障恢复和负载均衡等需求。复制也是Redis高可用的基础,哨兵和集群都是建立在复制基础上实现高可用的...

    李留广10222021-08-09