org.tinygroup.remoteconfig.zk.client.ZKManager Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of org.tinygroup.remoteconfig.zk Show documentation
There is a newer version: 3.4.9
Show newest version
package org.tinygroup.remoteconfig.zk.client;

import java.io.UnsupportedEncodingException;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.apache.commons.lang.StringUtils;
import org.apache.zookeeper.CreateMode;
import org.apache.zookeeper.KeeperException;
import org.apache.zookeeper.ZooDefs;
import org.apache.zookeeper.data.Stat;
import org.tinygroup.exception.BaseRuntimeException;
import org.tinygroup.remoteconfig.config.ConfigPath;
import org.tinygroup.remoteconfig.config.ConfigValue;
import org.tinygroup.remoteconfig.config.Module;
import org.tinygroup.remoteconfig.zk.utils.PathHelper;
import org.tinygroup.remoteconfig.zk.utils.SerializeUtil;

public class ZKManager extends BaseManager{

	public static ConfigValue get(String key ,ConfigPath configPath){
		try {
			key = PathHelper.createPath(key ,configPath);
			return getSimple(key);
		} catch (KeeperException e) {
			throw new BaseRuntimeException("0TE120119003", e ,key);
		} catch (InterruptedException e) {
			throw new BaseRuntimeException("0TE120119004", e ,key);
		} catch (UnsupportedEncodingException e) {
			throw new BaseRuntimeException("0TE120119003", e ,key);
		}
	}

	public static Map getAll(ConfigPath configPath) {
		Map dataMap = new HashMap();
		String node = PathHelper.createPath("" ,configPath);
		try {
			List subNodes = zooKeeper.getChildren(node, false);
			if (subNodes != null && !subNodes.isEmpty()) {
				for (String subNode : subNodes) {
					ConfigValue znodeValue = getSimple(node.concat("/").concat(subNode));
					if (znodeValue != null) {
						dataMap.put(subNode, znodeValue);
					}
				}
			}
		} catch (KeeperException e) {
			//当没有找到子项时，会抛出此异常，我们不做任何处理
		} catch (InterruptedException e) {
			throw new BaseRuntimeException("0TE120119006", e ,node);
		} catch (UnsupportedEncodingException e) {
			throw new BaseRuntimeException("0TE120119006", e ,node);
		}
		return dataMap;
	}

	public static void set(String key, ConfigValue value, ConfigPath configPath) {
		key = PathHelper.createPath(key ,configPath);
		try {
			Stat stat = zooKeeper.exists(key, false);
			if (stat == null) {
				addSimple(key ,value);
				return;
			}
			zooKeeper.setData(key, SerializeUtil.serialize(value),stat.getVersion());
		} catch (Exception e) {
			throw new BaseRuntimeException("0TE120119002", e ,key ,value);
		}
	}

	private static Stat addSimple(String node, ConfigValue value) throws BaseRuntimeException{
		try {
			Stat stat = zooKeeper.exists(node, false);
			if (stat == null) {
				//  create parent znodePath
				String parentPath = PathHelper.generateParentPath(node);
				if (StringUtils.isNotBlank(parentPath)) {
					Stat parentStat = zooKeeper.exists(parentPath, false);
					if (parentStat == null) {
						addModule(parentPath, new Module());
					}
				}
				zooKeeper.create(node, SerializeUtil.serialize(value), ZooDefs.Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT);
			}
			return zooKeeper.exists(node, true);
		} catch (KeeperException e) {
			throw new BaseRuntimeException("0TE120119011", e ,node ,value);
		} catch (InterruptedException e) {
			throw new BaseRuntimeException("0TE120119012", e ,node ,value);
		}
	}
	
	private static Stat addModule(String node, Module module) throws BaseRuntimeException{
		try {
			Stat stat = zooKeeper.exists(node, false);
			if (stat == null) {
				//  create parent znodePath
				String parentPath = PathHelper.generateParentPath(node);
				if (StringUtils.isNotBlank(parentPath)) {
					Stat parentStat = zooKeeper.exists(parentPath, false);
					if (parentStat == null) {
						addModule(parentPath ,new Module());
					}
				}
				String name = StringUtils.substringAfterLast(node, "/");
				if (StringUtils.isBlank(module.getName())) {
					module.setName(name);
				}
				if (StringUtils.isBlank(module.getModuleName())) {
					module.setModuleName(name);
				}
				zooKeeper.create(node, SerializeUtil.serialize(module), ZooDefs.Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT);
			}
			return zooKeeper.exists(node, true);
		} catch (KeeperException e) {
			throw new BaseRuntimeException("0TE120119011", e ,node ,module);
		} catch (InterruptedException e) {
			throw new BaseRuntimeException("0TE120119012", e ,node ,module);
		}
	}

	private static ConfigValue getSimple(String key) throws KeeperException, InterruptedException, UnsupportedEncodingException{
		Stat stat = zooKeeper.exists(key, false);
		if (stat != null) {
			byte[] resultData = zooKeeper.getData(key, false, stat);
			if (resultData != null) {
				Object obj =  SerializeUtil.unserialize(resultData);
				if (obj == null) {
					return new ConfigValue(new String(resultData ,"UTF-8"));
				}else if (obj instanceof String) {
					return new ConfigValue(obj.toString());
				}else if (obj instanceof ConfigValue) {
					return (ConfigValue) obj;
				}
			}
		}
		return null;
	}
	
}