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