基于zookeeper实现分布式锁

@RunWith(SpringRunner.class)
@SpringBootTest
public class TestApplicationTests {

    private ZooKeeper zooKeeper;

    @Test
    public void contextLoads() {
        try {
            zooKeeper = new ZooKeeper("127.0.0.1:2181", 2000, watchedEvent -> System.out.println("已连接"));
            String looks = "/locks";
            Stat exists = zooKeeper.exists(looks, false);
            if (exists == null) {
                zooKeeper.create(looks, "locks".getBytes(), ZooDefs.Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT);
            }
                    
            for (int i = 0; i < 20; i++) {
                new Thread(() -> {

                    try {
                        String thisLock = zooKeeper.create(looks + "/lookName", "lock".getBytes(),
                                ZooDefs.Ids.OPEN_ACL_UNSAFE, CreateMode.EPHEMERAL_SEQUENTIAL);
                        String lockName = thisLock.substring(looks.length() + 1);
                        System.out.println(Thread.currentThread().getName() + "创建" + thisLock);
                        List<String> children = zooKeeper.getChildren(looks, false);
                        while (!getLock(children, looks, lockName)) {
                            CountDownLatch countDownLatch = new CountDownLatch(1);

                            for (int z = 1; z < children.size(); z++) {
                                if (children.get(z).equals(lockName)) {
                                    System.out.println("正在等待:"+looks + "/"+children.get(z+1));
                                     zooKeeper.exists(looks + "/" + children.get(z + 1), new Watcher() {
                                        @Override
                                        public void process(WatchedEvent watchedEvent) {
                                            countDownLatch.countDown();
                                        }
                                    });


                                }
                            }
                            countDownLatch.await();
                            children = zooKeeper.getChildren(looks, false);
                        }
                        //模拟处理工作
                        Thread.sleep(1000);
                        zooKeeper.delete(thisLock, -1);
                        System.out.println("释放锁! 线程名:" + Thread.currentThread().getName()+" "+thisLock);

                    } catch (Exception e) {
                        e.printStackTrace();
                    }


                }, "线程" + i).start();
            }


        } catch (Exception e) {
            e.printStackTrace();
        } finally {

        }
    }

    private boolean getLock(List<String> children, String looks, String thisLock) throws KeeperException, InterruptedException {

        if (children.get(children.size()-1).equals(thisLock)) {
            System.out.println("获取到锁! 线程名:" + Thread.currentThread().getName());
            return true;
        } else {
            System.out.println("未获取到获取到锁! 线程名:" + Thread.currentThread().getName());
        }

        return false;
    }

}

 

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

qq_37879809

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值