com.taobao.common.tedis.group.DiamondConfigManager 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 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.diamond.manager.DiamondManager;
import com.taobao.diamond.manager.ManagerListener;
import com.taobao.diamond.manager.impl.DefaultDiamondManager;
/**
* HA????ʵ?֡?
*
* @author jianxing
*/
public class DiamondConfigManager implements ConfigManager {
static Log logger = LogFactory.getLog(DiamondConfigManager.class);
public static String defaultConfKey = "com.taobao.common.tedis.config";
volatile RandomRouter router;
DiamondManager diamondManager;
String configKey;
int timeout = 3000;
// ???????е?????
public volatile HAConfig haConfig;
Executor e = Executors.newSingleThreadExecutor();
public ManagerListenerImpl managerListener = new ManagerListenerImpl();
String appName;
String version;
public DiamondConfigManager(String appName, String version) {
init(appName, version);
}
private void init(String appName, String version) {
this.appName = appName;
this.version = version;
this.configKey = defaultConfKey + "." + appName + "." + version;
diamondManager = new DefaultDiamondManager(configKey, (ManagerListener) null);
// ip:port:r10,ip:port:r10
String configString = diamondManager.getConfigureInfomation(timeout);
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);
diamondManager.setManagerListener(managerListener);
}
@Override
public RandomRouter getRouter() {
return router;
}
public void setDefaultConfigKey(String key) {
this.configKey = key;
}
@Override
public void destroy() {
diamondManager.close();
}
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 ManagerListenerImpl implements ManagerListener {
@Override
public Executor getExecutor() {
return e;
}
@Override
public void receiveConfigInfo(String string) {
logger.warn("???ñ????" + string);
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