redis lua脚本实战秒杀扣减库存
作者：互联网
文章目录
前言
1.redisson介绍
2. redis lua脚本编写与执行
3.redis减库存lua脚本
4.实战
4.1 减库存逻辑
4.2 压测
前言
我们都知道redis是高性能高并发系统必不可少的kv中间件，它以高性能，高并发著称，我们常常用它做缓存，将热点数据或者是万年不变的数据缓存到redis中，查询的时候直接查询redis，减轻db的压力，分布式系统中我们也会拿它来做分布式锁，分布式id，幂等来解决一些分布式问题，redis也支持lua脚本，而且能够保证lua脚本执行过程中原子性，这就使得它的应用场景很多，也很典型，在redisson这个redis客户端中，它的各种分布式锁底层就是使用lua来实现的。本文主要是学习一下redis lua脚本的编写，以及在redisson这个redis客户端中是怎样使用的，实战一下秒杀场景redis减库存lua脚本的编写，并伪真实环境压测查看效果。
1.redisson介绍
redisson是一个redis的客户端，它拥有丰富的功能，而且支持redis的各种模式，什么单机，集群，哨兵的都支持，各种各样的分布式锁实现，什么分布式重入锁，红锁，读写锁，信号量的，然后它操作redis的常用数据结构就跟操作jdk的各种集合一样简单。
这里我们稍微演示下，不做过多的介绍，api毕竟只是个api，有意思的都在它背后各种原理。
maven依赖
<dependency>
<groupId>org.redisson</groupId>
<artifactId>redisson</artifactId>
<version>3.8.1</version>
</dependency>
1
2
3
4
5
创建RedissonClient对象
Config config = new Config();
config.useSingleServer()
.setAddress("redis://xxx:xx");
RedissonClient redissonClient = Redisson.create(config);
1
2
3
4
它的功能超级多，下面只是列举了一些常用的，还有什么Bloom过滤器，队列等等。
// string
redissonClient.getBucket("name").set("zhangsan");
redissonClient.getBucket("name").get();
// hash
RMap<Object, Object> user = redissonClient.getMap("user");
user.put("name","zhangsan");
user.put("age",11);
// list
RList<String> names = redissonClient.getList("names");
names.add("zhangsan");
names.add("lisi");
names.add("wangwu");
// set
RSet<Object> nameSet = redissonClient.getSet("names");
nameSet.add("lisi");
nameSet.add("lisi");
//lock
RLock lock = redissonClient.getLock("lock");
lock.lock();
lock.unlock();
// 分布式id
RAtomicLong id = redissonClient.getAtomicLong("id");
long l = id.incrementAndGet();
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
2. redis lua脚本编写与执行
其实redis中的lua脚本并不难，你也不需要把lua语言再去重学一遍，全凭感觉就好了，使用的时候去查下语法就ok了。
脚本中就一个redis.call() 应该算是函数吧（方法也可以），比如我要使用lua脚本实现set动作，就可以这样写
return redis.call('set','name','zhangsan')；
1
其实就是跟redis交互命令一个样子，再使用lua语言做一些条件分支，循环啥的，就完成了一些稍微复杂的逻辑。
下面自己写一遍扣减库存的逻辑，你就会这个玩意了。
上面是介绍了lua脚本的编写，下面我们介绍下这个执行。
不管是redis自己带的那个客户端，还是jedis，jediscluster，redistemplate，redisson这些客户端都是支持lua脚本api的，其实就是eval，evalsha，scriptload 这几个命令用的比较频繁，我这里把菜鸟教程上面关于介绍redis脚本命令截图过来
多说无益，这里直接使用redisson客户端实践一下。
public class RedisLua {
private static final Config config ;
private static final RedissonClient redisson ;
static {
config = new Config();
config.useSingleServer()
.setAddress("redis://ip:port");
redisson = Redisson.create(config);
}
public static void main(String[] args) throws InterruptedException {
redisson.getBucket("name").set(11);
RScript script = redisson.getScript();
String result = script.eval(RScript.Mode.READ_ONLY, new StringCodec(),"return redis.call('get','name');", RScript.ReturnType.VALUE);
System.out.println(result);
}
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
可以看到就是使用getScript方法获取一个script对象，然后调用script 对象的eval方法，这个script对象其实还是用好多个方法的，evalSha等等，可以自己研究下。然后就是通过lua脚本获取我上面set进去的name值。
我们在看上面菜鸟教程对脚本命令的介绍的时候，还发现有key… arg…这些东西，这个我认为就是动态替换（传参）的。
比如我现在不获取name这个key的值了，我要获取age 的，或者是我要直接set一个值，这个时候我就可以lua脚本中有两个变量与你传的参数对等起来
KEYS[1] ARGV[1]
你可以看作是数组，不过它的位置是从1开始的。
RScript script = redisson.getScript();
List<Object> keys=new ArrayList<>();
keys.add("age");
String re = script.eval(
RScript.Mode.READ_WRITE,
new StringCodec(),
"return redis.call('set',KEYS[1],ARGV[1]);",
RScript.ReturnType.VALUE,
keys,1);
Object age = redisson.getBucket("age").get();
System.out.println(age);
1
2
3
4
5
6
7
8
9
10
11
KEYS[1] 就对应这age， ARGV[1]就对应1，同理，KEYS[2] 就对应keys集合中的第二个元素。
3.redis减库存lua脚本
先介绍下下单减库存是怎么干的吧，其实一般库存有可用库存与预占库存，再下单的时候，就将可用库存减去你购买的商品数量看看是否是小于0，如果是小于0的话，说明库存不够了，就不让下单购买了，如果可用库存充足，可用库存减去购买商品数量，预占库存加上你购买商品数量，当用户超时未支付或者是手动取消订单的时候，就会去预占库减去用户购买商品数，可用库存加上商品数，其实还有一个已售库存，商家发货，已售库存加上商品数，预占减去商品数，大体上是这个逻辑。
现在可以想下，如果让你来实现下单预占库存的功能，你会怎么做，数据库三个库存字段这个不用说了。
首先你得先把可售库存查询出来，然后与购买商品数量进行比较，如果是可售库存大于这个购买商品数量，就可以购买，更新可售库存与预占库存。
如果上面这段代码逻辑你不加一些特殊手段的处理，那ok，高并发场景下绝对会出现超卖现象。
如果是秒杀场景呢？血亏。
这个时候我们可能会增加一些特殊手段来解决，比如说加锁，加分布式锁，将这一段的业务逻辑锁住，这个时候就不会出现那种超卖现象了， 但是这个中情况如果售罄的话，也会一直查询数据库。秒杀的时候，流量那么高，你不能让这么大的流量直接查库，如果商品售罄直接返回就可以了，不用再查询数据库了。
一般秒杀的时候，会将商品的库存同步推到redis中，流量过来的时候，会先扣减redis的库存，如果redis成功了，才扣减数据库中的库存，如果redis中的库存没了，直接返回就ok，这样大流量就不会直接冲击数据库了，那么redis要实现这段逻辑的话，就需要lua脚本的原子性了。
接下来我们就实现一下lua脚本扣减库存逻辑。
public static final String LOCK_STOCK_LUA= "local counter = redis.call('hget',KEYS[1],ARGV[1]); \n" +
"local result = counter - ARGV[2];" +
"if(result>=0 ) then \n" +
" redis.call('hset',KEYS[1],ARGV[1],result);\n" +
" redis.call('hincrby',KEYS[1],ARGV[3],ARGV[2]);\n" +
" return 1;\n" +
"end;\n" +
"return 0;\n";
1
2
3
4
5
6
7
8
我这里已经写好了，直接贴出来。
数据设计大体是这个样子的,使用hash数据结构
商品:{
“可售库存”:100,
“预占库存”:0,
“已售库存”:0
}
local counter = redis.call('hget',KEYS[1],ARGV[1]);
1
获取可售库存数量
local result = counter - ARGV[2];
if(result>=0 ) then
1
2
可售库存减去要购买的商品数量，如果是大于0的话，说明库存还够。
redis.call('hset',KEYS[1],ARGV[1],result);
redis.call('hincrby',KEYS[1],ARGV[3],ARGV[2]);
return 1;
1
2
3
重新设置可售库存数量，增加预占库存，然后返回1
如果是库存不够的话，直接返回0了就。
4.实战
4.1 减库存逻辑
减库存逻辑其实就是先是用lua脚本减redis库存，如果成功再去减数据库中的真实库存，如果减redis库存失败，库存不足，就不会再走后面减真实库存的逻辑了。
这块的话，我是写了一个库存服务，实现了这段逻辑，但是总感觉有各种数据不一致的问题，当然不是超卖，而是少卖问题，这里就不发出来了。
4.2 压测
我们这个实战是在阿里云进行的
redis选的是容器服务，按秒计费，配置是0.5c1g
mysql也是选择的容器服务，配置是0.5c1g
库存服务是云服务器，按小时计费的那种，配置是2c4g，因为要部署多个服务跟实例，选择的比较大。
压测也是使用的阿里云的性能测试服务。
redis监控，可以看到，这点并发对redis来说就是毛毛雨。cpu才使用7%
云服务器这块手速有点慢，没截图出来，cpu跟内存都在50%左右。
mysql数据库，可以看到cpu飙上去了，内存飙上去了。
数据库数据：
可以发现并没有出现超卖现象。
————————————————
版权声明：本文为CSDN博主「$码出未来」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/yuanshangshenghuo/article/details/115831912
标签：脚本,库存,扣减,redis,ARGV,lua,redisson 来源： https://blog.csdn.net/xiyang_1990/article/details/119897534