com.alogic.zk.ZooKeeperCluster Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of alogic-zk Show documentation
Zookeeper相关的API
The newest version!
package com.alogic.zk;

import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

import com.alogic.json.JsonFactory;
import org.apache.zookeeper.WatchedEvent;
import org.apache.zookeeper.Watcher;
import org.apache.zookeeper.Watcher.Event.EventType;
import org.apache.zookeeper.Watcher.Event.KeeperState;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.alogic.remote.backend.AppBackends;
import com.alogic.remote.backend.Backend;
import com.alogic.remote.cluster.Cluster;
import com.anysoft.util.BaseException;
import com.anysoft.util.JsonTools;
import com.anysoft.util.MapProperties;
import com.anysoft.util.Properties;
import com.anysoft.util.PropertiesConstants;
import com.anysoft.util.Settings;
import com.anysoft.util.UPath;

/**
 * 基于zk的Cluster
 * @author yyduan
 *
 */
public class ZooKeeperCluster extends Cluster.Abstract implements Watcher{
	/**
	 * a logger of log4j
	 */
	protected static final Logger LOG = LoggerFactory.getLogger(ZooKeeperCluster.class);
	
	/**
	 * 在Zookeeper中的根节点位置
	 */	
	protected String rootPath = "${zookeeper.arm.root}";
	
	/**
	 * zk连接
	 */
	protected ZooKeeperConnector conn = null;
	
	/**
	 * 监听器
	 */
	protected com.anysoft.util.Watcher listener = null;
	
	/**
	 * 后端节点缓存数据
	 */
	protected Map backends = new ConcurrentHashMap();
	
	@Override
	public void configure(Properties p) {
		super.configure(p);
		
		rootPath = PropertiesConstants.getString(p, "arm.root", rootPath);
		if (rootPath.length() <= 0){
			rootPath = "/alogic/global/arm";
		}	
		
		conn = new ZooKeeperConnector(p);
	}
	
	@Override
	public AppBackends load(String appId) {
		AppBackends found = backends.get(appId);
		if (found == null){
			synchronized(this){
				found = backends.get(appId);
				if (found == null){
					found = loadFromZK(appId);
					if (found != null){
						backends.put(appId, found);
					}
				}
			}
		}
		return found;
	}
	
	protected AppBackends loadFromZK(String appId){
		AppBackends found = null;
		
		try {
			UPath path = new UPath(rootPath + "/" + appId);
			boolean exist = existPath(path, null ,true);
			if (exist){
				found = new AppBackends(appId);
				String [] children = loadChildren(path, this,true);
				if (children != null){
					for (String child:children){
						UPath childPath = path.append(child);
						String data = loadData(childPath, null,true);
						if (data != null){
							Backend backend = createBackend(data);
							if (backend != null){
								found.addBackend(backend);
							}
						}
					}
				}
			}
		}catch (BaseException ex){
			LOG.error("Can not load app backends:" + appId,ex);
		}
		return found;		
	}

	@SuppressWarnings("unchecked")
	protected Backend createBackend(String data) {
		JsonFactory jsonFactory = Settings.getToolkit(JsonFactory.class);
		Object jsonObj = jsonFactory.fromJsonString(data);
		if (jsonObj != null && jsonObj instanceof Map){
			Map mapObj = (Map) jsonObj;
			String ip = JsonTools.getString(mapObj, "ip", "");
			String port = JsonTools.getString(mapObj, "port", "");
			if (ip.length() > 0 && port.length() > 0){
				Backend.Default backend = new Backend.Default();
				backend.configure(new MapProperties(mapObj,Settings.get()));
				return backend;
			}
		}
		return null;
	}

	
	/**
	 * 装入ZooKeeper节点数据
	 * @param path 节点路径
	 * @param watcher 监听器
	 * @param ignoreException 是否忽略异常
	 * @return 节点数据
	 */
	protected String loadData(UPath path,Watcher watcher,boolean ignoreException){
		if (!conn.isConnected()){
			conn.connect();
		}
		return conn.loadData(path, watcher,ignoreException);		
	}
	
	/**
	 * 装入指定节点的所有子节点
	 * @param path
	 * @param watcher
	 * @param ignoreException 是否忽略异常
	 * @return 子节点列表
	 */
	protected String [] loadChildren(UPath path,Watcher watcher,boolean ignoreException){
		if (!conn.isConnected()){
			conn.connect();
		}
		return conn.loadChildren(path, watcher,ignoreException);
	}
	
	/**
	 * 是否存在指定的路径
	 * @param path 路径
	 * @param watcher 监听器
	 * @param ignoreException 是否忽略异常
	 * @return 是否存在指定的路径
	 */
	protected boolean existPath(UPath path,Watcher watcher,boolean ignoreException){
		if (!conn.isConnected()){
			conn.connect();
		}
		return conn.existPath(path, watcher,ignoreException);
	}
	
	@Override
	public void process(WatchedEvent event) {
		EventType type = event.getType();
		switch (type){
			case NodeChildrenChanged:
			{
				UPath path = new UPath(event.getPath());
				AppBackends appBackends = load(path.getId());
				if (appBackends != null){
					fireChangeEvent(path.getId(), appBackends);
				}
				break;
			}
			case NodeCreated:
			case NodeDataChanged:				
			case NodeDeleted:
			{
				//重新订阅
				if (!conn.isConnected()){
					conn.connect();
				}
				conn.existPath(new UPath(event.getPath()),this,true);
				break;
			}
			case None:
			{
				KeeperState state = event.getState();
				if (state == KeeperState.Expired){
					//过期了
					fireAllChangedEvent();
				}
			}
		}
	}
	
	@Override
	public void addWatcher(com.anysoft.util.Watcher watcher) {	
		listener = watcher;
	}

	@Override
	public void removeWatcher(com.anysoft.util.Watcher watcher) {
		listener = null;
	}

	/**
	 * 触发变更事件
	 * @param id
	 * @param data
	 */
	protected void fireChangeEvent(String id,AppBackends data){
		backends.remove(id);
		if (listener != null){
			listener.changed(id, data);
		}
	}
	
	protected void fireAllChangedEvent(){
		backends.clear();
		if (listener != null){
			listener.allChanged();
		}
	}
}