springboot lua检查redis库存

devtools/2024/9/23 0:10:03/

需求

最近需求需要实现检查多个马戏场次下的座位等席对应库存渠道的库存余量，考虑到性能，决定采用Lua脚本实现库存检查。

数据结构

库存层级结构
redis库存hash类型结构

实现

lua脚本

lua">--- 字符串分割为数组
local function split(str, char)local arr = {};local pattern = string.format('([^%s]+)', char);for item in string.gmatch(str, pattern) dotable.insert(arr, item);endreturn arr;
end
--- 数组转成map
local function toMap(tb)if (tb == nil) thenreturn {};endlocal map = {};for i = 1, #tb doif (i % 2 == 1) thenmap[tb[i]] = tb[i+1];endendreturn map;
end
--- 初始化数据，入参01:1,02:2.01:3
local argArr = {};
for rowItem in string.gmatch(ARGV[1], '([^.]+)') dolocal row = {};for keyVal in string.gmatch(rowItem, '([^,]+)') do-- 解析出每个key与valuelocal tempArr = split(keyVal, ':');row[tempArr[1]] = tempArr[2];endtable.insert(argArr, row);
end
--- 检查库存
for i = 1, #KEYS do--local rData = redis.call("HMGET", KEYS[i], '01 02');local rdsMap = toMap(redis.call("HGETALL", KEYS[i]));for tier, quantity in pairs(argArr[i]) do--redis.log(redis.LOG_NOTICE, string.format('key is:%s,tier is:%s,quantity is:%s', KEYS[i], tier, quantity));if (rdsMap[tier] == nil or tonumber(quantity) > tonumber(rdsMap[tier])) thenreturn string.format("库存key:%s,tier:%s不足", KEYS[i], tier);endend
end
return '';

遍历键值对时需要使用pairs而不是ipairs，二者区别如下：
for k,v in pairs(argArr)的遍历顺序并非是table类型数据的排列顺序，而是根据table中key的hash值排列的顺序来遍历的。
for k,v in ipairs(argArr)必须要求table中的key为有序的，而且必须是从1开始，ipairs只会从1开始按连续的key顺序遍历到key不连续为止。

入参设计
由于RedisTemplate直接传入集合为参数序列化后会有中括号，而lua脚本table类型使用花括号定义，因此采用字符串作为入参，在lua脚本中解析字符串为table类型。例如参数01:1,02:2.01:3代表二维数组，第一行为01:1,02:2，其中01为渠道，1为渠道对应的库存，英文逗号作为渠道分隔符。
springboot调用lua脚本

@Autowired
private StringRedisTemplate redisJsonTemplate;@Test
public void checkInventory() {List<String> keyList = List.of("stock:GZ7:433680411156197407", "stock:GZ7:433680411156197408");// 执行 lua 脚本DefaultRedisScript<String> redisScript = new DefaultRedisScript<>();// 指定 lua 脚本redisScript.setScriptSource(new ResourceScriptSource(new ClassPathResource("/lua/TheaterStockCheck.lua")));// 指定返回类型redisScript.setResultType(String.class);Object result = redisJsonTemplate.execute(redisScript, keyList, "01:1,0101:2.01:3");//01:1,02:2.01:3System.out.println(result);
}