zookeeper实现互斥锁

简单的说，zookeeper就是为了解决集群环境中数据一致性的问题。

举个很简单栗子：

有一个变量A，分别存在于两台服务器中，某个程序须要用到变量A，就随机地访问其中一台服务器并取得变量A的值，对吧？如今有两个客户端分别修改两台服务器上的变量A，其中一个客户端把其中一台服务器的变量A的值修改成1，另外一个客户端把另外一台服务器的变量A修改成2。那么以后，程序访问取得这个变量的值，究竟是1仍是2呢？zookeeper就是为了解决这种问题的。

本文主要是讲一个demo，利用zookeeper的特性来实现一个互斥锁，原理性的东西本文不赘述。

咱们用多线程的方式来模拟多个客户端争抢锁，线程类代码以下：

public class Thread1 implements Runnable {
        ZooKeeper zk = null;

        @Override
        public void run() {
            try {
                zk = new ZooKeeper("127.0.0.1:2181", 2000, new Watcher() {
                    @Override
                    public void process(WatchedEvent event) {
                        // 获取事件类型
                        String type = event.getType().toString();
                        /**
                         * 咱们使用节点/lock/mylock为锁的标志
                         * 事件类型为“None”表示线程刚启动，线程启动后就当即去抢锁
                         * 事件类型为“NodeDeleted”表示有节点被删除了
                         * ，也就表示有线程释放锁了，当前线程获得事件通知后就当即抢锁
                         * 。if块里面的代码都是抢锁、处理业务逻辑、释放锁这个流程
                         */
                        if ("None".equals(type) || ("NodeDeleted".equals(type) && event.getPath().equals("/lock/mylock"))) {
                            try {
                                // 建立一个/lock/mylock节点，即执行抢锁。若是节点建立失败，则会抛出一个异常，进入下面的catch块中
                                zk.create("/lock/mylock", new byte[0], Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT);

                                // 继续监控/lock/mylock节点的变化状况，由于zookeeper的watcher是一次性的，因此用完后要从新watch
                                zk.exists("/lock/mylock", true);

                                // /lock/mylock节点建立成功，表示抢到了锁
                                System.out.println("----------------------------------");
                                System.out.println(Thread.currentThread().getName() + "抢到锁了");
                                System.out.println(Thread.currentThread().getName() + "处理业务逻辑……");
                                Thread.sleep(2000);

                                // 删除/lock/mylock节点，释放锁，以后全部的watch这个节点的线程都会收到事件通知，从而当即抢锁
                                zk.delete("/lock/mylock", -1);
                                System.out.println(Thread.currentThread().getName() + "已释放锁");
                            } catch (Exception e) {
                                System.out.println(Thread.currentThread().getName() + "没抢到锁");
                                try {
                                    // 抢锁失败，继续监控锁标志节点的变化状况
                                    zk.exists("/lock/mylock", true);
                                } catch (Exception e1) {
                                    e1.printStackTrace();
                                }
                            }
                        }
                    }
                });
            } catch (Exception e) {
                e.printStackTrace();
            }
        }

 

main函数代码以下：

for (int i = 0; i < 5; i++) {
            new Thread(new Thread1()).start();
        }
        System.in.read();

 

运行结果：

Thread-3-EventThread抢到锁了
Thread-3-EventThread处理业务逻辑……
Thread-2-EventThread没抢到锁
Thread-4-EventThread没抢到锁
Thread-0-EventThread没抢到锁
Thread-1-EventThread没抢到锁
Thread-3-EventThread已释放锁
----------------------------------
Thread-2-EventThread没抢到锁
Thread-0-EventThread没抢到锁
Thread-1-EventThread没抢到锁
Thread-3-EventThread没抢到锁
Thread-4-EventThread抢到锁了
Thread-4-EventThread处理业务逻辑……
Thread-4-EventThread已释放锁
----------------------------------
Thread-3-EventThread没抢到锁
Thread-1-EventThread没抢到锁
Thread-2-EventThread没抢到锁
Thread-4-EventThread没抢到锁
Thread-0-EventThread抢到锁了
Thread-0-EventThread处理业务逻辑……
Thread-0-EventThread已释放锁
----------------------------------
Thread-4-EventThread没抢到锁
Thread-3-EventThread没抢到锁
Thread-2-EventThread没抢到锁
Thread-0-EventThread没抢到锁
Thread-1-EventThread抢到锁了
Thread-1-EventThread处理业务逻辑……
Thread-1-EventThread已释放锁
----------------------------------
Thread-0-EventThread没抢到锁
Thread-2-EventThread没抢到锁
Thread-4-EventThread没抢到锁
Thread-1-EventThread没抢到锁
Thread-3-EventThread抢到锁了
Thread-3-EventThread处理业务逻辑……
Thread-3-EventThread已释放锁
----------------------------------
Thread-0-EventThread没抢到锁
Thread-4-EventThread没抢到锁
Thread-2-EventThread没抢到锁
Thread-3-EventThread没抢到锁
Thread-1-EventThread抢到锁了
Thread-1-EventThread处理业务逻辑……
Thread-1-EventThread已释放锁
----------------------------------
Thread-3-EventThread没抢到锁
Thread-2-EventThread没抢到锁
Thread-4-EventThread没抢到锁
Thread-1-EventThread没抢到锁
Thread-0-EventThread抢到锁了
Thread-0-EventThread处理业务逻辑……
Thread-0-EventThread已释放锁
----------------------------------
Thread-2-EventThread没抢到锁
Thread-0-EventThread没抢到锁
Thread-4-EventThread没抢到锁
Thread-1-EventThread没抢到锁
Thread-3-EventThread抢到锁了
Thread-3-EventThread处理业务逻辑……
Thread-3-EventThread已释放锁
----------------------------------
Thread-0-EventThread没抢到锁
Thread-2-EventThread没抢到锁
Thread-3-EventThread没抢到锁
Thread-1-EventThread抢到锁了
Thread-1-EventThread处理业务逻辑……
Thread-4-EventThread没抢到锁

咱们能够看到每次只有一个线程获取到了锁，并处理业务逻辑