zookeeper实现分布式锁-curator

最新推荐文章于 2024-05-14 06:43:20 发布

2401_83329718

最新推荐文章于 2024-05-14 06:43:20 发布

阅读量795

点赞数 14

分类专栏：程序员文章标签：分布式 zookeeper 云原生

本文链接：https://blog.csdn.net/2401_83329718/article/details/137897619

版权

程序员专栏收录该内容

726 篇文章 0 订阅

订阅专栏

2.Java封装基于ZK的分布式锁

2.1.创建项目导入依赖

org.springframework.boot

spring-boot-starter-parent

2.0.5.RELEASE

org.springframework.boot

spring-boot-starter-web

2.0.5.RELEASE

org.apache.zookeeper

zookeeper

3.4.9

org.projectlombok

lombok

1.18.12

provided

junit

2.2.封装zookeeper分布式锁

/**======================================================================================

方法描述：zookeeper分布式锁
逻辑步骤：

实现 AutoCloseable 可自动关闭资源

Watcher观察器

======================================================================================*/

//zookeeper分布式锁

//实现 AutoCloseable 可自动关闭资源

//Watcher观察器

@Slf4j

public class ZookeeperLock implements AutoCloseable, Watcher {

//zookeeper客户端对象

private ZooKeeper zooKeeper;

//当前节点

private String currentNode;

//通过构造器初始化zookeeper

ZookeeperLock(){

try {

//参数：链接字符串，会话超时，监听器

this.zooKeeper = new ZooKeeper(“172.16.2.54:2181”,10000,this);

} catch (IOException e) {

e.printStackTrace();

}

/**======================================================================================

方法描述：提供获取锁的方法: code作为业务相关的编码，比如传入订单ID，就锁住这个订单
逻辑步骤：

1.创建根节点

2.进来一个线程，就为线程在zookeeper的根节点创建一个瞬时有序节点

======================================================================================*/

public boolean getLock(String code){

try {

//1.创建根节点，如果不存在，stat为null 就创建

String path = “/” + code;

Stat exists = zooKeeper.exists(path, false);

//创建节点

if(null == exists){

// 节点路径，节点值，权限：不需要账户密码就能链接，创建模式：顺序临时节点

zooKeeper.create(path,code.getBytes() , ZooDefs.Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT);

}

//2.进来一个线程，就为线程在zookeeper创建一个瞬时有序节点，如： /xx/xx000000001 ；/xx/xx000000002

//把当前节点保存为成员变量，后面用来做判断

currentNode = zooKeeper.create(path+path,code.getBytes() , ZooDefs.Ids.OPEN_ACL_UNSAFE, CreateMode.EPHEMERAL_SEQUENTIAL);

//3.对所有的子节点进行排序

List children = zooKeeper.getChildren(path, false);

Collections.sort(children);

//4.如果排序过的children中的第一个和 currentNode 一致，拿到锁

String firstNode = children.get(0);

if(currentNode.endsWith(firstNode)){

return true;

}

//5.如果不是第一个节点，需要监听前一个节点

//用一个临时变量记录当前节点的上一个节点

String previousNode = firstNode;

for(String node : children){

if(currentNode.endsWith(node)){

//如果当前节点是node节点，那么就监听它的上一个节点：比如 currentNode 这里是 0003节点，那 node就是 0002节点

//第一个参数是监听的节点，第二个参数是是否要监听，zooKeeper在初始化的时候设置好了监听器

log.info(“监听上一个节点：{}”,node);

zooKeeper.exists(path+“/”+previousNode,true);

}else{

//把children中的节点复制给上一个节点

previousNode = node;

}

//代码到这里，节点已经做好监听了，只需要等待，等待上一个节点完成工作后唤醒他

synchronized (this){

//wait会释放锁

wait();

}

//到这里说明被唤醒，说明获取到锁

log.info(“拿到锁：{}”,currentNode);

return true;

} catch (KeeperException e) {

e.printStackTrace();

} catch (InterruptedException e) {

e.printStackTrace();

}

return false;

}

@Override

public void close() {

//释放锁：节点路径，节点版本号(-1匹配所有版本)

log.info(“释放节点：{}”,currentNode);

if(null != currentNode){

try {

zooKeeper.delete(currentNode ,-1);

} catch (InterruptedException e) {

e.printStackTrace();

} catch (KeeperException e) {

e.printStackTrace();

}

@Override

public void process(WatchedEvent watchedEvent) {

//当节点被释放，就会到这里被监听到

if(watchedEvent.getType() == Event.EventType.NodeDeleted){

synchronized (this){

//唤醒等待的线程

log.info(“当前节点：{}，唤醒”,watchedEvent.getPath());

notify();

}

2.3.测试代码

public class ZKTest {

@Test

public void testZK() throws Exception {

for(int i = 0 ; i < 10 ; i++){

new Thread(()->{

ZookeeperLock zookeeperLock = new ZookeeperLock();

boolean getLock = zookeeperLock.getLock(“order”);

System.out.println(“是否获取到锁：”+getLock);

zookeeperLock.close();

}).start();

}

Thread.sleep(5000);

}

打印日志：

09:18:24.500 [Thread-0] INFO cn.itsource.ZookeeperLock - /order/order0000000014监听上一个节点：/order/order0000000013

09:18:24.500 [Thread-9] INFO cn.itsource.ZookeeperLock - /order/order0000000017监听上一个节点：/order/order0000000016

09:18:24.500 [Thread-1] INFO cn.itsource.ZookeeperLock - /order/order0000000019监听上一个节点：/order/order0000000018

09:18:24.500 [Thread-8] INFO cn.itsource.ZookeeperLock - /order/order0000000018监听上一个节点：/order/order0000000017

09:18:24.500 [Thread-4] INFO cn.itsource.ZookeeperLock - /order/order0000000016监听上一个节点：/order/order0000000015

09:18:24.500 [Thread-2] INFO cn.itsource.ZookeeperLock - /order/order0000000020监听上一个节点：/order/order0000000019

09:18:24.500 [Thread-3] INFO cn.itsource.ZookeeperLock - /order/order0000000013监听上一个节点：/order/order0000000012

09:18:24.500 [Thread-7] INFO cn.itsource.ZookeeperLock - 释放节点：/order/order0000000011

09:18:24.500 [Thread-6] INFO cn.itsource.ZookeeperLock - /order/order0000000015监听上一个节点：/order/order0000000014

09:18:24.500 [Thread-5] INFO cn.itsource.ZookeeperLock - /order/order0000000012监听上一个节点：/order/order0000000011

09:18:24.509 [Thread-5-EventThread] INFO cn.itsource.ZookeeperLock - 当前节点：/order/order0000000011，唤醒

09:18:24.509 [Thread-5] INFO cn.itsource.ZookeeperLock - 拿到锁：/order/order0000000012

是否获取到锁：true

09:18:24.509 [Thread-5] INFO cn.itsource.ZookeeperLock - 释放节点：/order/order0000000012

09:18:24.511 [Thread-3-EventThread] INFO cn.itsource.ZookeeperLock - 当前节点：/order/order0000000012，唤醒

09:18:24.512 [Thread-3] INFO cn.itsource.ZookeeperLock - 拿到锁：/order/order0000000013

是否获取到锁：true

09:18:24.512 [Thread-3] INFO cn.itsource.ZookeeperLock - 释放节点：/order/order0000000013

09:18:24.514 [Thread-0-EventThread] INFO cn.itsource.ZookeeperLock - 当前节点：/order/order0000000013，唤醒

09:18:24.515 [Thread-0] INFO cn.itsource.ZookeeperLock - 拿到锁：/order/order0000000014

是否获取到锁：true

09:18:24.515 [Thread-0] INFO cn.itsource.ZookeeperLock - 释放节点：/order/order0000000014

09:18:24.517 [Thread-6-EventThread] INFO cn.itsource.ZookeeperLock - 当前节点：/order/order0000000014，唤醒

09:18:24.517 [Thread-6] INFO cn.itsource.ZookeeperLock - 拿到锁：/order/order0000000015

是否获取到锁：true

09:18:24.517 [Thread-6] INFO cn.itsource.ZookeeperLock - 释放节点：/order/order0000000015

09:18:24.520 [Thread-4-EventThread] INFO cn.itsource.ZookeeperLock - 当前节点：/order/order0000000015，唤醒

09:18:24.520 [Thread-4] INFO cn.itsource.ZookeeperLock - 拿到锁：/order/order0000000016

是否获取到锁：true

09:18:24.520 [Thread-4] INFO cn.itsource.ZookeeperLock - 释放节点：/order/order0000000016

09:18:24.522 [Thread-9] INFO cn.itsource.ZookeeperLock - 拿到锁：/order/order0000000017

是否获取到锁：true

09:18:24.522 [Thread-9] INFO cn.itsource.ZookeeperLock - 释放节点：/order/order0000000017

09:18:24.525 [Thread-8-EventThread] INFO cn.itsource.ZookeeperLock - 当前节点：/order/order0000000017，唤醒

09:18:24.525 [Thread-8] INFO cn.itsource.ZookeeperLock - 拿到锁：/order/order0000000018

是否获取到锁：true

09:18:24.525 [Thread-8] INFO cn.itsource.ZookeeperLock - 释放节点：/order/order0000000018

09:18:24.527 [Thread-1-EventThread] INFO cn.itsource.ZookeeperLock - 当前节点：/order/order0000000018，唤醒

09:18:24.527 [Thread-1] INFO cn.itsource.ZookeeperLock - 拿到锁：/order/order0000000019

是否获取到锁：true

09:18:24.527 [Thread-1] INFO cn.itsource.ZookeeperLock - 释放节点：/order/order0000000019

09:18:24.530 [Thread-2-EventThread] INFO cn.itsource.ZookeeperLock - 当前节点：/order/order0000000019，唤醒

09:18:24.530 [Thread-2] INFO cn.itsource.ZookeeperLock - 拿到锁：/order/order0000000020

是否获取到锁：true

09:18:24.530 [Thread-2] INFO cn.itsource.ZookeeperLock - 释放节点：/order/order0000000020

这个效果看起来是对的。但是如果自己基于zookeeper封装分布式锁未免太过麻烦，而且容易出BUG，Apache提供了一个基于Zookeeper的客户端工具curator已经实现了分布式锁的封装，我们使用它就可以了。

3.使用curator实现分布式锁

3.1.curator介绍

Apache Curator是用于Apache ZooKeeper（一种分布式协调服务）的Java / JVM客户端库。它包括一个高级API框架和实用程序，使使用Apache ZooKeeper变得更加轻松和可靠。解决了很多Zookeeper客户端非常底层的细节开发工作，包括连接重连、反复注册Watcher和NodeExistsException异常等等。

重点是它对分布式锁进行了封装：http://curator.apache.org/getting-started.html

3.1.导入依赖

org.apache.curator

curator-recipes

4.1.0

3.2.配置curator

//初始化方法start

@Bean(initMethod = “start”,destroyMethod = “close”)

public CuratorFramework curatorFramework(){

//重试策略

RetryPolicy retryPolicy = new ExponentialBackoffRetry(1000, 3);

//创建客户端

CuratorFramework client = CuratorFrameworkFactory.newClient(“172.16.2.54:2181”, retryPolicy);

return client;

}

3.3.分布式锁案例

@Slf4j

@RunWith(SpringRunner.class)

@SpringBootTest(classes = AppStart.class)

public class ZKTest {

@Autowired

private CuratorFramework curatorFramework;

/**======================================================================================

方法描述：使用curator实现分布式锁

======================================================================================*/

@Test

public void testCurator() throws Exception {

for(int i = 0 ; i < 10 ; i++){

new Thread(()->{

testCuratorLock();

}).start();

}

Thread.sleep(5000);

}

public void testCuratorLock(){

//分布式锁

InterProcessMutex lock = new InterProcessMutex(curatorFramework, “/order”);

try {

if ( lock.acquire(1, TimeUnit.SECONDS) ){

//处理业务逻辑

log.info(“获取到锁”);

}

} catch (Exception e) {

e.printStackTrace();

} finally{

try {

//释放锁

log.info(“释放锁”);

lock.release();

} catch (Exception e) {

e.printStackTrace();

}

打印效果

2021-05-15 09:54:32.251 INFO 22488 — [ Thread-4] cn.itsource.ZKTest : 获取到锁

2021-05-15 09:54:32.252 INFO 22488 — [ Thread-4] cn.itsource.ZKTest : 释放锁
自我介绍一下，小编13年上海交大毕业，曾经在小公司待过，也去过华为、OPPO等大厂，18年进入阿里一直到现在。

深知大多数Java工程师，想要提升技能，往往是自己摸索成长或者是报班学习，但对于培训机构动则几千的学费，着实压力不小。自己不成体系的自学效果低效又漫长，而且极易碰到天花板技术停滞不前！

因此收集整理了一份《2024年Java开发全套学习资料》，初衷也很简单，就是希望能够帮助到想自学提升又不知道该从何学起的朋友，同时减轻大家的负担。

既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，基本涵盖了95%以上Java开发知识点，真正体系化！

由于文件比较大，这里只是将部分目录截图出来，每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频，并且会持续更新！

如果你觉得这些内容对你有帮助，可以扫码获取！！（备注Java获取）

总结

三个工作日收到了offer，头条面试体验还是很棒的，这次的头条面试好像每面技术都问了我算法，然后就是中间件、MySQL、Redis、Kafka、网络等等。

第一个是算法

关于算法，我觉得最好的是刷题，作死的刷的，多做多练习，加上自己的理解，还是比较容易拿下的。

而且，我貌似是将《算法刷题LeetCode中文版》、《算法的乐趣》大概都过了一遍，尤其是这本

《算法刷题LeetCode中文版》总共有15个章节：编程技巧、线性表、字符串、栈和队列、树、排序、查找、暴力枚举法、广度优先搜索、深度优先搜索、分治法、贪心法、动态规划、图、细节实现题

最新出炉，头条三面技术四面HR，看我如何一步一步攻克面试官？

《算法的乐趣》共有23个章节：

最新出炉，头条三面技术四面HR，看我如何一步一步攻克面试官？

第二个是Redis、MySQL、kafka（给大家看下我都有哪些复习笔记）

基本上都是面试真题解析、笔记和学习大纲图，感觉复习也就需要这些吧（个人意见）

最新出炉，头条三面技术四面HR，看我如何一步一步攻克面试官？

第三个是网络（给大家看一本我之前得到的《JAVA核心知识整理》包括30个章节分类，这本283页的JAVA核心知识整理还是很不错的，一次性总结了30个分享的大知识点）

最新出炉，头条三面技术四面HR，看我如何一步一步攻克面试官？

《互联网大厂面试真题解析、进阶开发核心学习笔记、全套讲解视频、实战项目源码讲义》点击传送门即可获取！
解，还是比较容易拿下的。

而且，我貌似是将《算法刷题LeetCode中文版》、《算法的乐趣》大概都过了一遍，尤其是这本

[外链图片转存中…(img-Vatxv249-1713380977857)]

《算法的乐趣》共有23个章节：

[外链图片转存中…(img-kgMWLW0w-1713380977857)]

[外链图片转存中…(img-iEwm81OW-1713380977857)]

第二个是Redis、MySQL、kafka（给大家看下我都有哪些复习笔记）

基本上都是面试真题解析、笔记和学习大纲图，感觉复习也就需要这些吧（个人意见）

[外链图片转存中…(img-tIiRirFo-1713380977858)]

第三个是网络（给大家看一本我之前得到的《JAVA核心知识整理》包括30个章节分类，这本283页的JAVA核心知识整理还是很不错的，一次性总结了30个分享的大知识点）

[外链图片转存中…(img-7IFzks5f-1713380977858)]

《互联网大厂面试真题解析、进阶开发核心学习笔记、全套讲解视频、实战项目源码讲义》点击传送门即可获取！

2401_83329718

关注

14
点赞
踩
22

收藏

觉得还不错? 一键收藏
0
评论
zookeeper实现分布式锁-curator

Apache Curator是用于Apache ZooKeeper（一种分布式协调服务）的Java / JVM客户端库。它包括一个高级API框架和实用程序，使使用Apache ZooKeeper变得更加轻松和可靠。解决了很多Zookeeper客户端非常底层的细节开发工作，包括连接重连、反复注册Watcher和NodeExistsException异常等等。重点是它对分布式锁进行了封装：http://curator.apache.org/getting-started.html。
复制链接

扫一扫