分布式系统基础-- 注册中心Zookeeper应用4（分布式锁）

阳光穿透心脏的1/2处 2022-03-25 09:08 191阅读 0赞

前面讲了zookeeper的基础，现在基于zookeeper来实现分布式锁。

### 基于ZooKeeper分布式锁的流程： ###

在zookeeper指定节点（locks）下创建跟节点 /locks

获取locks下所有子节点children(孩子节点是临时节点)

通过当前的节点和子节点中最小的节点进行比较，如果相等，表示获得锁成功

监听当前节点的上一个节点

### 具体实现 ###

1.利用watcher机制手动实现（核心代码）

@Override
        public boolean tryLock() {
    
            try {
                //创建临时有序节点
                CURRENT_LOCK = zk.create(ROOT_LOCK + "/", "0".getBytes(),
                        ZooDefs.Ids.OPEN_ACL_UNSAFE, CreateMode.EPHEMERAL_SEQUENTIAL);
                System.out.println(Thread.currentThread().getName() + "->" +
                        CURRENT_LOCK + "，尝试竞争锁");
                List<String> childrens = zk.getChildren(ROOT_LOCK, false); //获取根节点下的所有子节点
                SortedSet<String> sortedSet = new TreeSet();//定义一个集合进行排序
                for (String children : childrens) {
                    sortedSet.add(ROOT_LOCK + "/" + children);
                }
                String firstNode = sortedSet.first(); //获得当前所有子节点中最小的节点
                SortedSet<String> lessThenMe = ((TreeSet<String>) sortedSet).headSet(CURRENT_LOCK); //
                if (CURRENT_LOCK.equals(firstNode)) {//通过当前的节点和子节点中最小的节点进行比较，如果相等，表示获得锁成功
                    return true;
                }
                if (!lessThenMe.isEmpty()) {
                    WAIT_LOCK = lessThenMe.last();//获得比当前节点更小的最后一个节点，设置给WAIT_LOCK
                }
            } catch (KeeperException e) {
                e.printStackTrace();
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            return false;
        }

2.利用curator客户端方式实现(核心代码)

@Override
            public void run() {
                InterProcessMutex lock = new InterProcessMutex(client, "/distributeLock");
                try {
                    if (lock.acquire(120, TimeUnit.SECONDS)) {
                        try {
                            //System.out.println("----" + this.name + "获得资源");
                            System.out.println("-----" + this.name + "正在处理资源");
                            Thread.sleep(10 * 1000);
                            // System.out.println("-----" + this.name + "资源使用完毕-------");
                            latch.countDown();
                        } finally {
                            lock.release();
                            System.out.println("-------" + this.name + "释放----------");
                        }
                    }
                } catch (Exception e) {
                    e.printStackTrace();
                }
            }

### 代码github地址： ###

[https://github.com/zhangyu345293721/distributelock][https_github.com_zhangyu345293721_distributelock]

[https_github.com_zhangyu345293721_distributelock]: https://github.com/zhangyu345293721/distributelock