使用zookeeper临时顺序节点实现分布式锁

目录

 

1、zookeeper节点特点

2、分布式锁实现

2.1 lock自定义

2.2 抽象方法,使用模板模式定义好获取锁的方法

2.3 具体的工具类锁

2.4 使用测试类

2.5 输出结果


1、zookeeper节点特点

  • 临时节点(EPHEMERAL) :session链接断开就没了;不能创建子节点;不能同名
  • 临时顺序节点(EPHEMERAL_SEQUENTIAL) :session链接断开就没了;不能创建子节点;同名节点会在后面添加上序号(分布式锁使用的好处)
  • 持久节点(PERSISTENT):session断联、服务端重启还在;可以创建子节点,子节点可以临时也可以持久;不能同名
  • 持久顺序节点(PERSISTENT_SEQUENTIAL):session断联、服务端重启还在;可以创建子节点,子节点可以临时也可以持久;同名节点会在后面添加上序号

1.1 图解

2、分布式锁实现

参考ReentrantLock实现的Lock接口,主要有tryLock(),lock(),unLock()方法

2.1 lock自定义

public interface Lock {
    /**
     * 获取锁
     */
    void getLock();

    /**
     * 释放锁
     */
    void unLock();
}

2.2 抽象方法,使用模板模式定义好获取锁的方法

public abstract class AbstractZkLock implements Lock {
    //使用了模板方法,开放使用的只有getLock和unLock方法
    @Override
    public void getLock() {
        if (tryLock()) {
            System.out.println("获取锁");
        } else {
            //未获取到锁等待
            waitLock();
            //获取到锁之后,唤醒
            getLock();
        }
    }

    /**
     * ②占有锁
     *
     * @return 获取锁返回true
     */
    public abstract boolean tryLock();

    /**
     * 等待
     */
    public abstract void waitLock();
}

2.3 具体的工具类锁

public class ZkDistributeLock extends AbstractZkLock {
    //连接zk服务器
    private ZkClient zkClient = new ZkClient("127.0.0.1:21810");
    //作为锁的父节点
    private static final String PATH_LOCK = "/zk_path_lock";

    //要获取锁的前一个节点
    private String beforePath = null;
    //当前要获取锁的节点
    private String currentPath = null;
    private CountDownLatch countDownLatch;

    ZkDistributeLock() {
        //先创建一个持久节点作为分布式锁的父节点
        if (!zkClient.exists(PATH_LOCK)) {
            zkClient.createPersistent(PATH_LOCK, PATH_LOCK);
        }
    }

    @Override
    public boolean tryLock() {
        //首次进来肯定是空的
        if (currentPath == null || currentPath.length() == 0) {
            //currentpath = /zk_path_lock/0000000144
            currentPath = zkClient.createEphemeralSequential(PATH_LOCK + "/", PATH_LOCK);
        }
        //获取所有临时节点并排序0000000196,0000000196,0000000196
        List<String> children = zkClient.getChildren(PATH_LOCK);
        //对所有子节点排序
        Collections.sort(children);
        //判断当前节点是不是第一个节点
        if (currentPath.equals(PATH_LOCK + "/" + children.get(0))) {
            return true;
        } else {
            //找到当前节点的下标
            System.out.println(children.toString());
            //currentPath要去掉前面的PATH_LOCK + "/"字符串
            int wz = Collections.binarySearch(children, currentPath.substring((PATH_LOCK + "/").length()));
            System.out.println(wz);
            beforePath = PATH_LOCK + '/' + children.get(wz - 1);
        }
        return false;
    }

    @Override
    public void waitLock() {
        IZkDataListener zkDataListener = new IZkDataListener() {
            @Override
            public void handleDataChange(String s, Object o) throws Exception {

            }

            @Override
            public void handleDataDeleted(String s) throws Exception {
                if (countDownLatch != null) {
                    countDownLatch.countDown();
                }
            }
        };
        //给排在前面的的节点增加数据删除的watcher,本质是启动另外一个线程去监听前置节点
        zkClient.subscribeDataChanges(beforePath, zkDataListener);
        try {
            if (zkClient.exists(beforePath)) {
                countDownLatch = new CountDownLatch(1);
                countDownLatch.await();
            }

        } catch (InterruptedException e) {
            e.printStackTrace();
        }
        zkClient.unsubscribeDataChanges(beforePath, zkDataListener);
    }

    @Override
    public void unLock() {
        zkClient.delete(currentPath);
        //每次实例化的时候,都会创建一个连接,所以要关闭
        zkClient.close();
    }
}

2.4 使用测试类

public class TestMain extends Thread {
    private ZkDistributeLock lock = new ZkDistributeLock();
    //全局订单id
    private static int count = 0;
    SimpleDateFormat dateFormat = new SimpleDateFormat("yyyy-MM-dd-HH-mm-ss");

    public static void main(String[] args) {
        for (int i = 0; i < 50; i++) {
            new Thread(new TestMain()).start();
        }
    }

    @Override
    public void run() {
        lock.getLock();
        try {
            String number = dateFormat.format(new Date()) + "-" + ++count;
            System.out.println(Thread.currentThread().getName() + ",生成订单ID:" + number);
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            lock.unLock();
        }
    }
}

2.5 输出结果

获取锁
Thread-2,生成订单ID:2021-04-03-21-08-57-1
获取锁
Thread-5,生成订单ID:2021-04-03-21-08-57-2
获取锁
Thread-8,生成订单ID:2021-04-03-21-08-57-3
获取锁
Thread-11,生成订单ID:2021-04-03-21-08-57-4
获取锁
Thread-14,生成订单ID:2021-04-03-21-08-57-5
获取锁
Thread-17,生成订单ID:2021-04-03-21-08-57-6
获取锁
Thread-20,生成订单ID:2021-04-03-21-08-57-7
获取锁
Thread-23,生成订单ID:2021-04-03-21-08-57-8
获取锁
Thread-26,生成订单ID:2021-04-03-21-08-57-9
获取锁
Thread-29,生成订单ID:2021-04-03-21-08-57-10
获取锁
Thread-32,生成订单ID:2021-04-03-21-08-57-11
获取锁
Thread-35,生成订单ID:2021-04-03-21-08-57-12
获取锁
Thread-38,生成订单ID:2021-04-03-21-08-57-13
获取锁
Thread-41,生成订单ID:2021-04-03-21-08-57-14
获取锁
Thread-44,生成订单ID:2021-04-03-21-08-57-15
获取锁
Thread-47,生成订单ID:2021-04-03-21-08-57-16
获取锁
Thread-50,生成订单ID:2021-04-03-21-08-57-17
获取锁
Thread-53,生成订单ID:2021-04-03-21-08-57-18
获取锁
Thread-56,生成订单ID:2021-04-03-21-08-57-19
获取锁
Thread-59,生成订单ID:2021-04-03-21-08-57-20
获取锁
Thread-62,生成订单ID:2021-04-03-21-08-57-21
获取锁
Thread-65,生成订单ID:2021-04-03-21-08-57-22
获取锁
Thread-68,生成订单ID:2021-04-03-21-08-57-23
获取锁
Thread-71,生成订单ID:2021-04-03-21-08-57-24
获取锁
Thread-74,生成订单ID:2021-04-03-21-08-57-25
获取锁
Thread-77,生成订单ID:2021-04-03-21-08-57-26
获取锁
Thread-80,生成订单ID:2021-04-03-21-08-57-27
获取锁
Thread-83,生成订单ID:2021-04-03-21-08-57-28
获取锁
Thread-86,生成订单ID:2021-04-03-21-08-57-29
获取锁
Thread-89,生成订单ID:2021-04-03-21-08-57-30
获取锁
Thread-92,生成订单ID:2021-04-03-21-08-57-31
获取锁
Thread-95,生成订单ID:2021-04-03-21-08-57-32
获取锁
Thread-98,生成订单ID:2021-04-03-21-08-57-33
获取锁
Thread-101,生成订单ID:2021-04-03-21-08-57-34
获取锁
Thread-104,生成订单ID:2021-04-03-21-08-57-35
获取锁
Thread-107,生成订单ID:2021-04-03-21-08-57-36
获取锁
Thread-110,生成订单ID:2021-04-03-21-08-57-37
获取锁
Thread-113,生成订单ID:2021-04-03-21-08-57-38
获取锁
Thread-116,生成订单ID:2021-04-03-21-08-57-39
获取锁
Thread-119,生成订单ID:2021-04-03-21-08-57-40
获取锁
Thread-122,生成订单ID:2021-04-03-21-08-57-41
获取锁
Thread-125,生成订单ID:2021-04-03-21-08-57-42
获取锁
Thread-128,生成订单ID:2021-04-03-21-08-57-43
获取锁
Thread-131,生成订单ID:2021-04-03-21-08-57-44
获取锁
Thread-134,生成订单ID:2021-04-03-21-08-57-45
获取锁
Thread-137,生成订单ID:2021-04-03-21-08-57-46
获取锁
Thread-140,生成订单ID:2021-04-03-21-08-57-47
获取锁
Thread-143,生成订单ID:2021-04-03-21-08-57-48
获取锁
Thread-146,生成订单ID:2021-04-03-21-08-57-49
获取锁
Thread-149,生成订单ID:2021-04-03-21-08-57-50

 

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值