腳本之家服務器常用軟件

快捷導航

Redis SETNX的實現(xiàn)示例

更新時間：2024年12月27日 11:04:28 作者：飛滕人生TYF

SETNX是Redis提供的原子操作,用于在指定鍵不存在時設置鍵值,并返回操作結果,文中通過示例代碼介紹的非常詳細,對大家的學習或者工作具有一定的參考學習價值,需要的朋友們下面隨著小編來一起學習學習吧

SETNX 是 Redis 提供的一種原子操作，全稱是 “SET if Not eXists”，用于在指定的鍵不存在時設置鍵值，并返回操作結果。它是實現(xiàn)分布式鎖和冪等性控制的核心工具之一。

1. SETNX 的基本功能

語法

SETNX key value

key：需要設置的鍵。
value：需要設置的值。

返回值

1：如果鍵不存在，設置成功。
0：如果鍵已經(jīng)存在，不執(zhí)行任何操作。

使用示例

SETNX lock_key "123"

執(zhí)行結果

如果 lock_key 不存在，則設置鍵值為 "123"，并返回 1。
如果 lock_key 已存在，則不執(zhí)行任何操作，返回 0。

2. SETNX 的特性

原子性：
- SETNX 是 Redis 的原子操作，多個客戶端并發(fā)訪問時，只會有一個操作成功。
冪等性：
- 如果鍵已存在，則后續(xù)的 SETNX 調用不會影響當前值。
輕量級鎖：
- SETNX 常用于實現(xiàn)分布式鎖，通過確保某個鍵唯一存在來鎖定資源。

3. 結合 EXPIRE 的分布式鎖

SETNX 本身不能設置過期時間，因此為了避免死鎖問題（如客戶端異常未釋放鎖），可以結合 EXPIRE 設置鎖的自動過期時間。

問題

如果一個客戶端使用 SETNX 獲取鎖，卻因異常無法釋放鎖，其他客戶端可能會永遠無法獲取鎖。

解決方案 1：SETNX + EXPIRE

使用 SETNX 設置鎖。
如果鎖設置成功，立即設置過期時間。

if redis.call("SETNX", KEYS[1], ARGV[1]) == 1 then
    redis.call("EXPIRE", KEYS[1], ARGV[2])
    return 1
else
    return 0
end

缺點：

SETNX 和 EXPIRE 是兩個獨立操作，在高并發(fā)情況下可能出現(xiàn)非原子性問題。

解決方案 2：SETNX 改用 SET（推薦）

Redis 提供了改進版本的 SET 命令，可以直接設置鍵值并附加過期時間：

SET key value NX EX seconds

NX：表示僅當鍵不存在時才執(zhí)行設置操作（相當于 SETNX）。
EX seconds：設置過期時間，單位為秒。

示例

SET lock_key "123" NX EX 10

如果 lock_key 不存在，設置值為 "123"，且鍵將在 10 秒后過期。

優(yōu)點

原子操作，無需再單獨調用 EXPIRE。

4. 使用 SETNX 實現(xiàn)分布式鎖

SETNX 的一個典型應用是分布式鎖，保證在分布式系統(tǒng)中對共享資源的互斥訪問。

4.1 基本實現(xiàn)

獲取鎖：
- 使用 SETNX 嘗試設置一個鍵。
- 設置成功，表示成功獲取鎖。
釋放鎖：
- 檢查當前鎖是否屬于自己（通過唯一標識區(qū)分），如果是，則刪除鎖。

實現(xiàn)邏輯

String lockKey = "lock_key";
String requestId = UUID.randomUUID().toString();
int expireTime = 10;

// 獲取鎖
if (redisTemplate.opsForValue().setIfAbsent(lockKey, requestId, expireTime, TimeUnit.SECONDS)) {
    try {
        // 處理業(yè)務邏輯
    } finally {
        // 釋放鎖
        if (requestId.equals(redisTemplate.opsForValue().get(lockKey))) {
            redisTemplate.delete(lockKey);
        }
    }
} else {
    // 獲取鎖失敗
    System.out.println("Lock is already held by another process.");
}

4.2 Lua 腳本保證原子性

為了確保釋放鎖的操作是原子的，可以使用 Lua 腳本完成判斷和刪除：

if redis.call("get", KEYS[1]) == ARGV[1] then
    return redis.call("del", KEYS[1])
else
    return 0
end

調用示例

String script = "if redis.call('get', KEYS[1]) == ARGV[1] then " +
                "return redis.call('del', KEYS[1]) " +
                "else return 0 end";
redisTemplate.execute(new DefaultRedisScript<>(script, Long.class), 
                      Collections.singletonList(lockKey), requestId);

5. SETNX 的典型應用場景

5.1 分布式鎖

確保資源互斥訪問，防止并發(fā)修改造成數(shù)據(jù)錯誤。

5.2 請求去重

對同一用戶的重復請求設置唯一標識，防止重復處理。
示例：使用 SETNX 設置請求 ID，只有第一次請求會被處理。

5.3 冪等性控制

確保某些操作（如支付、扣款）不會因重復請求而執(zhí)行多次。

6. SETNX 的優(yōu)缺點

優(yōu)點	缺點
原子性強，適合高并發(fā)場景	無法直接設置過期時間
實現(xiàn)簡單，易于集成到業(yè)務邏輯中	需要結合 EXPIRE 或改用 `SET` 命令
性能高，Redis 本身支持高吞吐量	需要額外處理死鎖或鎖釋放的邊界條件

7. SETNX 的改進建議

盡量使用 SET key value NX EX seconds 替代 SETNX：
- 提供了原生的過期時間設置，簡化了分布式鎖的實現(xiàn)。
結合 Lua 腳本：
- 使用 Lua 腳本處理復雜邏輯，保證操作的原子性。
監(jiān)控鎖的狀態(tài)：
- 對于長期持有鎖的操作，應增加心跳機制，防止鎖意外釋放。
鎖爭搶優(yōu)化：
- 避免高并發(fā)環(huán)境下大量線程重復嘗試獲取鎖，可以結合延時隊列或限流機制。