Alawn/Druid
1
0
Fork 0
mirror of https://github.com/alibaba/druid.git synced 2023-03-04 19:17:16 +00:00
Code Issues Releases Wiki Activity

cache the reflect class and reflect method.

Browse Source
This commit is contained in:
septer
2012-08-10 18:35:36 +08:00
parent 9e2aa378a8
commit b272bfd4b7
2 changed files with 151 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/main/java/com/alibaba/druid/stat/DruidDataSourceStatJNDIStatStrategy.java
+115 -34
View File
@@ -1,5 +1,8 @@
package com.alibaba.druid.stat;
import java.lang.reflect.Method;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
@@ -10,66 +13,144 @@ import com.alibaba.druid.util.ReflectionUtils;
public class DruidDataSourceStatJNDIStatStrategy implements DruidDataSourceStatStrategy {
private final static Log LOG = LogFactory.getLog(DruidDataSourceStatJNDIStatStrategy.class);
private final static Log LOG = LogFactory.getLog(DruidDataSourceStatJNDIStatStrategy.class);
private Object getStatDefaultStrategy() {
private final static List<Object> strategyList = new ArrayList<Object>();
private final static HashMap<Object, HashMap<String, Method>> classMethodHM = new HashMap<Object, HashMap<String, Method>>();
private final static String[] methodNameArr = new String[] { "getSqlStatData",
"getSqlStatDataList", "getActiveConnectionStackTraceByDataSourceId", "returnJSONBasicStat",
"getDataSourceStatList", "getDataSourceStatData", "getPoolingConnectionInfoByDataSourceId", "resetAll",
"getDruidDataSourceById", "getSqlStatById" };
public DruidDataSourceStatJNDIStatStrategy(){
init();
}
private void init() {
try {
Class<?> clazz = ReflectionUtils.getClassFromWebContainerOrCurrentClassLoader("com.alibaba.druid.stat.DruidDataSourceStatDefaultStrategy");
return clazz.newInstance();
// cache the class
String className = "com.alibaba.druid.stat.DruidDataSourceStatDefaultStrategy";
Class<?> clazzFromWebContainer = ReflectionUtils.getClassFromWebContainer(className);
Class<?> clazzFromCurClassLoader = ReflectionUtils.getClassFromCurrentClassLoader(className);
if (clazzFromWebContainer != null) strategyList.add(clazzFromWebContainer.newInstance());
if (clazzFromCurClassLoader != null) strategyList.add(clazzFromCurClassLoader.newInstance());
// cache the method
Object obj = null;
for (int i = 0; i < strategyList.size(); i++) {
obj = strategyList.get(i);
HashMap<String, Method> methodHM = null;
String methodName = null;
for (int j = 0; j < methodNameArr.length; j++) {
methodHM = new HashMap<String, Method>();
methodName = methodNameArr[j];
methodHM.put(methodName, ReflectionUtils.getObjectMethod(obj, methodName));
}
classMethodHM.put(obj, methodHM);
}
} catch (Exception e) {
LOG.error(e.getMessage(), e);
return null;
}
}
@SuppressWarnings("unchecked")
private Object invoke(String methodName, Integer methodParam) {
Object result = null;
try {
HashMap<String, Method> hm = null;
Object invokeRes = null;
Object obj = null;
for (int i = 0; i < strategyList.size(); i++) {
obj = strategyList.get(i);
hm = classMethodHM.get(obj);
if (hm.containsKey(methodName)) {
if (methodParam == null) {
invokeRes = hm.get(methodName).invoke(obj);
} else {
invokeRes = hm.get(methodName).invoke(obj, methodParam);
}
if (invokeRes instanceof Map) {
result = invokeMapResHandler(null, (Map<String, Object>) invokeRes);
} else if (invokeRes instanceof List) {
result = invokeListResHandler(null, (List<Map<String, Object>>) invokeRes);
}
}
}
} catch (Exception e) {
LOG.error(e.getMessage(), e);
}
return result;
}
private Map<String, Object> invokeMapResHandler(Map<String, Object> finalRs, Map<String, Object> invokeRes) {
Map<String, Object> res = null;
if (finalRs == null) {
res = finalRs;
} else {
finalRs.putAll(invokeRes);
res = finalRs;
}
return res;
}
private List<?> invokeListResHandler(List<Map<String, Object>> finalRs, List<Map<String, Object>> invokeRes) {
List<Map<String, Object>> res = null;
if (finalRs == null) {
res = finalRs;
} else {
finalRs.addAll(invokeRes);
res = finalRs;
}
return res;
}
@SuppressWarnings("unchecked")
public Map<String, Object> getSqlStatData(Integer id) {
return (Map<String, Object>) ReflectionUtils.callObjectMethod(getStatDefaultStrategy(), "getSqlStatData", id);
}
@SuppressWarnings("unchecked")
public List<Map<String, Object>> getSqlStatDataList() {
return (List<Map<String, Object>>) ReflectionUtils.callObjectMethod(getStatDefaultStrategy(), "getSqlStatDataList");
}
@SuppressWarnings("unchecked")
public List<String> getActiveConnectionStackTraceByDataSourceId(Integer id) {
return (List<String>) ReflectionUtils.callObjectMethod(getStatDefaultStrategy(),
"getActiveConnectionStackTraceByDataSourceId", id);
}
@SuppressWarnings("unchecked")
public Map<String, Object> returnJSONBasicStat() {
return (Map<String, Object>) ReflectionUtils.callObjectMethod(getStatDefaultStrategy(), "returnJSONBasicStat");
}
@SuppressWarnings("unchecked")
public List<Object> getDataSourceStatList() {
return (List<Object>) ReflectionUtils.callObjectMethod(getStatDefaultStrategy(), "getDataSourceStatList");
return (Map<String, Object>) invoke("getSqlStatData", id);
}
@SuppressWarnings("unchecked")
public Map<String, Object> getDataSourceStatData(Integer id) {
return (Map<String, Object>) ReflectionUtils.callObjectMethod(getStatDefaultStrategy(), "getDataSourceStatData", id);
return (Map<String, Object>) invoke("getDataSourceStatData", id);
}
@SuppressWarnings("unchecked")
public Map<String, Object> returnJSONBasicStat() {
return (Map<String, Object>) invoke("returnJSONBasicStat", null);
}
@SuppressWarnings("unchecked")
public List<Map<String, Object>> getSqlStatDataList() {
return (List<Map<String, Object>>) invoke("getSqlStatDataList", null);
}
@SuppressWarnings("unchecked")
public List<String> getActiveConnectionStackTraceByDataSourceId(Integer id) {
return (List<String>) invoke("getActiveConnectionStackTraceByDataSourceId", id);
}
@SuppressWarnings("unchecked")
public List<Object> getDataSourceStatList() {
return (List<Object>) invoke("getDataSourceStatList", null);
}
@SuppressWarnings("unchecked")
public List<Map<String, Object>> getPoolingConnectionInfoByDataSourceId(Integer id) {
return (List<Map<String, Object>>) ReflectionUtils.callObjectMethod(getStatDefaultStrategy(),
"getPoolingConnectionInfoByDataSourceId",
id);
return (List<Map<String, Object>>) invoke("getPoolingConnectionInfoByDataSourceId", id);
}
public void resetAll() {
ReflectionUtils.callObjectMethod(getStatDefaultStrategy(), "resetAll");
invoke("resetAll", null);
}
public DruidDataSource getDruidDataSourceById(Integer id) {
return (DruidDataSource) ReflectionUtils.callObjectMethod(getStatDefaultStrategy(), "getDruidDataSourceById");
return (DruidDataSource) invoke("getDruidDataSourceById", id);
}
public JdbcSqlStat getSqlStatById(Integer id) {
return (JdbcSqlStat) ReflectionUtils.callObjectMethod(getStatDefaultStrategy(), "getSqlStatById", id);
return (JdbcSqlStat) invoke("getSqlStatById", id);
}
}
src/main/java/com/alibaba/druid/util/ReflectionUtils.java
+36 -7
View File
@@ -14,19 +14,30 @@ public class ReflectionUtils {
private final static Log LOG = LogFactory.getLog(ReflectionUtils.class);
public static Class<?> getClassFromWebContainerOrCurrentClassLoader(String className) {
public static Class<?> getClassFromWebContainer(String className) {
Class<?> result = null;
try {
result = HttpServletRequest.class.getClassLoader().loadClass(className);
} catch (ClassNotFoundException e) {
if (LOG.isDebugEnabled()) LOG.debug("can'r find class in web container classLoader ", e);
}
return result;
}
public static Class<?> getClassFromCurrentClassLoader(String className) {
Class<?> result = null;
try {
result = Thread.currentThread().getContextClassLoader().loadClass(className);
} catch (ClassNotFoundException e) {
LOG.error("can'r find class in current thread context classLoader ", e);
}
return result;
}
public static Class<?> getClassFromWebContainerOrCurrentClassLoader(String className) {
Class<?> result = getClassFromWebContainer(className);
if (result == null) {
try {
result = Thread.currentThread().getContextClassLoader().loadClass(className);
} catch (ClassNotFoundException e) {
LOG.error("can'r find class in current thread context classLoader ", e);
}
result = getClassFromCurrentClassLoader(className);
}
return result;
}
@@ -51,7 +62,25 @@ public class ReflectionUtils {
return null;
}
}
public static Method getObjectMethod(Object obj, String methodName) {
try {
return obj.getClass().getMethod(methodName);
} catch (Exception e) {
LOG.warn("getObjectMethod fail:class=" + obj.getClass().getName() + " method=" + methodName, e);
return null;
}
}
public static Method getObjectMethod(Object obj, String methodName, Integer id) {
try {
return obj.getClass().getMethod(methodName, Integer.class);
} catch (Exception e) {
LOG.warn("getObjectMethod fail:class=" + obj.getClass().getName() + " method=" + methodName, e);
return null;
}
}
public static Object callObjectMethod(Object obj, String methodName, Integer id) {
try {
Method m = obj.getClass().getMethod(methodName, Integer.class);