[유치수] [NOBTS] refactoring

git-svn-id: http://svn.bds.nhncorp.com/pe/hippo-tomcat-profiler/trunk@461 84d0f5b1-2673-498c-a247-62c4ff18d310
This commit is contained in:
Chisu Yu
2012-07-06 09:58:04 +00:00
parent 74126a7f7c
commit 727b2f91a3
34 changed files with 5519 additions and 5184 deletions
@@ -5,100 +5,142 @@ import javassist.CtClass;
import javassist.CtMethod;
import com.profiler.config.TomcatProfilerConstant;
import com.profiler.logging.Logger;
import com.profiler.modifier.AbstractModifier;
public class MySQLConnectionImplModifier extends AbstractModifier{
public static byte[] modify(ClassPool classPool,ClassLoader classLoader,String javassistClassName,byte[] classFileBuffer) {
log("MySQLConnectionImplModifier modifing");
// printClassInfo(javassistClassName);
return changeMethods(classPool,classLoader,javassistClassName,classFileBuffer);
public class MySQLConnectionImplModifier extends AbstractModifier {
private static final Logger logger = Logger.getLogger(MySQLConnectionImplModifier.class);
public static byte[] modify(ClassPool classPool, ClassLoader classLoader, String javassistClassName, byte[] classFileBuffer) {
logger.info("MySQLConnectionImplModifier modifing. %s", javassistClassName);
return changeMethods(classPool, classLoader, javassistClassName, classFileBuffer);
}
private static byte[] changeMethods(ClassPool classPool,ClassLoader classLoader,String javassistClassName,byte[] classfileBuffer) {
private static byte[] changeMethods(ClassPool classPool, ClassLoader classLoader, String javassistClassName, byte[] classfileBuffer) {
try {
CtClass cc = classPool.get(javassistClassName);
updateGetInstanceMethod(classPool,cc);
updateCreateStatementMethod(classPool,cc);
updateCloseMethod(classPool,cc);
updateGetInstanceMethod(classPool, cc);
updateCreateStatementMethod(classPool, cc);
updateCloseMethod(classPool, cc);
byte[] newClassfileBuffer = cc.toBytecode();
// cc.writeFile();
// cc.writeFile();
printClassConvertComplete(javassistClassName);
return newClassfileBuffer;
} catch (Exception e) {
logger.error(e.getMessage());
e.printStackTrace();
}
return null;
}
private static void updateCreateStatementMethod(ClassPool classPool,CtClass cc) throws Exception {
CtMethod serviceMethod=cc.getDeclaredMethod("createStatement", null);
log("*** Changing createStatement method ");
// serviceMethod.insertBefore(getCreateStatementMethodBeforeInsertCode());
private static void updateCreateStatementMethod(ClassPool classPool, CtClass cc) throws Exception {
CtMethod serviceMethod = cc.getDeclaredMethod("createStatement", null);
logger.info("Changing createStatement method");
serviceMethod.insertBefore(getCreateStatementMethodBeforeInsertCode());
serviceMethod.insertAfter(getCreateStatementMethodAfterInsertCode());
}
@SuppressWarnings("unused")
private static String getCreateStatementMethodBeforeInsertCode() {
StringBuilder sb=new StringBuilder();
// sb.append("{");
// sb.append("System.out.println(\"-----ConnectionImpl.createStatement() method is called\");");
// sb.append("}");
StringBuilder sb = new StringBuilder();
if (logger.isDebugEnabled()) {
sb.append("{");
sb.append("System.out.println(\"-----ConnectionImpl.createStatement() method is called\");");
sb.append("}");
}
return sb.toString();
}
private static String getCreateStatementMethodAfterInsertCode() {
StringBuilder sb=new StringBuilder();
StringBuilder sb = new StringBuilder();
sb.append("{");
// sb.append("System.out.println(\"-----ConnectionImpl.createStatement() method is ended\");");
sb.append(TomcatProfilerConstant.CLASS_NAME_REQUEST_DATA_TRACER+".put("+TomcatProfilerConstant.REQ_DATA_TYPE_DB_CREATE_STATEMENT+");");
if (logger.isDebugEnabled()) {
sb.append("System.out.println(\"-----ConnectionImpl.createStatement() method is ended\");");
}
sb.append(TomcatProfilerConstant.CLASS_NAME_REQUEST_DATA_TRACER + ".put(" + TomcatProfilerConstant.REQ_DATA_TYPE_DB_CREATE_STATEMENT + ");");
sb.append("}");
return sb.toString();
}
private static void updateGetInstanceMethod(ClassPool classPool,CtClass cc) throws Exception {
CtClass[] params=new CtClass[5];
params[0]=classPool.getCtClass("java.lang.String");
params[1]=classPool.getCtClass("int");
params[2]=classPool.getCtClass("java.util.Properties");
params[3]=classPool.getCtClass("java.lang.String");
params[4]=classPool.getCtClass("java.lang.String");
CtMethod serviceMethod=cc.getDeclaredMethod("getInstance", params);
log("*** Changing getInstance method ");
// serviceMethod.insertBefore(getGetInstanceMethodBeforeInsertCode());
private static void updateGetInstanceMethod(ClassPool classPool, CtClass cc) throws Exception {
CtClass[] params = new CtClass[5];
params[0] = classPool.getCtClass("java.lang.String");
params[1] = classPool.getCtClass("int");
params[2] = classPool.getCtClass("java.util.Properties");
params[3] = classPool.getCtClass("java.lang.String");
params[4] = classPool.getCtClass("java.lang.String");
CtMethod serviceMethod = cc.getDeclaredMethod("getInstance", params);
logger.info("Changing getInstance method ");
serviceMethod.insertBefore(getGetInstanceMethodBeforeInsertCode());
serviceMethod.insertAfter(getGetInstanceMethodAfterInsertCode());
}
@SuppressWarnings("unused")
private static String getGetInstanceMethodBeforeInsertCode() {
StringBuilder sb=new StringBuilder();
// sb.append("{");
// sb.append("System.out.println(\"-----ConnectionImpl.getInstance() method is called\");");
// sb.append("}");
StringBuilder sb = new StringBuilder();
if (logger.isDebugEnabled()) {
sb.append("{");
sb.append("System.out.println(\"-----ConnectionImpl.getInstance() method is called\");");
sb.append("}");
}
return sb.toString();
}
private static String getGetInstanceMethodAfterInsertCode() {
StringBuilder sb=new StringBuilder();
StringBuilder sb = new StringBuilder();
sb.append("{");
// sb.append("System.out.println(\"-----ConnectionImpl.getInstance() method is ended\");");
sb.append(TomcatProfilerConstant.CLASS_NAME_REQUEST_DATA_TRACER+".putConnection("+TomcatProfilerConstant.REQ_DATA_TYPE_DB_GET_CONNECTION+",$5);");
if (logger.isDebugEnabled()) {
sb.append("System.out.println(\"-----ConnectionImpl.getInstance() method is ended\");");
}
sb.append(TomcatProfilerConstant.CLASS_NAME_REQUEST_DATA_TRACER + ".putConnection(" + TomcatProfilerConstant.REQ_DATA_TYPE_DB_GET_CONNECTION + ",$5);");
sb.append("}");
return sb.toString();
}
private static void updateCloseMethod(ClassPool classPool,CtClass cc) throws Exception {
CtMethod serviceMethod=cc.getDeclaredMethod("close", null);
log("*** Changing close method ");
// serviceMethod.insertBefore(getCloseMethodBeforeInsertCode());
private static void updateCloseMethod(ClassPool classPool, CtClass cc) throws Exception {
CtMethod serviceMethod = cc.getDeclaredMethod("close", null);
logger.info("Changing close method ");
serviceMethod.insertBefore(getCloseMethodBeforeInsertCode());
serviceMethod.insertAfter(getCloseMethodAfterInsertCode());
}
@SuppressWarnings("unused")
private static String getCloseMethodBeforeInsertCode() {
StringBuilder sb=new StringBuilder();
// sb.append("{");
// sb.append("System.out.println(\"-----ConnectionImpl.close() method is called\");");
// sb.append("}");
StringBuilder sb = new StringBuilder();
if (logger.isDebugEnabled()) {
sb.append("{");
sb.append("System.out.println(\"-----ConnectionImpl.close() method is called\");");
sb.append("}");
}
return sb.toString();
}
private static String getCloseMethodAfterInsertCode() {
StringBuilder sb=new StringBuilder();
StringBuilder sb = new StringBuilder();
sb.append("{");
// sb.append("System.out.println(\"-----ConnectionImpl.close() method is ended\");");
sb.append(TomcatProfilerConstant.CLASS_NAME_REQUEST_DATA_TRACER+".put("+TomcatProfilerConstant.REQ_DATA_TYPE_DB_CLOSE_CONNECTION+");");
if (logger.isDebugEnabled()) {
sb.append("System.out.println(\"-----ConnectionImpl.close() method is ended\");");
}
sb.append(TomcatProfilerConstant.CLASS_NAME_REQUEST_DATA_TRACER + ".put(" + TomcatProfilerConstant.REQ_DATA_TYPE_DB_CLOSE_CONNECTION + ");");
sb.append("}");
return sb.toString();
}
}