用zookeeper实现简单的检测离线或者上线的功能

心跳检测

机器间的心跳检测机制是指在分布式环境中，不同机器（或进程）之间需要检测到彼此是否在正常运行，例如A机器需要知道B机器是否正常运行。在传统的开发中，我们通常是通过主机直接是否可以相互PING通来判断，更复杂一点的话，则会通过在机器之间建立长连接，通过TCP连接固有的心跳检测机制来实现上层机器的心跳检测，这些都是非常常见的心跳检测方法。

下面来看看如何使用ZK来实现分布式机器（进程）间的心跳检测。

基于ZK的临时节点的特性，可以让不同的进程都在ZK的一个指定节点下创建临时子节点，不同的进程直接可以根据这个临时子节点来判断对应的进程是否存活。通过这种方式，检测和被检测系统直接并不需要直接相关联，而是通过ZK上的某个节点进行关联，大大减少了系统耦合。

以下代码主要是利用zookeeper的心跳机制和临时节点，来实现两个客户端client1 、client2之间上线下线通知的功能：

主要实现类

package com.zhuyun.keepalive;

import org.apache.zookeeper.CreateMode;
import org.apache.zookeeper.KeeperException;
import org.apache.zookeeper.WatchedEvent;
import org.apache.zookeeper.Watcher;
import org.apache.zookeeper.Watcher.Event.EventType;
import org.apache.zookeeper.ZooDefs.Ids;
import org.apache.zookeeper.ZooKeeper;
import org.apache.zookeeper.data.Stat;

public class Executor implements Watcher{
	ZooKeeper zk;
	String hostPort;
	String znode1;
	String znode2;
	
	public Executor(String hostPort,String znode1, String znode2) throws Exception{
		this.hostPort = hostPort;
		this.znode1 = znode1;
		this.znode2 = znode2;
		
		zk = new ZooKeeper(hostPort, 3000, this);
		Stat exists = zk.exists(znode1, false);
		if (exists == null) {			//假如节点不存在，则创建临时节点
			zk.create(znode1, null, Ids.OPEN_ACL_UNSAFE, CreateMode.EPHEMERAL);
		}
		//监听第二个节点
		zk.exists(znode2, true);
	}
	
	@Override
	public void process(WatchedEvent event) {
		try {
			
		if (event.getType() == EventType.NodeDeleted) {	
			//临时节点被删除了，则节点不在线；触发后，watcher被删除，继续添加watcher
				System.out.println("节点" + znode2 + "当前不在线");
				zk.exists(znode2, true);
		}else if (event.getType() == EventType.NodeCreated) {
			//临时节点被创建了，则节点上线了；触发后，watcher被删除，继续添加watcher
			System.out.println("节点" + znode2 + "上线了");
			zk.exists(znode2, true);
		}
		
		} catch (KeeperException | InterruptedException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
	}
	
	
}

以下是client1的代码：

package com.zhuyun.keepalive;

public class Client1 {
	public static void main(String[] args) throws Exception {
		new Executor("192.168.10.203:2181", "/com/zhuyun/keepalive/client1","/com/zhuyun/keepalive/client2");
		
		System.in.read();
	}
}

以下是client1打印结果：

节点/com/zhuyun/keepalive/client2当前不在线
节点/com/zhuyun/keepalive/client2上线了

以下是client2的代码

package com.zhuyun.keepalive;

public class Client2 {
	public static void main(String[] args) throws Exception {
		new Executor("192.168.10.203:2181", "/com/zhuyun/keepalive/client2","/com/zhuyun/keepalive/client1");
		
		System.in.read();
	}
}

以下是client2的打印结果：

节点/com/zhuyun/keepalive/client1当前不在线