ZooKeeper 分布式锁 Curator 源码 05:分布式读写锁和联锁
前言
Curator 同样支持分布式读写锁 和联锁,只需要使用 InterProcessReadWriteLock 即可,来一起看看它的源码以及实现方式。
使用方式
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();
}
}
源码
读锁写锁都是基于 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
的逻辑,所以这里写锁和写锁,以及读锁和写锁都是互斥的。
联锁
联锁的使用,就是将 InterProcessLock
放到集合中,然后进行统一加锁。
加锁就遍历集合,依次进行加锁。
总结
本文介绍了读写锁和联锁,其实都是基于最基础的可重入锁进行封装,理解了可重入锁的概念,后面的简单看下思想即可。