Redis實現可重入的分散式鎖

风小雅發表於2024-07-09
Redis分散式

加鎖指令碼

-- 加鎖指令碼
-- 成功返回1，失敗返回-1
local key = KEYS[1]
local requestId = KEYS[2]
-- 單位毫秒
local ttl = tonumber(KEYS[3])
local result = redis.call('setnx', key, requestId)
if result == 1 then
    redis.call('pexpire', key, ttl)
else
    result = -1
    local value = redis.call('get', key)
    -- 可重入
    if (value == requestId) then
        result = 1;
        redis.call('pexpire', key, ttl)
    end
end
return result;

解鎖指令碼

-- 解鎖指令碼
-- 成功返回1，失敗返回-1
local key = KEYS[1]
local requestId = KEYS[2]
local value = redis.call('get', key)
if value == requestId then
    redis.call('del', key)
    return 1
end
return -1

執行指令碼程式碼

import redis.clients.jedis.Jedis;
import redis.clients.jedis.JedisPool;
import redis.clients.jedis.JedisPoolConfig;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Paths;

public class RedisLuaScriptExample {
    public static void main(String[] args) {
        JedisPool pool = new JedisPool(new JedisPoolConfig(), "localhost");
        try (Jedis jedis = pool.getResource()) {
            // 讀取 Lua 指令碼檔案
            String script = new String(Files.readAllBytes(Paths.get("lock.lua")));

            // 載入 Lua 指令碼到 Redis
            String scriptSha = jedis.scriptLoad(script, "0");

            // 執行 Lua 指令碼
            jedis.evalsha(script, 0); // 改一下引數

        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            pool.close();
        }
    }
}

相關文章