com.taobao.common.tedis.group.ZookeeperConfigManager Maven / Gradle / Ivy
/**
* (C) 2011-2012 Alibaba Group Holding Limited.
*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU General Public License
* version 2 as published by the Free Software Foundation.
*
*/
package com.taobao.common.tedis.group;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.Executor;
import java.util.concurrent.Executors;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.apache.zookeeper.WatchedEvent;
import org.apache.zookeeper.Watcher;
import com.taobao.common.tedis.config.ConfigManager;
import com.taobao.common.tedis.config.HAConfig;
import com.taobao.common.tedis.config.HAConfig.ServerInfo;
import com.taobao.common.tedis.config.HAConfig.ServerProperties;
import com.taobao.common.tedis.dislock.ZKClient;
import com.taobao.common.tedis.dislock.ZKException;
import com.taobao.common.tedis.util.ZKUtil;
/**
* ZKʵ??
*
* @author juxin.zj E-mail:[email protected]
* @since 2012-4-19 ????11:43:59
* @version 1.0
*/
public class ZookeeperConfigManager implements ConfigManager {
static Log logger = LogFactory.getLog(ZookeeperConfigManager.class);
volatile RandomRouter router;
private ZKClient zkClient;
private String zkAddress;
private int zkTimeout = 500000;
int timeout = 3000;
// ???????е?????
public volatile HAConfig haConfig;
Executor e = Executors.newSingleThreadExecutor();
String appName;
String version;
public ZookeeperConfigManager(String appName, String version) {
try {
init(appName, version);
} catch (Exception e) {
logger.error("Init zeekeeper error.", e);
}
}
private void init(String appName, String version) throws Exception {
this.appName = appName;
this.version = version;
zkClient = new ZKClient(zkAddress, zkTimeout);
zkClient.init();
String path = ZKUtil.contact("tedis-config", appName);
path = ZKUtil.contact(path, version);
// ip:port:r10,ip:port:r10
String configString = new String(zkClient.getData(ZKUtil.normalize(path), new ManagerWatcher()));
this.haConfig = parseConfig(configString);
if (this.haConfig.password != null) {
for (ServerProperties sp : haConfig.getServers()) {
sp.password = this.haConfig.password;
}
}
this.router = new RandomRouter(haConfig.getServers(), haConfig.failover);
}
public RandomRouter getRouter() {
return router;
}
public void destroy() {
}
public static HAConfig parseConfig(String configString) {
HAConfig config = new HAConfig();
// timeout
Pattern p_timeout = Pattern.compile("timeout=([\\s\\S]+?);");
Matcher m_timeout = p_timeout.matcher(configString);
if (m_timeout.find()) {
String s_timeout = m_timeout.group(1);
logger.info("timeout=" + s_timeout);
try {
config.timeout = Integer.parseInt(s_timeout.trim());
} catch (Exception ex) {
logger.error("timeout????????:", ex);
}
}
// pool_size
Pattern p_pool_size = Pattern.compile("pool_size=([\\s\\S]+?);");
Matcher m_pool_size = p_pool_size.matcher(configString);
if (m_pool_size.find()) {
String s_pool_size = m_pool_size.group(1);
logger.info("pool_size=" + s_pool_size);
try {
config.pool_size = Integer.parseInt(s_pool_size.trim());
} catch (Exception ex) {
logger.error("pool_size????????:", ex);
}
}
// password
Pattern p_password = Pattern.compile("password=([\\s\\S]+?);");
Matcher m_password = p_password.matcher(configString);
if (m_password.find()) {
String s_password = m_password.group(1);
logger.info("password=" + s_password);
try {
config.password = s_password.trim();
} catch (Exception ex) {
logger.error("password????????:", ex);
}
}
// servers
Pattern p = Pattern.compile("servers=([\\s\\S]+?);", Pattern.CASE_INSENSITIVE);
Matcher m = p.matcher(configString);
if (m.find()) {
String s_servers = m.group(1);
logger.info("servers=" + s_servers);
String[] array = s_servers.trim().split(",");
List servers = new ArrayList();
int groupSize = 0;
for (String s : array) {
String[] groups = s.split("\\|");
if (groupSize != 0 && groups.length != groupSize) {
logger.error("???ô????group size??һ??");
}
groupSize = groups.length;
ServerProperties sp = new ServerProperties();
sp.servers = new ServerInfo[groupSize];
for (int i = 0; i < groupSize; i++) {
String[] ss = groups[i].split(":");
ServerInfo server = new ServerInfo();
if (ss.length >= 2) {
server.addr = ss[0];
server.port = Integer.parseInt(ss[1]);
sp.pool_size = config.pool_size;
sp.timeout = config.timeout;
sp.password = config.password;
if (ss.length == 3) {
sp.readWeight = Integer.parseInt(ss[2].toLowerCase().replace("r", "").trim());
}
} else {
logger.error("???ô???:" + s);
}
sp.servers[i] = server;
}
servers.add(sp);
}
config.groups = servers;
} else {
logger.error("servers???ý???????:" + configString);
}
// fail over????
Pattern p_failover = Pattern.compile("failover=([\\s\\S]+?);", Pattern.CASE_INSENSITIVE);
Matcher m_failover = p_failover.matcher(configString);
if (m_failover.find()) {
try {
String s_failover = m.group(1);
config.failover = Boolean.parseBoolean(s_failover.trim());
} catch (Throwable t) {
logger.error("failover???ؽ???????", t);
}
}
return config;
}
public class ManagerWatcher implements Watcher {
@Override
public void process(WatchedEvent event) {
String string = null;
try {
string = new String(zkClient.getData(event.getPath()));
} catch (ZKException e) {
logger.error("Get config error.", e);
return;
}
logger.warn("???ñ????" + string);
if (string != null) {
haConfig = parseConfig(string);
RandomRouter old = router;
router = new RandomRouter(haConfig.getServers(), haConfig.failover);
try {
Thread.sleep(1000);
} catch (InterruptedException ex) {
logger.warn("InterruptedException:", ex);
}
old.destroy();
}
}
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy