ZK(ZooKeeper)分布式锁实现

  • Znode监听: 每个线程在抢占所之前,会创建属于当前线程的ZNode节点,在释放锁的时候,会删除创建的ZNode,当我们创建的序号不是最小的时候,会等待watch通知,也就是上一个ZNode的状态通知,当前一个ZNode删除的时候,会触发回调机制,告诉下一个ZNode,你可以获取锁开始工作了

  • 临时节点自动删除:ZooKeeper还有一个好处,当我们客户端断开连接之后,我们出创建的临时节点会进行自动删除操作,所以我们在使用分布式锁的时候,一般都是会去创建临时节点,这样可以避免因为网络异常等原因,造成的死锁。

  • 羊群效应: ZooKeeper节点的顺序访问性,后面监听前面的方式,可以有效的避免 羊群效应,什么是羊群效应:当某一个节点挂掉了,所有的节点都要去监听,然后做出回应,这样会给服务器带来比较大压力,如果有了临时顺序节点,当一个节点挂掉了,只有它后面的那一个节点才做出反应。

我们现在看一下下面一张图:

在这里插入图片描述

在上图中,ZooKeeper里面有一把锁节点 testLock,这个锁就是ZooKeeper的一个节点,当两个客户端来获取这把锁的时候,会对ZooKeeper进行加锁的请求,也就是我们所说的 临时顺序节点

当我们在 /testLock目录下创建了一个顺序临时节点后,ZK会自动对这个临时节点维护 一个节点序号,并且这个节点是递增的,比如我们 clientA 创建了一个临时顺序节点,ZK内部会生成一个序号:/lock0000000001,那么 clientB 也生成了一个临时顺序节点,ZK会生成一个序号为 /lock0000000002,在这里数字都是依次递增的,从1开始递增,ZK内部会维护这个顺序。

下图所示:

在这里插入图片描述

这时候,ClientA会进行监听判断,在父节点下,我是不是最小的,如果是的话,那么俺就可以加锁了,因为我是最小的,其他的都比我大。我自己可以进行加锁,**你已经是一个成熟的临时节点了,要学会自己加锁。**咳,那么ZK是怎么进行判断的呢?宝贝,您往下看:

在这里插入图片描述

这个是cleintA已经加锁完成了,这个时候clientB也要过来加锁,那么他也要在/testLock,创建一个属于自己的临时节点,那么这个时候他的序号就会变成/lock0000000002,如下图所示:

在这里插入图片描述

这个时候就会出现我们前面所讲的,clientB 在加锁的时候会判断,自己是不是最小的,一看在当前父节点下不是最小的,啊~我还挺大的,还有比我小的!!!

加锁失败呀,咳咳,这个时候呢,clientB 就会去偷窥clientA,气氛逐渐暧昧起来,啊不是,是按照顺序去监听前一个节点(clientA),是否完成工作了,如果完成了,clientB才可以进行加锁工作,宝贝,你往下看图片:

在这里插入图片描述

clientA 加锁成功后,会进行自己的业务处理,当 clientA 处理完工作后,说我完事了,下一个,那么 clientA 是怎么完事的呢,他多长时间?不是,具体流程是怎样的?小农你不对劲,说什么呢!!!真羞涩

上面我们不是说了,当 clientB 加锁失败后,会给前一个节点(clientA)加上一个监听,当clientA被删除以后,就表示有人释放了锁,这个时候就会通知 clientB重新去获取锁。

在这里插入图片描述

这个时候clientB重新获取锁的时候,发现自己就是当前父节点下面最小的那个,于是clientB就开始加锁,开始工作等一系列操作,当clientB 完事以后,释放锁,也说了一句,下一个。

如下图所示:

在这里插入图片描述

当然除了 clientA、clientB还有C\D\E等,这字母看着好奇怪又好熟悉,原理都是一样的,都是最小节点进行解锁,如果不是,监听前一个节点是否释放,如果释放了,再次尝试加锁。如果前一节节点释放了,自己就是最小了,就排到前面去了,有点类似于 银行取号 的操作。

代码实现


使用ZooKeeper 创建临时顺序节点来实现分布式锁,大体的流程就是 先创建一个持久父节点,在当前节点下,创建临时顺序节点,找出最小的序列号,获取分布式锁,程序业务完成之后释放锁,通知下一个节点进行操作,使用的是watch来监控节点的变化,然后依次下一个最小序列节点进行操作。

首先我们需要创建一个持久父类节点:我这里是 /mxn

在这里插入图片描述

WatchCallBack

import org.apache.zookeeper.*;

import org.apache.zookeeper.data.Stat;

import java.util.Collections;

import java.util.List;

import java.util.concurrent.CountDownLatch;

/**

  • @program: mxnzookeeper

  • @ClassName WatchCallBack

  • @description:

  • @author: 微信搜索:牧小农

  • @create: 2021-10-23 10:48

  • @Version 1.0

**/

public class WatchCallBack implements Watcher, AsyncCallback.StringCallback ,AsyncCallback.Children2Callback ,AsyncCallback.StatCallback {

ZooKeeper zk ;

String threadName;

CountDownLatch cc = new CountDownLatch(1);

String pathName;

public String getPathName() {

return pathName;

}

public void setPathName(String pathName) {

this.pathName = pathName;

}

public String getThreadName() {

return threadName;

}

public void setThreadName(String threadName) {

this.threadName = threadName;

}

public ZooKeeper getZk() {

return zk;

}

public void setZk(ZooKeeper zk) {

this.zk = zk;

}

/** @Author 牧小农

  • @Description //TODO 尝试加锁方法

  • @Date 16:14 2021/10/24

  • @Param

  • @return

**/

public void tryLock(){

try {

System.out.println(threadName + " 开始创建。。。。");

//创建一个顺序临时节点

zk.create(“/lock”,threadName.getBytes(), ZooDefs.Ids.OPEN_ACL_UNSAFE, CreateMode.EPHEMERAL_SEQUENTIAL,this,“abc”);

//阻塞当前,监听前一个节点是否释放锁

cc.await();

} catch (InterruptedException e) {

e.printStackTrace();

}

}

/** @Author 牧小农

  • @Description //TODO 解锁方法

  • @Date 16:14 2021/10/24

  • @Param

  • @return

**/

public void unLock(){

try {

//释放锁,删除临时节点

zk.delete(pathName,-1);

//结束工作

System.out.println(threadName + " 结束工作了…");

} catch (InterruptedException e) {

e.printStackTrace();

} catch (KeeperException e) {

e.printStackTrace();

}

}

@Override

public void process(WatchedEvent event) {

//如果第一个节点释放了锁,那么第二个就会收到回调

//告诉它前一个节点释放了,你可以开始尝试获取锁

switch (event.getType()) {

case None:

break;

case NodeCreated:

break;

case NodeDeleted:

//当前节点重新获取锁

zk.getChildren(“/”,false,this ,“sdf”);

break;

case NodeDataChanged:

break;

case NodeChildrenChanged:

break;

}

}

@Override

public void processResult(int rc, String path, Object ctx, String name) {

if(name != null ){

System.out.println(threadName +" 线程创建了一个节点为 : " + name );

pathName = name ;

//监听前一个节点

zk.getChildren(“/”,false,this ,“sdf”);

}

}

//getChildren call back

@Override

public void processResult(int rc, String path, Object ctx, List children, Stat stat) {

//节点按照编号,升序排列

Collections.sort(children);

//对节点进行截取例如 /lock0000000022 截取后就是 lock0000000022

int i = children.indexOf(pathName.substring(1));

//是不是第一个,也就是说是不是最小的

if(i == 0){

//是第一个

System.out.println(threadName +" 现在我是最小的…");

try {

zk.setData(“/”,threadName.getBytes(),-1);

cc.countDown();

} catch (KeeperException e) {

e.printStackTrace();

} catch (InterruptedException e) {

e.printStackTrace();

}

}else{

//不是第一个

//监听前一个节点 看它是不是完成了工作进行释放锁了

zk.exists(“/”+children.get(i-1),this,this,“sdf”);

}

}

@Override

public void processResult(int rc, String path, Object ctx, Stat stat) {

//判断是否失败exists

}

}

TestLock

import com.mxn.zookeeper.config.ZKUtils;

import org.apache.zookeeper.ZooKeeper;

import org.junit.After;

import org.junit.Before;

import org.junit.Test;

/**

  • @program: mxnzookeeper

  • @ClassName TestLock

  • @description:

  • @author: 微信搜索:牧小农

  • @create: 2021-10-23 10:45

  • @Version 1.0

**/

public class TestLock {

ZooKeeper zk ;

@Before

public void conn (){

zk = ZKUtils.getZK();

}

@After

public void close (){

try {

zk.close();

} catch (InterruptedException e) {

e.printStackTrace();

}

}

@Test

public void lock(){

//创建十个线程

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

new Thread(){

@Override

public void run() {

WatchCallBack watchCallBack = new WatchCallBack();

watchCallBack.setZk(zk);

String threadName = Thread.currentThread().getName();

watchCallBack.setThreadName(threadName);

//线程进行抢锁操作

watchCallBack.tryLock();

try {

//进行业务逻辑处理

System.out.println(threadName+" 开始处理业务逻辑了…");

Thread.sleep(200);

}catch (Exception e){

e.printStackTrace();

}

//释放锁

watchCallBack.unLock();

}
自我介绍一下,小编13年上海交大毕业,曾经在小公司待过,也去过华为、OPPO等大厂,18年进入阿里一直到现在。

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

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

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

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

如果你觉得这些内容对你有帮助,可以扫码获取!!(备注Java获取)

img

最后我们该如何学习?

1、看视频进行系统学习

这几年的Crud经历,让我明白自己真的算是菜鸡中的战斗机,也正因为Crud,导致自己技术比较零散,也不够深入不够系统,所以重新进行学习是很有必要的。我差的是系统知识,差的结构框架和思路,所以通过视频来学习,效果更好,也更全面。关于视频学习,个人可以推荐去B站进行学习,B站上有很多学习视频,唯一的缺点就是免费的容易过时。

另外,我自己也珍藏了好几套视频资料躺在网盘里,有需要的我也可以分享给你:

1年半经验,2本学历,Curd背景,竟给30K,我的美团Offer终于来了

2、读源码,看实战笔记,学习大神思路

“编程语言是程序员的表达的方式,而架构是程序员对世界的认知”。所以,程序员要想快速认知并学习架构,读源码是必不可少的。阅读源码,是解决问题 + 理解事物,更重要的:看到源码背后的想法;程序员说:读万行源码,行万种实践。

Spring源码深度解析:

1年半经验,2本学历,Curd背景,竟给30K,我的美团Offer终于来了

Mybatis 3源码深度解析:

1年半经验,2本学历,Curd背景,竟给30K,我的美团Offer终于来了

Redis学习笔记:

1年半经验,2本学历,Curd背景,竟给30K,我的美团Offer终于来了

Spring Boot核心技术-笔记:

1年半经验,2本学历,Curd背景,竟给30K,我的美团Offer终于来了

3、面试前夕,刷题冲刺

面试的前一周时间内,就可以开始刷题冲刺了。请记住,刷题的时候,技术的优先,算法的看些基本的,比如排序等即可,而智力题,除非是校招,否则一般不怎么会问。

关于面试刷题,我个人也准备了一套系统的面试题,帮助你举一反三:

1年半经验,2本学历,Curd背景,竟给30K,我的美团Offer终于来了

只有技术过硬,在哪儿都不愁就业,“万般带不去,唯有业随身”学习本来就不是在课堂那几年说了算,而是在人生的旅途中不间断的事情。

人生短暂,别稀里糊涂的活一辈子,不要将就。
《互联网大厂面试真题解析、进阶开发核心学习笔记、全套讲解视频、实战项目源码讲义》点击传送门即可获取!
qF-1713380086207)]

Mybatis 3源码深度解析:

[外链图片转存中…(img-GW8rq39G-1713380086208)]

Redis学习笔记:

[外链图片转存中…(img-qJqKSmoY-1713380086208)]

Spring Boot核心技术-笔记:

[外链图片转存中…(img-1bqkuOs0-1713380086208)]

3、面试前夕,刷题冲刺

面试的前一周时间内,就可以开始刷题冲刺了。请记住,刷题的时候,技术的优先,算法的看些基本的,比如排序等即可,而智力题,除非是校招,否则一般不怎么会问。

关于面试刷题,我个人也准备了一套系统的面试题,帮助你举一反三:

[外链图片转存中…(img-naGLhaVT-1713380086208)]

只有技术过硬,在哪儿都不愁就业,“万般带不去,唯有业随身”学习本来就不是在课堂那几年说了算,而是在人生的旅途中不间断的事情。

人生短暂,别稀里糊涂的活一辈子,不要将就。
《互联网大厂面试真题解析、进阶开发核心学习笔记、全套讲解视频、实战项目源码讲义》点击传送门即可获取!

  • 10
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值