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没抢到锁

我们可以看到每次只有一个线程获取到了锁，并处理业务逻辑