mirror of
https://github.com/wahyd4/pinpoint.git
synced 2026-08-17 08:46:22 +10:00
[유치수] [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:
@@ -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();
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user