com.taobao.common.tedis.group.TedisGroup 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.
*
*/
/*
* To change this template, choose Tools | Templates
* and open the template in the editor.
*/
package com.taobao.common.tedis.group;
import java.lang.reflect.InvocationHandler;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.lang.reflect.Proxy;
import java.lang.reflect.UndeclaredThrowableException;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.TimeoutException;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import com.taobao.common.tedis.Group;
import com.taobao.common.tedis.Single;
import com.taobao.common.tedis.TedisConnectionException;
import com.taobao.common.tedis.TedisException;
import com.taobao.common.tedis.binary.RedisCommands;
import com.taobao.common.tedis.config.ConfigManager;
import com.taobao.common.tedis.config.Process;
import com.taobao.common.tedis.config.Process.Policy;
import com.taobao.common.tedis.config.Router;
import com.taobao.common.tedis.monitor.BufferedStatLogWriter;
/**
* ???ζ??²?atom?ķ???ϸ?ڡ? ??д?ֿ????????·?ɣ?дȫ??
*
* @author jianxing
*/
public class TedisGroup implements Group {
static final Log logger = LogFactory.getLog(TedisGroup.class);
private ConfigManager cm;
private RedisCommands tedis;
private String appName;
private String version;
private volatile boolean inited = false;
private final static int retryTimes = 3;
public TedisGroup() {}
public TedisGroup(String appName, String version) {
this.appName = appName;
this.version = version;
}
@Override
public void init() {
if (!inited) {
if (this.cm == null) {
this.cm = new DiamondConfigManager(appName, version);
}
try {
tedis = (RedisCommands) Proxy.newProxyInstance(RedisCommands.class.getClassLoader(), new Class[] { RedisCommands.class }, new RedisGroupInvocationHandler());
} catch (Exception e) {
throw new TedisException("init failed", e);
}
inited = true;
}
}
@Override
public RedisCommands getTedis() {
if (tedis == null) {
throw new TedisException("please invoke the init method first.");
}
return tedis;
}
public class RedisGroupInvocationHandler implements InvocationHandler {
@Override
public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
long time = System.currentTimeMillis();
String name = method.getName();
Router rr = cm.getRouter();
Process annotation = method.getAnnotation(Process.class);
Throwable exception = null;
if (annotation.value() == Policy.READ) {
int errorCount = 0;
while (rr.getReadData().props.size() > 0) {
Single s = rr.route();
try {
Object result = method.invoke(s.getTedis(), args);
statLog(name, true, time);
return result;
} catch (Throwable t) {
statLog(name, false, time);
errorCount++;
if (errorCount < retryTimes) {
continue;
}
exception = t;
logger.warn("read exception:" + s.getProperties(), t);
boolean connectionError = false;
try {
if (t instanceof InvocationTargetException) {// ????쳣
InvocationTargetException ite = (InvocationTargetException) t;
UndeclaredThrowableException ute = (UndeclaredThrowableException) ite.getTargetException();
if (ute.getUndeclaredThrowable() instanceof TimeoutException) {
connectionError = true;
rr.onError(s);
} else {
ExecutionException ee = (ExecutionException) ute.getUndeclaredThrowable();
InvocationTargetException ite_1 = (InvocationTargetException) ee.getCause();
if (ite_1.getCause() instanceof ClassCastException) {
connectionError = true;
rr.onError(s);
} else {
TedisException te = (TedisException) ite_1.getTargetException();
if (te instanceof TedisConnectionException || te.getCause() instanceof TedisConnectionException) {
connectionError = true;
rr.onError(s);
}
}
}
}
} catch (Throwable tt) {
logger.warn("????쳣:", tt);
// ???ܻ??׳?ת???쳣,????Ԥ??,???????ת???쳣,ֱ????connection error?????д????׳?
}
if (!connectionError) {
throw t;
}
}
}
throw new Exception("Read RouteData is empty," + rr, exception);
} else if (annotation.value() == Policy.WRITE) {
Single[] ss = rr.getWriteData().group;
if (ss == null || ss.length == 0) {
throw new Exception("write RouteData is empty," + rr, exception);
}
Object result = null;
int e = 0;
for (Single s : ss) {
for (int i = 0; i < retryTimes; i++) {
try {
result = method.invoke(s.getTedis(), args);
} catch (Throwable t) {
if (i == retryTimes - 1) {
e++;
statLog(name, false, time);
logger.warn("write exception:" + s.getProperties(), t);
exception = t;
try {
// ????쳣
InvocationTargetException ite = (InvocationTargetException) t;
UndeclaredThrowableException ute = (UndeclaredThrowableException) ite.getTargetException();
if (ute.getUndeclaredThrowable() instanceof TimeoutException) {
rr.onError(s);
} else {
ExecutionException ee = (ExecutionException) ute.getUndeclaredThrowable();
InvocationTargetException ite_1 = (InvocationTargetException) ee.getCause();
if (ite_1.getCause() instanceof ClassCastException) {
rr.onError(s);
} else {
TedisException te = (TedisException) ite_1.getTargetException();
if (te instanceof TedisConnectionException || te.getCause() instanceof TedisConnectionException) {
rr.onError(s);
}
}
}
} catch (Throwable tt) {
logger.warn("????쳣:", tt);
}
} else {
continue;
}
}
break;
}
}
if (e >= 2) {// ȫ???????쳣??,??֪???ö?
throw exception;
}
statLog(name, true, time);
return result;
} else if ("toString".equals(name)) {
return "";
} else if ("hashCode".equals(name)) {
Single s = rr.route();
if (s != null) {
return s.hashCode();
} else {
return 0;
}
} else if ("equals".equals(name)) {
Single s = rr.route();
if (args.length == 1) {
return s.equals(args[0]);
}
}
statLog(name, false, time);
throw new Exception("method don't match:" + name);
}
}
private void statLog(String methodName, Boolean flag, long time) {
BufferedStatLogWriter.add(appName, methodName, flag, 1, System.currentTimeMillis() - time);
}
@Override
public void destroy() {
Router rr = cm.getRouter();
rr.destroy();
// ConfigMananger.destroy();
}
public String getAppName() {
return appName;
}
public void setAppName(String appName) {
this.appName = appName;
}
public String getVersion() {
return version;
}
public void setVersion(String version) {
this.version = version;
}
@Override
public void setConfigManager(ConfigManager cm) {
this.cm = cm;
}
public ConfigManager getConfigManager() {
return cm;
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy