快捷導(dǎo)航

如何使用redis的setnx實現(xiàn)分布式鎖

更新時間：2024年06月03日 10:27:33 作者：ppo_wu

Redis Setnx（SET if Not eXists）命令在指定的 key 不存在時,為 key 設(shè)置指定的值,這篇文章主要介紹了使用redis的setnx實現(xiàn)分布式鎖,需要的朋友可以參考下

Redis Setnx（SET if Not eXists）命令在指定的 key 不存在時，為 key 設(shè)置指定的值。

語法

redis Setnx 命令基本語法如下：

redis 127.0.0.1:6379> SETNX KEY_NAME VALUE

可用版本

>= 1.0.0

返回值

設(shè)置成功，返回 1 。設(shè)置失敗，返回 0 。

實例

redis> EXISTS job                # job 不存在
(integer) 0
redis> SETNX job "programmer"    # job 設(shè)置成功
(integer) 1
redis> SETNX job "code-farmer"   # 嘗試覆蓋 job ，失敗
(integer) 0
redis> GET job                   # 沒有被覆蓋
"programmer"

在Redis中，SETNX 是 “Set If Not Exists”（如果不存在，則設(shè)置）的縮寫。這是一個原子操作，用于設(shè)置一個鍵的值，前提是這個鍵不存在。如果鍵已經(jīng)存在,.則不會執(zhí)行任何操作。

封裝方法trylock，用于獲取分布式鎖

/**
     * 嘗試獲  取一個鎖。
     *
     * @param name   鎖的名稱，通常是一個資源的標(biāo)識。
     * @param expire 鎖的過期時間，單位為毫秒。
     * @return 如果獲取鎖成功，返回一個唯一的token；如果失敗，則返回null。
     */
    public String tryLock(String name, long expire) {
        // 為鎖名稱添加后綴，以避免命名沖突
        name = name + "_lock";
        // 生成一個唯一的token，用于標(biāo)識持有鎖的客戶端
        String token = UUID.randomUUID().toString();
        // 獲取Redis連接工廠
        RedisConnectionFactory factory = stringRedisTemplate.getConnectionFactory();
        // 獲取Redis連接
        RedisConnection conn = factory.getConnection();
        try {
            // 使用SET命令嘗試以NX選項（只在鍵不存在時設(shè)置）設(shè)置鍵值對，如果成功，返回true
            // 這里使用了Expiration指定鍵的過期時間，以確保鎖在一段時間后自動釋放
            // 參考redis命令：SET key value [EX seconds] [PX milliseconds] [NX|XX]
            Boolean result = conn.set(
                    name.getBytes(), //key
                    token.getBytes(), //value
                    Expiration.from(expire, TimeUnit.MILLISECONDS),
                    RedisStringCommands.SetOption.SET_IF_ABSENT // NX
            );
            // 如果設(shè)置成功，返回生成的token
            if (result != null && result)
                return token;
        } finally {
            // 釋放Redis連接
            RedisConnectionUtils.releaseConnection(conn, factory, false);
        }
        // 如果未能成功獲取鎖，返回null
        return null;
    }

接下來，你可以在需要防止并發(fā)執(zhí)行的方法中使用tryLock方法：

public void exampleMethod(String taskName) {
		String lockKey = "myLockKey";
        // 嘗試獲取鎖
        String token = tryLock(lockKey, 10000); // 鎖過期時間為10秒
        if (token != null) {
            try {
                // 獲取鎖成功，執(zhí)行業(yè)務(wù)邏輯
                System.out.println("所獲取成功");
                // 模擬任務(wù)執(zhí)行
                // ...
            } finally {
                // 釋放鎖
                stringRedisTemplate.delete(lockKey+ "_lock");
            }
        } else {
            // 獲取鎖失敗，處理失敗邏輯
            System.out.println("獲取鎖失敗");
        }
    }

到此這篇關(guān)于使用redis的setnx實現(xiàn)分布式鎖的文章就介紹到這了,更多相關(guān)redis setnx分布式鎖內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: