ZooKeeper 分布式鎖 Curator 源碼 05：分布式讀寫鎖和聯(lián)鎖

作者：程序員小航 2021-07-16 07:57:34

開發(fā)

前端

分布式

分布式 Curator 同樣支持分布式讀寫鎖[1] 和聯(lián)鎖[2]，只需要使用 InterProcessReadWriteLock 即可，來一起看看它的源碼以及實現(xiàn)方式。

成都創(chuàng)新互聯(lián)公司專業(yè)為企業(yè)提供武義網(wǎng)站建設、武義做網(wǎng)站、武義網(wǎng)站設計、武義網(wǎng)站制作等企業(yè)網(wǎng)站建設、網(wǎng)頁設計與制作、武義企業(yè)網(wǎng)站模板建站服務，十載武義做網(wǎng)站經(jīng)驗，不只是建網(wǎng)站，更提供有價值的思路和整體網(wǎng)絡服務。

前言

Curator 同樣支持分布式讀寫鎖[1] 和聯(lián)鎖[2]，只需要使用 InterProcessReadWriteLock 即可，來一起看看它的源碼以及實現(xiàn)方式。

1.使用方式

  
 
 
 

   
  
  
  
public class CuratorDemo {
   
  
  
  

   
  
  
  
    public static void main(String[] args) throws Exception {
   
  
  
  

   
  
  
  
        String connectString = "127.0.0.1:2181,127.0.0.1:2182,127.0.0.1:2183";
   
  
  
  

   
  
  
  
        RetryPolicy retryPolicy = new ExponentialBackoffRetry(1000, 3);
   
  
  
  

   
  
  
  
        CuratorFramework client = CuratorFrameworkFactory
   
  
  
  
                .builder()
   
  
  
  
                .connectString(connectString)
   
  
  
  
                .retryPolicy(retryPolicy)
   
  
  
  
                .build();
   
  
  
  
        client.start();
   
  
  
  

   
  
  
  
  InterProcessReadWriteLock lock = new InterProcessReadWriteLock(client, "/locks/lock_01");
   
  
  
  
  lock.readLock().acquire();
   
  
  
  
  lock.readLock().release();
   
  
  
  
  lock.writeLock().acquire();
   
  
  
  
  lock.writeLock().release();
   
  
  
  

   
  
  
  
    }
   
  
  
  
}
  
 
 
 

2.源碼

讀鎖寫鎖都是基于 InterProcessMutex 實現(xiàn)的，所以基本都和 InterProcessMutex 沒有區(qū)別。不過這里生成的鎖名字不再是 -lock- 而是換成了 __WRIT__ 和 __READ__。

讀鎖加鎖節(jié)點名為 /locks/lock_01/_c_44a8eaf8-f177-403a-92bf-9119591b54d5-__READ__0000000000，寫鎖解鎖節(jié)點名為 _c_2e5dde98-c548-4f8b-a798-821ee8330eb6-__WRIT__0000000001。

其中創(chuàng)建節(jié)點時和可重入鎖 InterProcessMutex 沒有區(qū)別，唯一的區(qū)別就是在 internalLockLoop 方法中，判斷鎖獲取結果時有區(qū)別。

當可重入鎖時是在 StandardLockInternalsDriver#getsTheLock 判斷當前節(jié)點是否為最小節(jié)點。

而讀寫鎖是在 InterProcessReadWriteLock#InterProcessReadWriteLock 中重寫了 getsTheLock 方法。

讀鎖加鎖

讀鎖加鎖，當前線程直接返回成功，也就是說當前線程讀寫不互斥的。

如果是其他線程，則遍歷所有子節(jié)點。

子節(jié)點包含寫鎖，當前節(jié)點在子節(jié)點有序集合的索引小于寫鎖的索引則直接獲得鎖，否則獲取失敗;

子節(jié)點不包含寫鎖，則當前節(jié)點在子節(jié)點的有序集合的 index < Integer.MAX_VALUE (2147483647) 即可。

就是說讀鎖最多支持 2147483647 個。

寫鎖加鎖

寫鎖加鎖直接復用的可重入鎖 InterProcessMutex 的邏輯，所以這里寫鎖和寫鎖，以及讀鎖和寫鎖都是互斥的。

3.聯(lián)鎖

聯(lián)鎖的使用，就是將 InterProcessLock 放到集合中，然后進行統(tǒng)一加鎖。

加鎖就遍歷集合，依次進行加鎖。

4.總結

本文介紹了讀寫鎖和聯(lián)鎖，其實都是基于最基礎的可重入鎖進行封裝，理解了可重入鎖的概念，后面的簡單看下思想即可。

引用鏈接：

[1]Shared Reentrant Read Write Lock: https://curator.apache.org/curator-recipes/shared-reentrant-read-write-lock.html

[2]Multi Shared Lock: https://curator.apache.org/curator-recipes/multi-shared-lock.html

分享標題：ZooKeeper分布式鎖Curator源碼05：分布式讀寫鎖和聯(lián)鎖
分享鏈接：http://m.5511xx.com/article/cocjhcs.html