乐趣区

女朋友也能看懂的Zookeeper分布式锁原理

前言
关于分布式锁,在互联网行业的使用场景还是比较多的,比如电商的库存扣减,秒杀活动,集群定时任务执行等需要进程互斥的场景。而实现分布式锁的手段也很多,大家比较常见的就是 redis 跟 zookeeper,今天我们主要介绍的是基于 zookeeper 实现的分布式锁。
这篇文章主要借用 Curator 框架对 zk 分布式锁的实现思路,大家理解了以后完全可以自己手动实现一遍,但是在工作中还是建议使用成熟的开源框架,很多坑别人已经帮我们踩好了,除非万不得已,需要高度定制符合自己项目的需求的时候,才开始自行封装吧。
正文
zookeeper 简单介绍
既然是基于 zookeeper 的分布式锁,首先肯定要对这个 zookeeper 有一定了解,这里就不过多的进行讲解,只对其跟分布式锁有关联的特性做一个简单的介绍,更多详细的功能特性大家可以参阅官方文档。
zookeeper 维护着类似文件系统的数据结构,它总共有四种类型的节点

PERSISTENT:持久化的节点。一旦创建后,即使客户端与 zk 断开了连接,该节点依然存在。
PERSISTENT_SEQUENTIAL:持久化顺序编号节点。比 PERSISTENT 节点多了节点自动按照顺序编号。
EPHEMERAL:临时节点。当客户端与 zk 断开连接之后,该节点就被删除。
EPHEMERAL_SEQUENTIAL:临时顺序编号节点。比 EPHEMERAL 节点多了节点自动按照顺序编号。(分布式锁实现使用该节点类型)

Curator 实现分布式锁原理
好,当我们简单了解了 zk 的节点类型以后,现在正式的分析 Curator 分布式锁的实现原理。这里我们定义了一个“/curator_lock”锁节点用来存放相关客户端创建的临时顺序节点。
假设两个客户端 ClientA 跟 ClientB 同时去争夺一个锁,此时 ClientA 先行一步获得了锁,那么它将会在我们的 zk 上创建一个“/curator_lock/xxxxx-0000000000”的临时顺序节点。

接着它会拿到“/curator_lock/”锁节点下的所有子节点,因为这些节点是有序的,这时候会判断它所创建的节点是否排在第一位(也就是序号最小),由于 ClientA 是第一个创建节点的的客户端,必然是排在第一位,所以它也就拿到了锁。
[zk: localhost:2182(CONNECTED) 4] ls /curator_lock
[_c_f3f38067-8bff-47ef-9628-e638cfaad77e-lock-0000000000]
这个时候 ClientB 也来了,按照同样的步骤,先是在“/curator_lock/”下创建一个临时顺序节点“/curator_lock/xxxxx-0000000001”,接着也是获得该节点下的所有子节点信息,并比对自己生成的节点序号是否最小,由于此时序号最小的节点是 ClientA 创建的,并且还没释放掉,所以 ClientB 自己就拿不到锁。

[zk: localhost:2182(CONNECTED) 4] ls /curator_lock
[_c_2a8198e4-2039-4a3c-8606-39c65790d637-lock-0000000001,
_c_f3f38067-8bff-47ef-9628-e638cfaad77e-lock-0000000000]

既然 ClientB 拿不到锁,也不会放弃,它会对自己的前一个节点加上监听器(zk 提供的 api 实现),只要监听到前一个节点被删除了,也就是释放了锁,就会马上重新执行获取锁的操作。
当后面的 ClientC,ClientD… 过来的时候也是如此,变化的只是节点上的编号,它会根据 Client 连接的数量而不断增加。
可能大家还会担心,万一我的获取到锁的客户端宕机了怎么办,会不会不释放锁?其实上面已经解答了这个问题,由于 Curator 使用的是临时顺序节点来实现的分布式锁,只要客户端与 zk 连接断开,该节点也就消失了,相当于释放了锁。
下面代码展示了 Curator 的基本使用方法,仅作为参考实例,请勿在生产环境使用的这么随意。
CuratorFramework client = CuratorFrameworkFactory.newClient(“127.0.0.1:2182”,
5000,10000,
new ExponentialBackoffRetry(1000, 3));
client.start();
InterProcessMutex interProcessMutex = new InterProcessMutex(client, “/curator_lock”);
// 加锁
interProcessMutex.acquire();

// 业务逻辑

// 释放锁
interProcessMutex.release();
client.close();

总结
我们在搞懂了原理之后,就可以抛弃 Curator, 自己动手实现一个分布式锁了,相信大家实现基本的功能都是没问题的,但是要做到生产级别,可能还是要在细节上下功夫,比如说一些异常处理,性能优化等因素。

微信公众号《深夜里的程序猿》– 分享最干的干货

退出移动版