用redis實現悲觀鎖（後端語言以php為例）
鎖機制
通常使用的鎖分為樂觀鎖，悲觀鎖這兩種，簡單介紹下這兩種鎖，作為本文的背景知識，對這類知識已經有足夠了解的同學可以跳過這部分。
樂觀鎖
先來看下百度百科上的解釋：大多是基於數據版本（ Version ）記錄機制實現。何謂數據版本？即為數據增加一個版本標識，在基於數據庫表的版本解決方案中，一般是通過為數據庫表增加一個 “version” 字段來實現。讀取出數據時，將此版本號一同讀出，之後更新時，對此版本號加一。此時，將提交數據的版本數據與數據庫表對應記錄的當前版本信息進行比對，如果提交的數據版本號大於數據庫表當前版本號，則予以更新，否則認為是過期數據。
其實說白了，就是好比一個健身房裏只有一臺跑步機，在健身房門口有個排號機，每個進健身房的人都得先領一個號碼才能進入，如果跑步機上有人，則在一邊做做熱身、喝喝水，如果跑步機上沒人，則確認跑步機上當前顯示的號碼（上一個用過跑步機的人的號碼）是否比自己手持的小，如果小，則可以使用；否則，就意味著過號，而過號在現實中我們的都知道要麽走，要麽重排，就是不能插隊，在系統中也是一樣的，通常是返回錯誤。
悲觀鎖
同樣，來看下百度百科的解釋：具有強烈的獨占和排他特性。它指的是對數據被外界（包括本系統當前的其他事務，以及來自外部系統的事務處理）修改持保守態度，因此，在整個數據處理過程中，將數據處於鎖定狀態。悲觀鎖的實現，往往依靠數據庫提供的鎖機制（也只有數據庫層提供的鎖機制才能真正保證數據訪問的排他性，否則，即使在本系統中實現了加鎖機制，也無法保證外部系統不會修改數據）。
然後，也同樣通俗的解釋下，還是那個健身房。這次在門口不需要排號機了，而是掛著把鑰匙（只有一把），想進去的人必須拿到這把鑰匙才行，拿到鑰匙的人可以進入，不管是熱身、喝水還是跑步都可以，直到他出來把鑰匙掛回墻上，下一個才能去爭取，拿到的才可以再進去。聽著好像有點不人性化，所以悲觀鎖比較適合強一致性的場景，但效率比較低，特別是讀的並發低。樂觀鎖則適用於讀多寫少，並發沖突少的場景。
背景
先說下，本文的開發背景，方便大家了解為什麽要使用悲觀鎖以及文中鎖的詳細設計。
任務分發系統：任務池（mysql）中存在大量任務（文章），現在需要用戶協助編輯，系統基本需求如下（簡化版）：
1、推送用戶感興趣的分類下的任務到用戶編輯器中；
2、用戶編輯提交一個任務後，自動推送下一個任務；
3、每次只分配一個任務給用戶；
4、如果一個用戶占有某任務超過一定時間，則自動釋放任務，任務進任務池，重新循環；
5、……
目標
目標有兩個：
1、一個任務在同一時間段內只能被一個用戶所持有；
2、避免出現死任務，即避免任務被用戶長時間占有，無法釋放。
思路
由於系統並發量較大，並且有頻繁的寫操作，所以選擇悲觀鎖來控制每個任務只能同時被一個用戶領取。主要思路如下：
1、從任務池中找出一部分可分配的任務；
2、根據一定順序，選擇一個任務，作為候選推送任務；
3、嘗試對候選推送任務加鎖；
4、如果加鎖成功，則推送任務給用戶，並修改對應的任務狀態和用戶狀態；
5、如果加鎖失敗，則任務已被領取，重復2-5，直到推送成功。
實現
這裏只介紹下鎖的實現機制，其余業務邏輯略過。由於加鎖過程應該是不可拆解的，也就是常說的原子型操作，因此這裏選擇redis中的setnx操作作為加鎖的方法。
簡化版的代碼如下：
function lock($strMutex, $intTimeout) {
$objRedis = new Redis();
//使用setnx原子型操作加鎖
$intRet = $objRedis->setnx($strMutex, 1);
if ($intRet) {
//設置過期時間，防止死任務的出現
$objRedis->expire($strMutex, $intTimeout);
return true;
}
return false;
}- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
這段代碼有個問題，就是setnx成功，但expire失敗，這就可能存在死任務的情況。解決這個問題的一種通用方法是通過使用incr方法代替setnx，具體如下：
function lock($strMutex, $intTimeout, $intMaxTimes = 0) {
$objRedis = new Redis();
//使用incr原子型操作加鎖
$intRet = $objRedis->incr($strMutex);
if ($intRet === 1) {
//設置過期時間，防止死任務的出現
$objRedis->expire($strMutex, $intTimeout);
return true;
}
if ($intMaxTimes > 0 && $intRet >= $intMaxTimes && $objRedis->ttl($strMutex) === -1) {
//當設置了最大加鎖次數時，如果嘗試加鎖次數大於最大加鎖次數並且無過期時間則強制解鎖
$objRedis->del($strMutex);
}
return false;
}- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
- 13
- 14
- 15
- 16
這段代碼通過$intMaxTimes來保證即使在expire未成功的時候也能強制解鎖，保證系統不會出現死任務。
還有沒有更好的方法呢？
其實redis中的set操作已兼容了setnx，並且支持設置過期時間。
function lock($strMutex, $intTimeout) {
$objRedis = new Redis();
//使用setnx操作加鎖，同時設置過期時間
$strRet = $objRedis->set($strMutex, 1, ‘ex‘, $intTimeout, ‘nx‘);
if ($strRet === ‘OK‘) {
return true;
}
return false;
}- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
這個方法是我認為目前最好的，但是為什麽沒有直接介紹這個方法，而是先介紹incr那個方法呢？其實細心的同學可以看到上面那個方面有兩個加粗的字”通用“。之所以這麽說是因為set方法是從redis2.6.12版本才開始支持多參數的。
水平有限，歡迎指正~
如需轉發，請註明出處，thx~
參考資料：
http://redisdoc.com/string/set.html
用redis實現悲觀鎖（後端語言以php為例）