首頁 > 軟體

redis加鎖的幾種方式彙總

2023-03-09 06:06:36

redis常見的加鎖命令分別是INCR、SETNX、SET

1、INCR

這種鎖的加鎖思路是:

key不存在,那麼key的值會先被初始化為0,然後再執行INCR操作進行加一。

然後其它使用者在執行INCR操作進行加一時,如果返回的數值大於1,說明這個key正在被鎖定使用。

1、 使用者端A請求伺服器獲取key的值為1表示獲取了鎖    

2、 使用者端B也去請求伺服器獲取key的值為2表示獲取鎖失敗    

3、 使用者端A執行程式碼完成,刪除鎖    

4、 使用者端B在等待一段時間後在去請求的時候獲取key的值為1表示獲取鎖成功    

5、 使用者端B執行程式碼完成,刪除鎖 

$redis->incr($key);
$redis->expire($key, $ttl); //設定生成時間為1秒 

具體使用命令:

127.0.0.1:6379>INCR keyName

2、SETNX

這種加鎖的思路是,如果key不存在,將key設定為value,如果key存在,則SETNX不做任何動作。

SETNX是SET if Not eXists的縮寫。

1、 使用者端A請求伺服器設定key的值,如果設定成功就表示加鎖成功

2、 使用者端B也去請求伺服器設定key的值,如果返回失敗,那麼就代表加鎖失敗

3、 使用者端A執行程式碼完成,刪除鎖

4、 使用者端B在等待一段時間後在去請求設定key的值,設定成功

5、 使用者端B執行程式碼完成,刪除鎖

$redis->setNX($key, $value);
$redis->expire($key, $ttl);

具體使用命令:

redis> SETNX keyName value
(integer) 1 

設定成功,返回1;設定失敗,返回0

3、SET

上面兩種方法都有一個問題,會發現,都需要設定 key 過期。

那麼為什麼要設定key過期呢?

如果請求執行因為某些原因意外退出了,導致建立了鎖但是沒有刪除鎖,那麼這個鎖將一直存在,以至於以後快取再也得不到更新。

於是乎我們需要給鎖加一個過期時間以防不測。

但是藉助 Expire 來設定就不是原子性操作了。

所以還可以通過事務來確保原子性,但是還是有些問題,所以官方就參照了另外一個,使用 SET 命令本身已經從版本 2.6.12 開始包含了設定過期時間的功能。

1、 使用者端A請求伺服器設定key的值,如果設定成功就表示加鎖成功

2、 使用者端B也去請求伺服器設定key的值,如果返回失敗,那麼就代表加鎖失敗    

3、 使用者端A執行程式碼完成,刪除鎖    

4、 使用者端B在等待一段時間後在去請求設定key的值,設定成功

5、 使用者端B執行程式碼完成,刪除鎖

$redis->set($key, $value, array('nx', 'ex' => $ttl));  //ex表示秒

具體使用:

redis>set key value NX EX max-lock-time 實現加鎖    

命令解釋:

  • key:key就是redis的key值作為鎖的標識,value在這裡作為使用者端的標識,只有key-value都比配才有刪除鎖的權利【保證安全性】
  • max-lock-time:通過max-lock-time設定過期時間保證不會出現死鎖【避免死鎖】
  • NX:只有這個key不存才的時候才會進行操作,if not exists;
  • EX:設定key的過期時間為秒,具體時間由第5個引數決定

加鎖程式碼: 

 Jedis jedis = new Jedis("127.0.0.1", 6379);
 private static final String SUCCESS = "OK";
 /**
  * 加鎖操作
  * @param key 鎖標識
  * @param value 使用者端標識
  * @param timeOut 過期時間
  */
  
 public Boolean lock(String key,String value,Long timeOut){
     String var1 = jedis.set(key,value,"NX","EX",timeOut);
     if(LOCK_SUCCESS.equals(var1)){
         return true;
     }
     return false;
 }

解鎖程式碼:

 Jedis jedis = new Jedis("127.0.0.1", 6379); 
 private static final Long UNLOCK_SUCCESS = 1L;
 /**
  * 解鎖操作
  * @param key 鎖標識
  * @param value 使用者端標識
  * @return
  */
  
 public static Boolean unLock(String key,String value){
     String luaScript = "if redis.call("get",KEYS[1]) == ARGV[1] then return 
     redis.call("del",KEYS[1]) else  return 0 end";
     Object var2 = jedis.eval(luaScript,Collections.singletonList(key), Collections.singletonList(value));
     if (UNLOCK_SUCCESS == var2) {
         return true;
     }
     return false;
}

luaScript 這個字串是個lua指令碼,代表的意思是如果根據key拿到的value跟傳入的value相同就執行del,否則就返回0【保證安全性】

jedis.eval(String,list,list);這個命令就是去執行lua指令碼,KEYS的集合就是第二個引數,ARGV的集合就是第三引數【保證解鎖的原子操作】

上述就實現了怎麼使用redis去正確的實現分散式鎖,但是有個小缺陷就是鎖過期時間要設定為多少合適,這個其實還是需要去根據業務場景考量一下的。

總結

以上為個人經驗,希望能給大家一個參考,也希望大家多多支援it145.com。


IT145.com E-mail:sddin#qq.com