Redis中Lua腳本的常見場景

更新時間：2025年10月20日 08:31:22 作者：辭暮爾爾-煙火年年

Redis 的 Lua 腳本可以極大提升操作的原子性和效率,特別適用于需要多個 Redis 命令組合執(zhí)行的場景,本文就來詳細(xì)的介紹一下Redis中Lua腳本的常見場景,感興趣的可以了解一下

1. 分布式鎖

Redis 的 Lua 腳本常用于實現(xiàn)分布式鎖，以確保多個客戶端在并發(fā)訪問時的互斥性。

示例：分布式鎖的獲取與釋放

-- 獲取鎖
local lock_key = KEYS[1]
local lock_value = ARGV[1]
local ttl = tonumber(ARGV[2])

if redis.call("SETNX", lock_key, lock_value) == 1 then
    redis.call("PEXPIRE", lock_key, ttl)
    return 1
else
    return 0
end

redis-cli EVAL "local lock_key = KEYS[1]; local lock_value = ARGV[1]; local ttl = tonumber(ARGV[2]); if redis.call('SETNX', lock_key, lock_value) == 1 then redis.call('PEXPIRE', lock_key, ttl); return 1; else return 0; end" 1 mylock lock_value 30000

-- 釋放鎖
local lock_key = KEYS[1]
local lock_value = ARGV[1]

if redis.call("GET", lock_key) == lock_value then
    redis.call("DEL", lock_key)
    return 1
else
    return 0
end

redis-cli EVAL "local lock_key = KEYS[1]; local lock_value = ARGV[1]; if redis.call('GET', lock_key) == lock_value then redis.call('DEL', lock_key); return 1; else return 0; end" 1 mylock lock_value

2. 計數(shù)器

實現(xiàn)自增、自減等計數(shù)器功能。

示例：原子性的自增操作

local key = KEYS[1]
local increment = tonumber(ARGV[1])

local current = redis.call("GET", key)
if current == false then
    current = 0
else
    current = tonumber(current)
end

local new_value = current + increment
redis.call("SET", key, new_value)
return new_value

redis-cli EVAL "local key = KEYS[1]; local increment = tonumber(ARGV[1]); local current = redis.call('GET', key); if current == false then current = 0; else current = tonumber(current); end; local new_value = current + increment; redis.call('SET', key, new_value); return new_value;" 1 mycounter 1

3. 事務(wù)性操作

Lua 腳本可以確保多條命令的原子性，避免使用事務(wù)的復(fù)雜性。

示例：轉(zhuǎn)賬操作

local from_account = KEYS[1]
local to_account = KEYS[2]
local amount = tonumber(ARGV[1])

local from_balance = tonumber(redis.call("GET", from_account))
local to_balance = tonumber(redis.call("GET", to_account))

if from_balance >= amount then
    redis.call("DECRBY", from_account, amount)
    redis.call("INCRBY", to_account, amount)
    return 1
else
    return 0
end

redis-cli EVAL "local from_account = KEYS[1]; local to_account = KEYS[2]; local amount = tonumber(ARGV[1]); local from_balance = tonumber(redis.call('GET', from_account)); local to_balance = tonumber(redis.call('GET', to_account)); if from_balance >= amount then redis.call('DECRBY', from_account, amount); redis.call('INCRBY', to_account, amount); return 1; else return 0; end" 2 account1 account2 100

4. 排行榜

操作有序集合（sorted sets）實現(xiàn)排行榜功能。

示例：獲取排行榜前 N 名

local key = KEYS[1]
local limit = tonumber(ARGV[1])

return redis.call("ZRANGE", key, 0, limit - 1, "WITHSCORES")

redis-cli EVAL "local key = KEYS[1]; local limit = tonumber(ARGV[1]); return redis.call('ZRANGE', key, 0, limit - 1, 'WITHSCORES');" 1 leaderboard 10

5. 隊列操作

通過列表（list）實現(xiàn)任務(wù)隊列。

示例：推送和彈出任務(wù)

-- 推送任務(wù)到隊列
local queue_key = KEYS[1]
local task = ARGV[1]

redis.call("RPUSH", queue_key, task)
return redis.call("LLEN", queue_key)

redis-cli EVAL "local queue_key = KEYS[1]; local task = ARGV[1]; redis.call('RPUSH', queue_key, task); return redis.call('LLEN', queue_key);" 1 task_queue "task1"

-- 彈出任務(wù)
local queue_key = KEYS[1]

local task = redis.call("LPOP", queue_key)
if not task then
    return nil
else
    return task
end

redis-cli EVAL "local queue_key = KEYS[1]; local task = redis.call('LPOP', queue_key); if not task then return nil; else return task; end" 1 task_queue

6. 限流器

實現(xiàn)簡單的限流器，用于控制請求頻率。

示例：限流腳本

local key = KEYS[1]
local limit = tonumber(ARGV[1])
local interval = tonumber(ARGV[2])

local current = tonumber(redis.call("GET", key) or "0")
if current + 1 > limit then
    return false
else
    redis.call("INCR", key)
    if current == 0 then
        redis.call("EXPIRE", key, interval)
    end
    return true
end

redis-cli EVAL "local key = KEYS[1]; local limit = tonumber(ARGV[1]); local interval = tonumber(ARGV[2]); local current = tonumber(redis.call('GET', key) or '0'); if current + 1 > limit then return false; else redis.call('INCR', key); if current == 0 then redis.call('EXPIRE', key, interval); end; return true; end" 1 rate_limit_key 10 60

總結(jié)

Redis 的 Lua 腳本強大且靈活，適用于多種場景。通過合理使用 Lua 腳本，可以確保操作的原子性、減少網(wǎng)絡(luò)開銷和提高系統(tǒng)性能。上述示例涵蓋了常見的分布式鎖、計數(shù)器、事務(wù)性操作、排行榜、隊列操作和限流器等場景，為這些應(yīng)用場景提供了高效、可靠的解決方案。

到此這篇關(guān)于Redis中Lua腳本的常見場景的文章就介紹到這了,更多相關(guān)Redis Lua腳本常見場景內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: