如何用zookeeper 实现分布式锁

一、分布式锁介绍

        分布式锁主要用于在分布式环境中保护跨进程、跨主机、跨网络的共享资源实现互斥访问,以达到保证数据的一致性。


二、分布式锁设计原理图

1、基于排它锁:



让所有的客户端(A B C)都在指定的Locks节点下 创建一个同名的节点lock,利用ZK 节点名称的唯一性来做限制, 谁创建成功了谁就获取到了锁,否则就是没有获取到。

这种设计比较简单

2、基于共享锁:


1、在当前节点下创建临时有序节点C,若创建成功则会返回节点C的名称。

2、获取当前节点下的所有临时节点。利用zk 节点的 顺序一致性,获取到了最小的节点就算获取到了锁。

3、找到临时节点中的最小节点的名称,和第一步创建节点返回的 C节点名称进行比较,如果匹配成功,表示当前节点就是最小节点,就意味着获取到了锁。

否则就要获取到比刚刚创建的节点顺序小的节点。

4、然后获取到比当前节点顺序小的所有节点的一个集合。如果不为空,就获取这个集合节点中的最后一个节点B。然后让B去监听它的上一个A,A的会话超时或者A节点被删除(释放)了,B就获取到了锁,同理,当B被删除或者会话超时时,C就获取到了锁。

5、执行完业务逻辑操作后  执行 删除节点的方法  锁就被释放了,然后下个节点就能获取到锁了。

也可以看看这两篇文章理一理:

zookeeper实现分布式锁原理

三种实现分布式锁的方式



Zookeeper实现分布式锁的一般步骤如下: 1. 在Zookeeper中创建一个临时节点,节点名称可以是锁的名称,节点数据可以是当前客户端的ID,表示该客户端获取了锁。 2. 客户端获取锁时,先检查是否已经存在该锁,如果不存在,则创建该锁;如果已经存在,则等待。 3. 当客户端释放锁时,删除该节点。 4. 其他客户端在创建节点时,如果发现该锁已经存在,则设置Watcher,等待上一个持有锁的客户端释放锁之后,重新尝试获取锁。 以下是一个简单的Java代码示例,演示了如何使用Zookeeper实现分布式锁: ``` public class DistributedLock { private static final String LOCK_BASE_PATH = "/mylock"; private static final String LOCK_NAME_PREFIX = "lock_"; private ZooKeeper zk; private String lockPath; public DistributedLock(String zkUrl) throws IOException, InterruptedException, KeeperException { this.zk = new ZooKeeper(zkUrl, 5000, null); createLockBasePath(); } private void createLockBasePath() throws KeeperException, InterruptedException { if (zk.exists(LOCK_BASE_PATH, false) == null) { zk.create(LOCK_BASE_PATH, new byte[0], ZooDefs.Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT); } } public void lock() throws KeeperException, InterruptedException { String path = zk.create(LOCK_BASE_PATH + "/" + LOCK_NAME_PREFIX, new byte[0], ZooDefs.Ids.OPEN_ACL_UNSAFE, CreateMode.EPHEMERAL_SEQUENTIAL); lockPath = path; while (true) { List<String> children = zk.getChildren(LOCK_BASE_PATH, false); String minChild = Collections.min(children); if (lockPath.endsWith(minChild)) { return; } else { String prevChild = children.get(children.indexOf(lockPath.substring(LOCK_BASE_PATH.length() + 1)) - 1); zk.exists(LOCK_BASE_PATH + "/" + prevChild, new LockWatcher()); } } } public void unlock() throws KeeperException, InterruptedException { zk.delete(lockPath, -1); } private class LockWatcher implements Watcher { @Override public void process(WatchedEvent event) { synchronized (this) { notifyAll(); } } } } ``` 在上述代码中,我们使用了ZooKeeper的EPHEMERAL_SEQUENTIAL节点类型来创建临时节点,并通过节点名称来实现锁。在获取锁时,会不断检查当前节点是否是最小的节点,如果不是,则等待上一个节点的Watcher通知,重新尝试获取锁。 需要注意的是,这只是一个简单的示例代码,实际应用中可能需要考虑更多的情况,比如节点的超时时间、异常处理等。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值