ZooKeeper 分布式锁 Curator 源码 05：分布式读写锁和联锁

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

开发

前端

分布式

分布式 Curator 同样支持分布式读写锁[1] 和联锁[2]，只需要使用 InterProcessReadWriteLock 即可，来一起看看它的源码以及实现方式。

[[411638]]

前言

Curator 同样支持分布式读写锁[1] 和联锁[2]，只需要使用 InterProcessReadWriteLock 即可，来一起看看它的源码以及实现方式。

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 实现的，所以基本都和 InterProcessMutex 没有区别。不过这里生成的锁名字不再是 -lock- 而是换成了 __WRIT__ 和 __READ__。

读锁加锁节点名为 /locks/lock_01/_c_44a8eaf8-f177-403a-92bf-9119591b54d5-__READ__0000000000，写锁解锁节点名为 _c_2e5dde98-c548-4f8b-a798-821ee8330eb6-__WRIT__0000000001。

其中创建节点时和可重入锁 InterProcessMutex 没有区别，唯一的区别就是在 internalLockLoop 方法中，判断锁获取结果时有区别。

当可重入锁时是在 StandardLockInternalsDriver#getsTheLock 判断当前节点是否为最小节点。

而读写锁是在 InterProcessReadWriteLock#InterProcessReadWriteLock 中重写了 getsTheLock 方法。

读锁加锁

读锁加锁，当前线程直接返回成功，也就是说当前线程读写不互斥的。

如果是其他线程，则遍历所有子节点。

子节点包含写锁，当前节点在子节点有序集合的索引小于写锁的索引则直接获得锁，否则获取失败;

子节点不包含写锁，则当前节点在子节点的有序集合的 index < Integer.MAX_VALUE (2147483647) 即可。

就是说读锁最多支持 2147483647 个。

写锁加锁

写锁加锁直接复用的可重入锁 InterProcessMutex 的逻辑，所以这里写锁和写锁，以及读锁和写锁都是互斥的。

3.联锁

联锁的使用，就是将 InterProcessLock 放到集合中，然后进行统一加锁。

加锁就遍历集合，依次进行加锁。

4.总结

本文介绍了读写锁和联锁，其实都是基于最基础的可重入锁进行封装，理解了可重入锁的概念，后面的简单看下思想即可。

引用链接：

[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