通过redis+lua实现加减库存

Posted 2023-04-04

参考技术A

我们先分析场景，通过多种方案对比，选用了redis+lua的组合来满足我们的业务需要。利用redis单线程的特点，以及redis2.6版本后开始对lua的支持，我们采用redis执行lua脚本来确保我们查询+修改的串行执行。后面我们展示了code的实现案例，以及介绍了lua脚本的一些注意事项，可以依葫芦画瓢形式自己实现自己的需求。综合而言，我们分析场景应先分析其核心问题，然后利用一些更简洁的方法或小技巧来落地。

Redis+lua进行类似秒杀的实现

Redis+lua进行类似秒杀的实现

由于项目需要，需要多线程去获取和修改数据库的库存，考虑到给数据库加锁效率低，所以采用redis+lua来进行实现 。

redis的单线程操作特性来执行lua脚本，通过lua脚本来保证原子性。如果通过单纯的redis指令来进行更改，在读和写之间会存在多线程并发更新的问题。

1.首先定义redis数据结构

goodId:
 {
 "total":100,
 "released":0;
 }

• 其中goodId为商品id号，可根据此来查询相关的数据结构信息，total为总数，released为发放出去的数量，可使用数为total-released 2.编写lua脚本 ``` local n = tonumber(ARGV[1]) if not n or n == 0 then return 0 end local vals = redis.call("HMGET", KEYS[1], "total", "released"); local total = tonumber(vals[1]) local blocked = tonumber(vals[2]) if not total or not blocked then return 0 end if blocked + n <= total then redis.call("HINCRBY", KEYS[1], "released", n) return n; end return 0

- 执行脚本命令`EVAL script_string 1 goodId apply_count`
- 若库存足够则返回申请的数量，否则返回0，不返回可满足的剩余数

3.spring boot 调用 
- pom dependency 

<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-data-redis</artifactId>
<version>2.0.1.RELEASE</version>

- java code
```java
        long count = redisHelper.getStrCache().execute(new RedisCallback<Long>() {
            @Nullable
            @Override
            public Long doInRedis(RedisConnection redisConnection) throws DataAccessException {
                long ret = redisConnection.eval(script.getScriptAsString().getBytes(), ReturnType.INTEGER, 1, key.getBytes(), String.valueOf(count).getBytes());
                return ret;
            }
        });

4.redis->database

针对redis到databases的更新，思考了很久，没有找到较好的解决办法，先采用定时任务异步更新。至于数据是否丢失的问题，如果redis挂了，重启后redis会恢复数据，等下次定时任务就可以将数据库中的数据保持一致，缺点是redis挂了秒杀活动会失败。

至于redis到database更新的如何驱动，列出两种愚见:

• redis存一份相关hash键名单表，通过读取名单表来读取更新
• 通过流式读取databases中的表来读取更新。

欢迎各位提出问题，谢谢您的阅读