-- KEYS[1] = 鎖的 key（如 "my_lock"）
-- ARGV[1] = 鎖的 value（唯一標(biāo)識(shí)，如 UUID）
-- ARGV[2] = 過期時(shí)間（毫秒，如 "30000"）

local key = KEYS[1]
local value = ARGV[1]
local ttl = tonumber(ARGV[2])

-- 嘗試加鎖：只有當(dāng) key 不存在時(shí)才設(shè)置成功
if redis.call("SET", key, value, "NX", "PX", ttl) then
    return 1  -- 加鎖成功
else
    return 0  -- 加鎖失?。ㄒ驯徽加茫?
end

-- KEYS[1] = 鎖的 key
-- ARGV[1] = 鎖的 value（用于驗(yàn)證是否是自己的鎖）

local key = KEYS[1]
local value = ARGV[1]

-- 獲取當(dāng)前鎖的 value
local current_value = redis.call("GET", key)

-- 如果存在且等于自己的 value，則刪除
if current值 == value then
    redis.call("DEL", key)
    return 1  -- 解鎖成功
else
    return 0  -- 不是自己的鎖，拒絕刪除
end

import redis
import uuid
import time

# 初始化 Redis 連接
r = redis.StrictRedis(host='localhost', port=6379, decode_responses=True)

# 加載 Lua 腳本（推薦使用 script_load + evalsha 提高性能）
lock_script = r.register_script("""
if redis.call("SET", KEYS[1], ARGV[1], "NX", "PX", ARGV[2]) then
    return 1
else
    return 0
end
""")

unlock_script = r.register_script("""
if redis.call("GET", KEYS[1]) == ARGV[1] then
    return redis.call("DEL", KEYS[1])
else
    return 0
end
""")

def acquire_lock(lock_name, acquire_timeout=10, lock_timeout=30):
    """獲取分布式鎖"""
    identifier = str(uuid.uuid4())
    lock_timeout_ms = int(lock_timeout * 1000)
    end_time = time.time() + acquire_timeout

    while time.time() < end_time:
        # 執(zhí)行 Lua 腳本嘗試加鎖
        if lock_script(keys=[lock_name], args=[identifier, lock_timeout_ms]):
            return identifier  # 返回鎖標(biāo)識(shí)
        time.sleep(0.01)  # 短暫等待后重試

    return False

def release_lock(lock_name, identifier):
    """釋放分布式鎖"""
    unlock_script(keys=[lock_name], args=[identifier])

# ===== 使用示例 =====
lock_key = "order:123:lock"

# 嘗試獲取鎖
lock_id = acquire_lock(lock_key, acquire_timeout=5, lock_timeout=30)
if lock_id:
    try:
        print("? 獲取鎖成功，執(zhí)行業(yè)務(wù)邏輯...")
        time.sleep(5)  # 模擬業(yè)務(wù)處理
    finally:
        # 釋放鎖
        release_lock(lock_key, lock_id)
        print("?? 鎖已釋放")
else:
    print("? 獲取鎖超時(shí)")

-- KEYS[1] = 鎖 key
-- ARGV[1] = 客戶端 ID
-- ARGV[2] = 鎖過期時(shí)間（毫秒）

local key = KEYS[1]
local client_id = ARGV[1]
local ttl = tonumber(ARGV[2])

-- 檢查是否已持有鎖
local current_client = redis.call("HGET", key, client_id)

if current_client then
    -- 已持有：重入次數(shù) +1，刷新 TTL
    redis.call("HINCRBY", key, client_id, 1)
    redis.call("PEXPIRE", key, ttl)
    return 1
else
    -- 未持有：嘗試獲取鎖
    if redis.call("HLEN", key) == 0 then
        -- 鎖空閑，創(chuàng)建
        redis.call("HMSET", key, client_id, 1)
        redis.call("PEXPIRE", key, ttl)
        return 1
    else
        -- 鎖被他人持有
        return 0
    end
end

-- KEYS[1] = 鎖 key
-- ARGV[1] = 客戶端 ID

local key = KEYS[1]
local client_id = ARGV[1]

-- 檢查是否持有鎖
local counter = redis.call("HGET", key, client_id)

if not counter then
    return 0  -- 未持有鎖
end

counter = tonumber(counter)

if counter > 1 then
    -- 重入次數(shù) >1，減1
    redis.call("HINCRBY", key, client_id, -1)
    return 1
else
    -- 最后一次，刪除整個(gè) key
    redis.call("DEL", key)
    return 1
end