基於Redis實現分佈式鎖之前,這些坑你一定得知道

開頭

基於Redis的分佈式鎖對大家來說並不陌生,可是你的分佈式鎖有失敗的時候嗎?在失敗的時候可曾懷疑過你在用的分佈式鎖真的靠譜嗎?以下是結合自己的踩坑經驗總結的一些經驗之談。

你真的需要分佈式鎖嗎?

用到分佈式鎖說明遇到了多個進程共同訪問同一個資源的問題, 一般是在兩個場景下會防止對同一個資源的重複訪問:

  1. 提高效率。比如多個節點計算同一批任務,如果某個任務已經有節點在計算了,那其他節點就不用重複計算了,以免浪費計算資源。不過重複計算也沒事,不會造成其他更大的損失。也就是允許偶爾的失敗。
  2. 保證正確性。這種情況對鎖的要求就很高了,如果重複計算,會對正確性造成影響。這種不允許失敗。

引入分佈式鎖勢必要引入一個第三方的基礎設施,比如MySQL,Redis,Zookeeper等,這些實現分佈式鎖的基礎設施出問題了,也會影響業務,所以在使用分佈式鎖前可以考慮下是否可以不用加鎖的方式實現?不過這個不在本文的討論範圍內,本文假設加鎖的需求是合理的,並且偏向於上面的第二種情況,爲什麼是偏向?因爲不存在100%靠譜的分佈式鎖,看完下面的內容就明白了。

從一個簡單的分佈式鎖實現說起

分佈式鎖的Redis實現很常見,自己實現和使用第三方庫都很簡單,至少看上去是這樣的,這裏就介紹一個最簡單靠譜的Redis實現。

最簡單的實現

實現很經典了,這裏只提兩個要點?

  1. 加鎖和解鎖的鎖必須是同一個,常見的解決方案是給每個鎖一個鑰匙(唯一ID),加鎖時生成,解鎖時判斷。
  2. 不能讓一個資源永久加鎖。常見的解決方案是給一個鎖的過期時間。當然了還有其他方案,後面再說。

一個可複製粘貼的實現方式如下:

加鎖

public static boolean tryLock(String key, String uniqueId, int seconds) {
    return "OK".equals(jedis.set(key, uniqueId, "NX", "EX", seconds));
}

這裏其實是調用了 SET key value PX milliseoncds NX,不明白這個命令的參考下SET key value [EX seconds|PX milliseconds] [NX|XX] [KEEPTTL]

解鎖

public static boolean releaseLock(String key, String uniqueId) {
    String luaScript = "if redis.call('get', KEYS[1]) == ARGV[1] then " +
            "return redis.call('del', KEYS[1]) else return 0 end";
    return jedis.eval(
        luaScript, 
        Collections.singletonList(key), 
        Collections.singletonList(uniqueId)
    ).equals(1L);
}

這段實現的精髓在那個簡單的lua腳本上,先判斷唯一ID是否相等再操作。

靠譜嗎?

這樣的實現有什麼問題呢?

  1. 單點問題。上面的實現只要一個master節點就能搞定,這裏的單點指的是單master,就算是個集羣,如果加鎖成功後,鎖從master複製到slave的時候掛了,也是會出現同一資源被多個client加鎖的。
  2. 執行時間超過了鎖的過期時間。上面寫到爲了不出現一直上鎖的情況,加了一個兜底的過期時間,時間到了鎖自動釋放,但是,如果在這期間任務並沒有做完怎麼辦?由於GC或者網絡延遲導致的任務時間變長,很難保證任務一定能在鎖的過期時間內完成。

如何解決這兩個問題呢?試試看更復雜的實現吧。

Redlock算法

對於第一個單點問題,順着redis的思路,接下來想到的肯定是Redlock了。Redlock爲了解決單機的問題,需要多個(大於2)redis的master節點,多個master節點互相獨立,沒有數據同步。

Redlock的實現如下:

  1. 獲取當前時間。
  2. **依次獲取N個節點的鎖。**每個節點加鎖的實現方式同上。這裏有個細節,就是每次獲取鎖的時候的過期時間都不同,需要減去之前獲取鎖的操作的耗時,
  • 比如傳入的鎖的過期時間爲500ms,
  • 獲取第一個節點的鎖花了1ms,那麼第一個節點的鎖的過期時間就是499ms,
  • 獲取第二個節點的鎖花了2ms,那麼第二個節點的鎖的過期時間就是497ms
  • 如果鎖的過期時間小於等於0了,說明整個獲取鎖的操作超時了,整個操作失敗
  1. **判斷是否獲取鎖成功。**如果client在上述步驟中獲取到了(N/2 + 1)個節點鎖,並且每個鎖的過期時間都是大於0的,則獲取鎖成功,否則失敗。失敗時釋放鎖。
  2. **釋放鎖。**對所有節點發送釋放鎖的指令,每個節點的實現邏輯和上面的簡單實現一樣。爲什麼要對所有節點操作?因爲分佈式場景下從一個節點獲取鎖失敗不代表在那個節點上加速失敗,可能實際上加鎖已經成功了,但是返回時因爲網絡抖動超時了。

以上就是大家常見的redlock實現的描述了,一眼看上去就是簡單版本的多master版本,如果真是這樣就太簡單了,接下來分析下這個算法在各個場景下是怎樣被玩壞的。

分佈式鎖的坑

高併發場景下的問題

以下問題不是說在併發不高的場景下不容易出現,只是在高併發場景下出現的概率更高些而已。**性能問題。**性能問題來自於兩個方面。

  1. 獲取鎖的時間上。如果redlock運用在高併發的場景下,存在N個master節點,一個一個去請求,耗時會比較長,從而影響性能。這個好解決。通過上面描述不難發現,從多個節點獲取鎖的操作並不是一個同步操作,可以是異步操作,這樣可以多個節點同時獲取。即使是並行處理的,還是得預估好獲取鎖的時間,保證鎖的TTL > 獲取鎖的時間+任務處理時間。
  2. 被加鎖的資源太大。加鎖的方案本身就是會爲了正確性而犧牲併發的,犧牲和資源大小成正比。這個時候可以考慮對資源做拆分,拆分的方式有兩種:
  3. 從業務上將鎖住的資源拆分成多段,每段分開加鎖。比如,我要對一個商戶做若干個操作,操作前要鎖住這個商戶,這時我可以將若干個操作拆成多個獨立的步驟分開加鎖,提高併發。
  4. 用分桶的思想,將一個資源拆分成多個桶,一個加鎖失敗立即嘗試下一個。比如批量任務處理的場景,要處理200w個商戶的任務,爲了提高處理速度,用多個線程,每個線程取100個商戶處理,就得給這100個商戶加鎖,如果不加處理,很難保證同一時刻兩個線程加鎖的商戶沒有重疊,這時可以按一個維度,比如某個標籤,對商戶進行分桶,然後一個任務處理一個分桶,處理完這個分桶再處理下一個分桶,減少競爭。

**重試的問題。**無論是簡單實現還是redlock實現,都會有重試的邏輯。如果直接按上面的算法實現,是會存在多個client幾乎在同一時刻獲取同一個鎖,然後每個client都鎖住了部分節點,但是沒有一個client獲取大多數節點的情況。解決的方案也很常見,在重試的時候讓多個節點錯開,錯開的方式就是在重試時間中加一個隨機時間。這樣並不能根治這個問題,但是可以有效緩解問題,親試有效。

節點宕機

對於單master節點且沒有做持久化的場景,宕機就掛了,這個就必須在實現上支持重複操作,自己做好冪等。

對於多master的場景,比如redlock,我們來看這樣一個場景:

  1. 假設有5個redis的節點:A、B、C、D、E,沒有做持久化。
  2. client1從A、B、C 3個節點獲取鎖成功,那麼client1獲取鎖成功。
  3. 節點C掛了。
  4. client2從C、D、E獲取鎖成功,client2也獲取鎖成功,那麼在同一時刻client1和client2同時獲取鎖,redlock被玩壞了。

怎麼解決呢?最容易想到的方案是打開持久化。持久化可以做到持久化每一條redis命令,但這對性能影響會很大,一般不會採用,如果不採用這種方式,在節點掛的時候肯定會損失小部分的數據,可能我們的鎖就在其中。 另一個方案是延遲啓動。就是一個節點掛了修復後,不立即加入,而是等待一段時間再加入,等待時間要大於宕機那一刻所有鎖的最大TTL。 但這個方案依然不能解決問題,如果在上述步驟3中B和C都掛了呢,那麼只剩A、D、E三個節點,從D和E獲取鎖成功就可以了,還是會出問題。那麼只能增加master節點的總量,緩解這個問題了。增加master節點會提高穩定性,但是也增加了成本,需要在兩者之間權衡。

任務執行時間超過鎖的TTL

之前產線上出現過因爲網絡延遲導致任務的執行時間遠超預期,鎖過期,被多個線程執行的情況。 這個問題是所有分佈式鎖都要面臨的問題,包括基於zookeeper和DB實現的分佈式鎖,這是鎖過期了和client不知道鎖過期了之間的矛盾。 在加鎖的時候,我們一般都會給一個鎖的TTL,這是爲了防止加鎖後client宕機,鎖無法被釋放的問題。但是所有這種姿勢的用法都會面臨同一個問題,就是沒發保證client的執行時間一定小於鎖的TTL。雖然大多數程序員都會樂觀的認爲這種情況不可能發生,我也曾經這麼認爲,直到被現實一次又一次的打臉。

Martin Kleppmann也質疑過這一點,這裏直接用他的圖:

  1. Client1獲取到鎖
  2. Client1開始任務,然後發生了STW的GC,時間超過了鎖的過期時間
  3. Client2 獲取到鎖,開始了任務
  4. Client1的GC結束,繼續任務,這個時候Client1和Client2都認爲自己獲取了鎖,都會處理任務,從而發生錯誤。

Martin Kleppmann舉的是GC的例子,我碰到的是網絡延遲的情況。不管是哪種情況,不可否認的是這種情況無法避免,一旦出現很容易懵逼。

如何解決呢?一種解決方案是不設置TTL,而是在獲取鎖成功後,給鎖加一個watchdog,watchdog會起一個定時任務,在鎖沒有被釋放且快要過期的時候會續期。這樣說有些抽象,下面結合redisson源碼說下:

public class RedissonLock extends RedissonExpirable implements RLock {
    ...
    @Override
    public void lock() {
        try {
            lockInterruptibly();
        } catch (InterruptedException e) {
            Thread.currentThread().interrupt();
        }
    }

    @Override
    public void lock(long leaseTime, TimeUnit unit) {
        try {
            lockInterruptibly(leaseTime, unit);
        } catch (InterruptedException e) {
            Thread.currentThread().interrupt();
        }
    }
    ...
 }

redisson常用的加鎖api是上面兩個,一個是不傳入TTL,這時是redisson自己維護,會主動續期;另外一種是自己傳入TTL,這種redisson就不會幫我們自動續期了,或者自己將leaseTime的值傳成-1,但是不建議這種方式,既然已經有現成的API了,何必還要用這種奇怪的寫法呢。 接下來分析下不傳參的方法的加鎖邏輯:

public class RedissonLock extends RedissonExpirable implements RLock {

    ...
        
    public static final long LOCK_EXPIRATION_INTERVAL_SECONDS = 30;
    protected long internalLockLeaseTime = TimeUnit.SECONDS.toMillis(LOCK_EXPIRATION_INTERVAL_SECONDS);

        
    @Override
    public void lock() {
        try {
            lockInterruptibly();
        } catch (InterruptedException e) {
            Thread.currentThread().interrupt();
        }
    }
        
    @Override
    public void lockInterruptibly() throws InterruptedException {
        lockInterruptibly(-1, null);
    }
    
    @Override
    public void lockInterruptibly(long leaseTime, TimeUnit unit) throws InterruptedException {
        long threadId = Thread.currentThread().getId();
        Long ttl = tryAcquire(leaseTime, unit, threadId);
        // lock acquired
        if (ttl == null) {
            return;
        }

        RFuture<RedissonLockEntry> future = subscribe(threadId);
        commandExecutor.syncSubscription(future);

        try {
            while (true) {
                ttl = tryAcquire(leaseTime, unit, threadId);
                // lock acquired
                if (ttl == null) {
                    break;
                }

                // waiting for message
                if (ttl >= 0) {
                    getEntry(threadId).getLatch().tryAcquire(ttl, TimeUnit.MILLISECONDS);
                } else {
                    getEntry(threadId).getLatch().acquire();
                }
            }
        } finally {
            unsubscribe(future, threadId);
        }
//        get(lockAsync(leaseTime, unit));
    }

    private Long tryAcquire(long leaseTime, TimeUnit unit, long threadId) {
        return get(tryAcquireAsync(leaseTime, unit, threadId));
    }

    private <T> RFuture<Long> tryAcquireAsync(long leaseTime, TimeUnit unit, final long threadId) {
        if (leaseTime != -1) {
            return tryLockInnerAsync(leaseTime, unit, threadId, RedisCommands.EVAL_LONG);
        }
        RFuture<Long> ttlRemainingFuture = tryLockInnerAsync(LOCK_EXPIRATION_INTERVAL_SECONDS, TimeUnit.SECONDS, threadId, RedisCommands.EVAL_LONG);
        ttlRemainingFuture.addListener(new FutureListener<Long>() {
            @Override
            public void operationComplete(Future<Long> future) throws Exception {
                if (!future.isSuccess()) {
                    return;
                }

                Long ttlRemaining = future.getNow();
                // lock acquired
                if (ttlRemaining == null) {
                    scheduleExpirationRenewal(threadId);
                }
            }
        });
        return ttlRemainingFuture;
    }

    private void scheduleExpirationRenewal(final long threadId) {
        if (expirationRenewalMap.containsKey(getEntryName())) {
            return;
        }

        Timeout task = commandExecutor.getConnectionManager().newTimeout(new TimerTask() {
            @Override
            public void run(Timeout timeout) throws Exception {
                
                RFuture<Boolean> future = commandExecutor.evalWriteAsync(getName(), LongCodec.INSTANCE, RedisCommands.EVAL_BOOLEAN,
                        "if (redis.call('hexists', KEYS[1], ARGV[2]) == 1) then " +
                            "redis.call('pexpire', KEYS[1], ARGV[1]); " +
                            "return 1; " +
                        "end; " +
                        "return 0;",
                          Collections.<Object>singletonList(getName()), internalLockLeaseTime, getLockName(threadId));
                
                future.addListener(new FutureListener<Boolean>() {
                    @Override
                    public void operationComplete(Future<Boolean> future) throws Exception {
                        expirationRenewalMap.remove(getEntryName());
                        if (!future.isSuccess()) {
                            log.error("Can't update lock " + getName() + " expiration", future.cause());
                            return;
                        }
                        
                        if (future.getNow()) {
                            // reschedule itself
                            scheduleExpirationRenewal(threadId);
                        }
                    }
                });
            }
        }, internalLockLeaseTime / 3, TimeUnit.MILLISECONDS);

        if (expirationRenewalMap.putIfAbsent(getEntryName(), task) != null) {
            task.cancel();
        }
    }

    
    ...
}

可以看到,最後加鎖的邏輯會進入到org.redisson.RedissonLock#tryAcquireAsync中,在獲取鎖成功後,會進入scheduleExpirationRenewal,這裏面初始化了一個定時器,dely的時間是internalLockLeaseTime / 3。在redisson中,internalLockLeaseTime是30s,也就是每隔10s續期一次,每次30s。 如果是基於zookeeper實現的分佈式鎖,可以利用zookeeper檢查節點是否存活,從而實現續期,zookeeper分佈式鎖沒用過,不詳細說。

不過這種做法也無法百分百做到同一時刻只有一個client獲取到鎖,如果續期失敗,比如發生了Martin Kleppmann所說的STW的GC,或者client和redis集羣失聯了,只要續期失敗,就會造成同一時刻有多個client獲得鎖了。在我的場景下,我將鎖的粒度拆小了,redisson的續期機制已經夠用了。 如果要做得更嚴格,得加一個續期失敗終止任務的邏輯。這種做法在以前Python的代碼中實現過,Java還沒有碰到這麼嚴格的情況。

這裏也提下Martin Kleppmann的解決方案,我自己覺得這個方案並不靠譜,原因後面會提到。 他的方案是讓加鎖的資源自己維護一套保證不會因加鎖失敗而導致多個client在同一時刻訪問同一個資源的情況。

在客戶端獲取鎖的同時,也獲取到一個資源的token,這個token是單調遞增的,每次在寫資源時,都檢查當前的token是否是較老的token,如果是就不讓寫。對於上面的場景,Client1獲取鎖的同時分配一個33的token,Client2獲取鎖的時候分配一個34的token,在client1 GC期間,Client2已經寫了資源,這時最大的token就是34了,client1 從GC中回來,再帶着33的token寫資源時,會因爲token過期被拒絕。這種做法需要資源那一邊提供一個token生成器。 對於這種fencing的方案,我有幾點問題:

  1. 無法保證事務。示意圖中畫的只有34訪問了storage,但是在實際場景中,可能出現在一個任務內多次訪問storage的情況,而且必須是原子的。如果client1帶着33token在GC前訪問過一次storage,然後發生了GC。client2獲取到鎖,帶着34的token也訪問了storage,這時兩個client寫入的數據是否還能保證數據正確?如果不能,那麼這種方案就有缺陷,除非storage自己有其他機制可以保證,比如事務機制;如果能,那麼這裏的token就是多餘的,fencing的方案就是多此一舉。
  2. 高併發場景不實用。因爲每次只有最大的token能寫,這樣storage的訪問就是線性的,在高併發場景下,這種方式會極大的限制吞吐量,而分佈式鎖也大多是在這種場景下用的,很矛盾的設計。
  3. 這是所有分佈式鎖的問題。這個方案是一個通用的方案,可以和Redlock用,也可以和其他的lock用。所以我理解僅僅是一個和Redlock無關的解決方案。

系統時鐘漂移

這個問題只是考慮過,但在實際項目中並沒有碰到過,因爲理論上是可能出現的,這裏也說下。 redis的過期時間是依賴系統時鐘的,如果時鐘漂移過大時會影響到過期時間的計算。

爲什麼系統時鐘會存在漂移呢?先簡單說下系統時間,linux提供了兩個系統時間:clock realtime和clock monotonic。clock realtime也就是xtime/wall time,這個時間時可以被用戶改變的,被NTP改變,gettimeofday拿的就是這個時間,redis的過期計算用的也是這個時間。 clock monotonic ,直譯過來時單調時間,不會被用戶改變,但是會被NTP改變。

最理想的情況時,所有系統的時鐘都時時刻刻和NTP服務器保持同步,但這顯然時不可能的。導致系統時鐘漂移的原因有兩個:

  1. 系統的時鐘和NTP服務器不同步。這個目前沒有特別好的解決方案,只能相信運維同學了。
  2. clock realtime被人爲修改。在實現分佈式鎖時,不要使用clock realtime。不過很可惜,redis使用的就是這個時間,我看了下Redis 5.0源碼,使用的還是clock realtime。Antirez說過改成clock monotonic的,不過大佬還沒有改。也就是說,人爲修改redis服務器的時間,就能讓redis出問題了。

總結

本文從一個簡單的基於redis的分佈式鎖出發,到更復雜的Redlock的實現,介紹了在使用分佈式鎖的過程中才踩過的一些坑以及解決方案。

粉絲福利

爲粉絲講解福利資源:特講解教程教你如何學習 ,**源碼、分佈式、微服務、性能優化、多線程併發,從0到1,**帶你領略底層精髓詳情點擊鏈接:https://shimo.im/docs/VqQR6tPrpR3C3tjq/

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章