zookeepr 实现分布式锁

hfmbook

于 2019-07-15 19:29:08 发布

阅读量144

点赞数

分类专栏： JAVA

本文链接：https://blog.csdn.net/hfmbook/article/details/96007646

版权

JAVA 专栏收录该内容

71 篇文章 1 订阅

订阅专栏

引入jar包

<dependency>
            <groupId>org.apache.curator</groupId>
            <artifactId>curator-framework</artifactId>
            <version>4.0.0</version>
        </dependency>

        <dependency>
            <groupId>org.apache.curator</groupId>
            <artifactId>curator-recipes</artifactId>
            <version>4.0.0</version>
        </dependency>

实现代码


import org.apache.curator.framework.CuratorFramework;
import org.apache.curator.framework.CuratorFrameworkFactory;
import org.apache.curator.framework.recipes.locks.InterProcessMutex;
import org.apache.curator.retry.ExponentialBackoffRetry;

import java.util.concurrent.CountDownLatch;

public class ZookeeperMain {

    public static final String ZK_CONNECTION = "127.0.0.1:2181";

    public static void main(String[] args) throws Exception{
        CuratorFramework curatorFramework = CuratorFrameworkFactory.builder().
                connectString(ZK_CONNECTION).sessionTimeoutMs(50000000).
                retryPolicy(new ExponentialBackoffRetry(1000, 3)).build();
        curatorFramework.start();
        final InterProcessMutex lock = new InterProcessMutex(curatorFramework, "/locks");

        int count = 10 ;
        CountDownLatch latch = new CountDownLatch(count);
        for (int i = 0; i < count; i++) {
            new Thread(() -> {
                System.out.println(Thread.currentThread().getName() + "->尝试竞争锁");
                try {
                    lock.acquire();
                    System.out.println(Thread.currentThread().getName() + "->成功获得了锁");

                    Thread.sleep( 40000 );
                } catch (Exception e) {
                    e.printStackTrace();
                } finally {
                    try {
                        latch.countDown();
                        lock.release();
                    } catch (Exception e) {
                    }
                }
            }, "Thread-" + i).start();
        }
        latch.await();
        curatorFramework.close();
    }
}