Zookeeper:实现“分布式锁”的 Demo_zookeeper实现分布式锁demo(1)

img
img

网上学习资料一大堆,但如果学到的知识不成体系,遇到问题时只是浅尝辄止,不再深入研究,那么很难做到真正的技术提升。

需要这份系统化资料的朋友,可以戳这里获取

一个人可以走的很快,但一群人才能走的更远!不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人,都欢迎加入我们的的圈子(技术交流、学习资源、职场吐槽、大厂内推、面试辅导),让我们一起学习成长!

分布式锁本质,就是多个资源竞争者对一份资源的排他占有

  • 我们设置多个线程,分别在同一 path 下创建节点
  • 没个线程获取当前 path 下子节点,看最小子节点是否为自身,是则加锁成功(更好的方式是用 Watcher 对前一个地址监控,这里图方便用子节点排序取最小的方式 )
  • 线程加锁成功后,执行任务,执行完毕后解锁
3、Demo 前提

参考:Mac通过Docker安装Zookeeper集群


二、代码实现

1、引用 Maven 依赖
        <!-- 选择对应的Zookeeper版本 -->
        <dependency>
            <groupId>org.apache.zookeeper</groupId>
            <artifactId>zookeeper</artifactId>
            <version>3.7.0</version>
        </dependency>

2、ConnectionWatcher 类创建 Zookeeper 连接
import java.io.IOException;
import java.util.concurrent.CountDownLatch;

import org.apache.zookeeper.WatchedEvent;
import org.apache.zookeeper.Watcher;
import org.apache.zookeeper.ZooKeeper;

public class ConnectionWatcher implements Watcher {

    private final CountDownLatch connectedSignal = new CountDownLatch(1);
    private static final int SESSION\_TIMEOUT = 5000;
    protected ZooKeeper zk;


    public void connect(String hosts) throws IOException, InterruptedException {
        zk = new ZooKeeper(hosts, SESSION\_TIMEOUT, this);
        connectedSignal.await();
    }

    @Override
    public void process(WatchedEvent event) {
        if (event.getState() == Event.KeeperState.SyncConnected) {
            connectedSignal.countDown();
        }
    }

    public void close() throws InterruptedException {
        zk.close();
    }

}

3、ActiveKeyValueStore 类读写 Zookeeper 数据
import java.nio.charset.Charset;
import java.nio.charset.StandardCharsets;
import java.util.List;

import org.apache.zookeeper.CreateMode;
import org.apache.zookeeper.KeeperException;
import org.apache.zookeeper.Watcher;
import org.apache.zookeeper.ZooDefs;
import org.apache.zookeeper.data.Stat;

public class ActiveKeyValueStore extends ConnectionWatcher {

    private static final Charset CHARSET = StandardCharsets.UTF\_8;

    int state = 0;

    /\*\*
 \* 写入节点数据
 \*
 \* @param path 节点地址
 \* @param value 数据值
 \* @throws InterruptedException 中断异常
 \* @throws KeeperException ZooKeeper异常
 \*/
    public void write(String path, String value) throws InterruptedException, KeeperException {
        Stat stat = zk.exists(path, false);
        if (stat == null) {
            if (value == null) {
                zk.create(path, null,
                        ZooDefs.Ids.OPEN\_ACL\_UNSAFE, CreateMode.PERSISTENT);
            } else {
                zk.create(path, value.getBytes(CHARSET),
                        ZooDefs.Ids.OPEN\_ACL\_UNSAFE, CreateMode.PERSISTENT);
            }

        } else {
            if (value == null) {
                zk.setData(path, null, -1);
            } else {
                zk.setData(path, value.getBytes(CHARSET), -1);
            }

        }
    }
  
   public boolean lock(String path, String name) throws InterruptedException, KeeperException {
        boolean flag = tryLock(path, name);
        if (flag) {
            state++;
        }
        return flag;
    }

    public boolean tryLock(String path, String name) throws InterruptedException, KeeperException {

        String lockPath = path + "/" + name;
        zk.create(lockPath, null, ZooDefs.Ids.OPEN\_ACL\_UNSAFE, CreateMode.EPHEMERAL);

        List<String> waits = readChildren(path, null);
        Collections.sort(waits);
        if (waits.get(0).equals(name)) {
            return true;
        }
        CountDownLatch latch = new CountDownLatch(1);
        for (int i = 0; i < waits.size(); i++) {
            String cur = waits.get(i);
            if (!cur.equalsIgnoreCase(name)) {
                continue;
            }
            String prePath = path + "/" + waits.get(i - 1);
            zk.exists(prePath, new Watcher() {
                @Override
                public void process(WatchedEvent event) {
                    latch.countDown();
                }
            });
            break;
        }
        latch.await();
        return true;
    }

    public boolean unlock(String path, String name) {
        if (state > 1) {
            state--;
            return true;
        }

        String lockPath = path + "/" + name;

        try {
            Stat stat = zk.exists(lockPath, false);
            int version = stat.getVersion();
            zk.delete(lockPath, version);
            state--;
            return true;
        } catch (Exception e) {
            System.out.println("unlock:" + lockPath + " ,exception,");
        }
        return false;
    }

    /\*\*
 \* 获取所有子节点
 \*
 \* @param path 节点地址
 \* @param watcher watcher
 \* @return 所有子节点
 \* @throws InterruptedException 中断异常
 \* @throws KeeperException ZooKeeper异常
 \*/
    public List<String> readChildren(String path, Watcher watcher) throws InterruptedException, KeeperException {
        List<String> childrens = null;
        if (watcher == null) {
            childrens = zk.getChildren(path, false);
        } else {
            childrens = zk.getChildren(path, watcher, null);
        }
        return childrens;
    }
}

4、ZkLock 类实现分布式锁
import lombok.SneakyThrows;
import org.apache.zookeeper.KeeperException;

import java.io.IOException;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;

public class ZkLock {


    /\*\*
 \* 开启的线程数,模拟多客户端操作
 \*/
    private static final int CLIENTS\_NUM = 3;

    private final ActiveKeyValueStore store;

    public ZkLock(String hosts) throws IOException, InterruptedException {


![img](https://img-blog.csdnimg.cn/img_convert/3676d0ce3d2eae084b2eab09b5e988bb.png)
![img](https://img-blog.csdnimg.cn/img_convert/fdc8062d601795efee5ec83fd51cac85.png)
![img](https://img-blog.csdnimg.cn/img_convert/3c4a246376a03a4e7ca8ef4c19391cda.png)

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

**由于文件比较多,这里只是将部分目录截图出来,全套包含大厂面经、学习笔记、源码讲义、实战项目、大纲路线、讲解视频,并且后续会持续更新**

**[需要这份系统化资料的朋友,可以戳这里获取](https://bbs.csdn.net/topics/618545628)**

片转存中...(img-qcG3zBey-1714981201132)]

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

**由于文件比较多,这里只是将部分目录截图出来,全套包含大厂面经、学习笔记、源码讲义、实战项目、大纲路线、讲解视频,并且后续会持续更新**

**[需要这份系统化资料的朋友,可以戳这里获取](https://bbs.csdn.net/topics/618545628)**

  • 7
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值