springboot lua檢查redis庫存的實現(xiàn)示例

更新時間：2024年09月12日 10:06:18 作者：morganEngineer

本文主要介紹了springboot lua檢查redis庫存的實現(xiàn)示例,為了優(yōu)化性能,通過Lua腳本實現(xiàn)對多個馬戲場次下的座位等席的庫存余量檢查,感興趣的可以了解一下

需求

最近需求需要實現(xiàn)檢查多個馬戲場次下的座位等席對應(yīng)庫存渠道的庫存余量，考慮到性能，決定采用Lua腳本實現(xiàn)庫存檢查。

數(shù)據(jù)結(jié)構(gòu)

庫存層級結(jié)構(gòu)

在這里插入圖片描述

redis庫存hash類型結(jié)構(gòu)

在這里插入圖片描述

實現(xiàn)

lua腳本

--- 字符串分割為數(shù)組
local function split(str, char)
    local arr = {};
    local pattern = string.format('([^%s]+)', char);
    for item in string.gmatch(str, pattern) do
        table.insert(arr, item);
    end
    return arr;
end
--- 數(shù)組轉(zhuǎn)成map
local function toMap(tb)
    if (tb == nil) then
        return {};
    end
    local map = {};
    for i = 1, #tb do
        if (i % 2 == 1) then
            map[tb[i]] = tb[i+1];
        end
    end
    return map;
end
--- 初始化數(shù)據(jù)，入?yún)?1:1,02:2.01:3
local argArr = {};
for rowItem in string.gmatch(ARGV[1], '([^.]+)') do
    local row = {};
    for keyVal in string.gmatch(rowItem, '([^,]+)') do
    -- 解析出每個key與value
        local tempArr = split(keyVal, ':');
        row[tempArr[1]] = tempArr[2];
    end
    table.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])) then
            return string.format("庫存key:%s,tier:%s不足", KEYS[i], tier);
        end
    end
end
return '';

遍歷鍵值對時需要使用pairs而不是ipairs，二者區(qū)別如下：
for k,v in pairs(argArr)的遍歷順序并非是table類型數(shù)據(jù)的排列順序，而是根據(jù)table中key的hash值排列的順序來遍歷的。
for k,v in ipairs(argArr)必須要求table中的key為有序的，而且必須是從1開始，ipairs只會從1開始按連續(xù)的key順序遍歷到key不連續(xù)為止。

入?yún)⒃O(shè)計
由于RedisTemplate直接傳入集合為參數(shù)序列化后會有中括號，而lua腳本table類型使用花括號定義，因此采用字符串作為入?yún)ⅲ趌ua腳本中解析字符串為table類型。例如參數(shù)01:1,02:2.01:3代表二維數(shù)組，第一行為01:1,02:2，其中01為渠道，1為渠道對應(yīng)的庫存，英文逗號作為渠道分隔符。
springboot調(diào)用lua腳本

@Autowired
private StringRedisTemplate redisJsonTemplate;

@Test
public void checkInventory() {
    List<String> keyList = List.of("stock:GZ7:433680411156197407", "stock:GZ7:433680411156197408");
    // 執(zhí)行 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:3
    System.out.println(result);
}

調(diào)用lua腳本需要使用string序列化方式，使用jdk序列化方式會導(dǎo)致序列化后的數(shù)據(jù)lua腳本無法解析

測試效果

redis庫存數(shù)據(jù)

在這里插入圖片描述