用redis setNX命令來加鎖

加鎖和釋放鎖，都需要原子操作，不然在中途中斷就會(huì)發(fā)生異常，

刪鎖的時(shí)候也只能刪除自己對(duì)應(yīng)線程的鎖。

下面是用sexNX命令加鎖的模板

加鎖失敗后，休眠一段時(shí)候后，再次調(diào)用（自旋鎖）

這個(gè)還是會(huì)有一個(gè)問題，

如果加鎖的時(shí)間是30秒，業(yè)務(wù)執(zhí)行的時(shí)候了40秒，這個(gè)其他的線程就能搶到鎖，所以加鎖的時(shí)間必須大于業(yè)務(wù)執(zhí)行的時(shí)間

官方文檔

http://redis.cn/commands/set.html

建議 使用redisson來用分布式鎖

redis幾種加鎖的實(shí)現(xiàn)

1. redis加鎖分類

redis能用的的加鎖命令分表是INCR、SETNX、SET

2. 第一種鎖命令I(lǐng)NCR

這種加鎖的思路是， key 不存在，那么 key 的值會(huì)先被初始化為 0 ，然后再執(zhí)行 INCR 操作進(jìn)行加一。 

然后其它用戶在執(zhí)行 INCR 操作進(jìn)行加一時(shí)，如果返回的數(shù)大于 1 ，說明這個(gè)鎖正在被使用當(dāng)中。

• 1、 客戶端A請(qǐng)求服務(wù)器獲取key的值為1表示獲取了鎖
• 2、 客戶端B也去請(qǐng)求服務(wù)器獲取key的值為2表示獲取鎖失敗
• 3、 客戶端A執(zhí)行代碼完成，刪除鎖
• 4、 客戶端B在等待一段時(shí)間后在去請(qǐng)求的時(shí)候獲取key的值為1表示獲取鎖成功
• 5、 客戶端B執(zhí)行代碼完成，刪除鎖 

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

3. 第二種鎖SETNX

這種加鎖的思路是，如果 key 不存在，將 key 設(shè)置為 value 

如果 key 已存在，則 SETNX 不做任何動(dòng)作

• 1、 客戶端A請(qǐng)求服務(wù)器設(shè)置key的值，如果設(shè)置成功就表示加鎖成功
• 2、 客戶端B也去請(qǐng)求服務(wù)器設(shè)置key的值，如果返回失敗，那么就代表加鎖失敗
• 3、 客戶端A執(zhí)行代碼完成，刪除鎖
• 4、 客戶端B在等待一段時(shí)間后在去請(qǐng)求設(shè)置key的值，設(shè)置成功
• 5、 客戶端B執(zhí)行代碼完成，刪除鎖 

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

4. 第三種鎖SET

上面兩種方法都有一個(gè)問題，會(huì)發(fā)現(xiàn)，都需要設(shè)置 key 過期。那么為什么要設(shè)置key過期呢？如果請(qǐng)求執(zhí)行因?yàn)槟承┰蛞馔馔顺隽耍瑢?dǎo)致創(chuàng)建了鎖但是沒有刪除鎖，那么這個(gè)鎖將一直存在，以至于以后緩存再也得不到更新。于是乎我們需要給鎖加一個(gè)過期時(shí)間以防不測(cè)。 

但是借助 Expire 來設(shè)置就不是原子性操作了。所以還可以通過事務(wù)來確保原子性，但是還是有些問題，所以官方就引用了另外一個(gè)，使用 SET 命令本身已經(jīng)從版本 2.6.12 開始包含了設(shè)置過期時(shí)間的功能。

• 1、 客戶端A請(qǐng)求服務(wù)器設(shè)置key的值，如果設(shè)置成功就表示加鎖成功
• 2、 客戶端B也去請(qǐng)求服務(wù)器設(shè)置key的值，如果返回失敗，那么就代表加鎖失敗
• 3、 客戶端A執(zhí)行代碼完成，刪除鎖
• 4、 客戶端B在等待一段時(shí)間后在去請(qǐng)求設(shè)置key的值，設(shè)置成功
• 5、 客戶端B執(zhí)行代碼完成，刪除鎖 

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

5. 其它問題

雖然上面一步已經(jīng)滿足了我們的需求，但是還是要考慮其它問題？ 

• 1、 redis發(fā)現(xiàn)鎖失敗了要怎么辦？中斷請(qǐng)求還是循環(huán)請(qǐng)求？ 
• 2、 循環(huán)請(qǐng)求的話，如果有一個(gè)獲取了鎖，其它的在去獲取鎖的時(shí)候，是不是容易發(fā)生搶鎖的可能？ 
• 3、 鎖提前過期后，客戶端A還沒執(zhí)行完，然后客戶端B獲取到了鎖，這時(shí)候客戶端A執(zhí)行完了，會(huì)不會(huì)在刪鎖的時(shí)候把B的鎖給刪掉？

6. 解決辦法

• 針對(duì)問題1：使用循環(huán)請(qǐng)求，循環(huán)請(qǐng)求去獲取鎖 
• 針對(duì)問題2：針對(duì)第二個(gè)問題，在循環(huán)請(qǐng)求獲取鎖的時(shí)候，加入睡眠功能，等待幾毫秒在執(zhí)行循環(huán) 
• 針對(duì)問題3：在加鎖的時(shí)候存入的key是隨機(jī)的。這樣的話，每次在刪除key的時(shí)候判斷下存入的key里的value和自己存的是否一樣

? ? ? ? do { ?//針對(duì)問題1，使用循環(huán)
? ? ? ? ? ? $timeout = 10;
? ? ? ? ? ? $roomid = 10001;
? ? ? ? ? ? $key = 'room_lock';
? ? ? ? ? ? $value = 'room_'.$roomid; ?//分配一個(gè)隨機(jī)的值針對(duì)問題3
? ? ? ? ? ? $isLock = Redis::set($key, $value, 'ex', $timeout, 'nx');//ex 秒
? ? ? ? ? ? if ($isLock) {
? ? ? ? ? ? ? ? if (Redis::get($key) == $value) { ?//防止提前過期，誤刪其它請(qǐng)求創(chuàng)建的鎖
? ? ? ? ? ? ? ? ? ? //執(zhí)行內(nèi)部代碼
? ? ? ? ? ? ? ? ? ? Redis::del($key);
? ? ? ? ? ? ? ? ? ? continue;//執(zhí)行成功刪除key并跳出循環(huán)
? ? ? ? ? ? ? ? }
? ? ? ? ? ? } else {
? ? ? ? ? ? ? ? usleep(5000); //睡眠，降低搶鎖頻率，緩解redis壓力，針對(duì)問題2
? ? ? ? ? ? }
? ? ? ? } while(!$isLock);

7. 另外一個(gè)鎖

以上的鎖完全滿足了需求，但是官方另外還提供了一套加鎖的算法，這里以PHP為例

? ? $servers = [
? ? ? ? ['127.0.0.1', 6379, 0.01],
? ? ? ? ['127.0.0.1', 6389, 0.01],
? ? ? ? ['127.0.0.1', 6399, 0.01],
? ? ];
?
? ? $redLock = new RedLock($servers);
?
? ? //加鎖
? ? $lock = $redLock->lock('my_resource_name', 1000);
?
? ? //刪除鎖
? ? $redLock->unlock($lock)

總結(jié)

以上為個(gè)人經(jīng)驗(yàn)，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。

最新評(píng)論