redis 不可重入分布式鎖(setNx()和getset()方法實現)


通常如果在單機環境,使用synchronized或juc ReentrantLock 實現鎖機制,但如果是分布式系統,則需要借助第三方工具實現,比如redis、zookeeper等。
redis為單進程單線程模式,操作都是原子過程,采用隊列模式將並發訪問變成串行訪問,且多客戶端對redis的連接並不存在競爭關系。
---------------------

一、redis 基本命令
setnx 命令(『SET if Not eXists』(如果不存在,則 SET)的簡寫):
設置成功,返回 1 。
設置失敗,返回 0 。
該命令是原子操作。

getset 命令:
自動將key對應到value並且返回原來key對應的value。如果key存在但是對應的value不是字符串,就返回錯誤。
* 返回值 *:
返回之前的舊值,如果之前Key不存在將返回null。
該命令是原子操作。

get 命令:
get <key> 獲取key的值,如果存在,則返回;如果不存在,則返回null;

del 命令:
del <key> 刪除key及key對應的值,如果key不存在,程序忽略

二、使用SETNX實現分布式鎖
redis官網推薦的setnx實現分布式鎖如下,多個進程執行以下Redis命令:
SETNX lock.foo {current Unix time + lock timeout + 1}

如果 SETNX 返回1,說明該進程獲得鎖,SETNX將鍵 lock.foo 的值設置為鎖的超時時間(當前時間 + 鎖的有效時間)。 
如果 SETNX 返回0,說明其他進程已經獲得了鎖,進程不能進入臨界區。進程可以在一個循環中不斷地嘗試 SETNX 操作,以獲得鎖。

三、死鎖問題
基於setnx、getset命令的分布式鎖實現過程中如果細節不反復測試,很容易出現死鎖問題:
考慮一種情況,如果進程獲得鎖后,斷開了與 Redis 的連接(可能是進程掛掉,或者網絡中斷),如果沒有有效的釋放鎖的機制,那么其他進程都會處於一直等待的狀態,即出現“死鎖”。上面在使用 SETNX 獲得鎖時,我們將鍵 lock.foo 的值設置為鎖的有效時間,進程獲得鎖后,其他進程還會不斷的檢測鎖是否已超時,如果超時,那么等待的進程也將有機會獲得鎖。然而,鎖超時時,我們不能簡單地使用 DEL 命令刪除鍵 lock.foo 以釋放鎖。考慮以下情況,進程P1已經首先獲得了鎖 lock.foo,然后進程P1掛掉了。進程P2,P3正在不斷地檢測鎖是否已釋放或者已超時,執行流程如下:
P2和P3進程讀取鍵 lock.foo 的值,檢測鎖是否已超時(通過比較當前時間和鍵 lock.foo 的值來判斷是否超時)
P2和P3進程發現鎖 lock.foo 已超時
P2執行 DEL lock.foo命令
P2執行 SETNX lock.foo命令,並返回1,即P2獲得鎖
P3執行 DEL lock.foo命令將P2剛剛設置的鍵 lock.foo 刪除(這步是由於P3剛才已檢測到鎖已超時)
P3執行 SETNX lock.foo命令,並返回1,即P3獲得鎖
P2和P3同時獲得了鎖
從上面的情況可以得知,在檢測到鎖超時后,進程不能直接簡單地執行 DEL 刪除鍵的操作以獲得鎖。為了解決上述算法可能出現的多個進程同時獲得鎖的問題,我們再來看以下的算法。 
我們同樣假設進程P1已經首先獲得了鎖 lock.foo,然后進程P1掛掉了。接下來的情況:
進程P4執行 SETNX lock.foo 以嘗試獲取鎖
由於進程P1已獲得了鎖,所以P4執行 SETNX lock.foo 返回0,即獲取鎖失敗
P4執行 GET lock.foo 來檢測鎖是否已超時,如果沒超時,則等待一段時間,再次檢測
如果P4檢測到鎖已超時,即當前的時間大於鍵 lock.foo 的值,P4會執行以下操作 
GETSET lock.foo {current Unix timestamp + lock timeout + 1}
由於 GETSET 操作在設置鍵的值的同時,還會返回鍵的舊值,通過比較鍵 lock.foo 的舊值是否小於當前時間,可以判斷進程是否已獲得鎖。
假如另一個進程P5也檢測到鎖已超時,並在P4之前執行了 GETSET 操作,那么P4的 GETSET 操作返回的是一個大於當前時間的時間戳,這樣P4就不會獲得鎖而繼續等待。注意到,即使P4接下來將鍵 lock.foo 的值設置了比P5設置的更大的值也沒影響。
另外,值得注意的是,在進程釋放鎖,即執行 DEL lock.foo 操作前,需要先判斷鎖是否已超時。如果鎖已超時,那么鎖可能已由其他進程獲得,這時直接執行 DEL lock.foo 操作會導致把其他進程已獲得 的鎖釋放掉。

 

四、代碼實現
本文基於jedis2.9, 以java語言代碼實現,如下:

測試代碼入口:
private static final String lockKey = "lock:user_13";  
    public static void main(String[] args) {  
        Jedis jedis = new Jedis("127.0.0.1", 6379);  
          
        DistributedLockHandler distributedLockHandler = new DistributedLockHandler().setJedis(jedis);  
        try{  
            boolean getLock = distributedLockHandler.tryLock(lockKey, Long.valueOf(5));  
  
            if(getLock){  
                // Do your job  
                System.out.println("Do your job........");  
            }  
  
        }catch(Exception e){  
            System.out.println(e);  
        }finally {  
            distributedLockHandler.realseLock(lockKey);  
        }  
          
    }       

分布式鎖實現代碼:
public class DistributedLockHandler {  
  
    private static final Integer Lock_Timeout = 3;  
    private Jedis jedis;  
  
    /** 
     * 外部調用加鎖的方法 
     * @param lockKey 鎖的key 
     * @param timeout 超時時間(放置時間長度,如:5L) 
     * @return 
     */  
    public boolean tryLock(String lockKey, Long timeout) {  
        try {  
            Long currentTime = System.currentTimeMillis();//開始加鎖的時間  
            boolean result = false;  
              
            while (true) {  
                if ((System.currentTimeMillis() - currentTime) / 1000 > timeout) {//當前時間超過了設定的超時時間  
                    System.out.println("Execute DistributedLockHandler.tryLock method, Time out.");  
                    break;  
                } else {  
                    result = innerTryLock(lockKey);  
                    if (result) {  
                        break;  
                    } else {  
                        System.out.println("Try to get the Lock,and wait 100 millisecond....");  
                        Thread.sleep(100);  
                    }  
                }  
            }  
            return result;  
        } catch (Exception e) {  
            System.out.println("Failed to run DistributedLockHandler.getLock method."+ e);  
            return false;  
        }  
    }  
      
    /** 
     * 釋放鎖 
     * @param lockKey 鎖的名字 
     */  
    public void realseLock(String lockKey) {  
        if(!checkIfLockTimeout(System.currentTimeMillis(), lockKey)){  
            jedis.del(lockKey);  
        }  
    }  
      
    /** 
     * 內部獲取鎖的實現方法 
     * @param lockKey 鎖的名字 
     * @return 
     */  
    private boolean innerTryLock(String lockKey) {  
          
        long currentTime = System.currentTimeMillis();//當前時間  
        String lockTimeDuration = String.valueOf(currentTime + Lock_Timeout + 1);//鎖的持續時間  
        Long result = jedis.setnx(lockKey, lockTimeDuration);  
          
        if (result == 1) {  
            return true;  
        } else {  
            if (checkIfLockTimeout(currentTime, lockKey)) {  
                String preLockTimeDuration = jedis.getSet(lockKey, lockTimeDuration);  
                if (currentTime > Long.valueOf(preLockTimeDuration)) {  
                    return true;  
                }  
            }  
            return false;  
        }  
          
    }  
  
    /** 
     * 判斷加鎖是否超時 
     * @param currentTime 當前時間 
     * @param lockKey 鎖的名字 
     * @return 
     */  
    private boolean checkIfLockTimeout(Long currentTime, String lockKey) {  
        if (currentTime > Long.valueOf(jedis.get(lockKey))) {//當前時間超過鎖的持續時間  
            return true;  
        } else {  
            return false;  
        }  
    }  
  
    public DistributedLockHandler setJedis(Jedis jedis) {  
        this.jedis = jedis;  
        return this;  
    }  
}  

上述代碼已完整實現基於redis setnx的分布式鎖,不過該鎖機制是不可重入的,與juc中ReentrantLock 可重入鎖仍存在距離,之后再分章節實現基於setnx的可重入鎖。


參考:
1. http://redis.io/commands/setnx
2. http://redis.io/topics/distlock
--------------------- 

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM