home/pinpoint
2
0
Fork 0
mirror of https://github.com/wahyd4/pinpoint.git synced 2026-08-17 08:46:22 +10:00
Code Issues Packages Projects Releases Wiki Activity

[유치수] [NOBTS] refactoring

Browse Source 
git-svn-id: http://svn.bds.nhncorp.com/pe/hippo-tomcat-profiler/trunk@471 84d0f5b1-2673-498c-a247-62c4ff18d310
This commit is contained in:
Chisu Yu
2012-07-25 06:39:24 +00:00
parent a6cbef8def
commit 8caba0a28a
24 changed files with 366 additions and 1071 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/main/java/com/profiler/modifier/db/mysql/MySQLConnectionImplModifier.java
+13 -100
View File
@@ -13,21 +13,22 @@ public class MySQLConnectionImplModifier extends AbstractModifier {
private static final Logger logger = Logger.getLogger(MySQLConnectionImplModifier.class);
public byte[] modify(ClassPool classPool, ClassLoader classLoader, String javassistClassName, byte[] classFileBuffer) {
logger.info("MySQLConnectionImplModifier modifing. %s", javassistClassName);
checkLibrary(classPool, javassistClassName, classLoader);
return changeMethods(classPool, classLoader, javassistClassName, classFileBuffer);
logger.info("Modifing. %s", javassistClassName);
checkLibrary(classPool, javassistClassName, classLoader);
return changeMethods(classPool, classLoader, javassistClassName, classFileBuffer);
}
private 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);
byte[] newClassfileBuffer = cc.toBytecode();
// cc.writeFile();
printClassConvertComplete(javassistClassName);
return newClassfileBuffer;
return cc.toBytecode();
} catch (Exception e) {
logger.error(e.getMessage());
e.printStackTrace();
@@ -36,37 +37,8 @@ public class MySQLConnectionImplModifier extends AbstractModifier {
}
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());
}
private static String getCreateStatementMethodBeforeInsertCode() {
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();
sb.append("{");
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();
CtMethod method = cc.getDeclaredMethod("createStatement", null);
method.insertAfter("{" + TomcatProfilerConstant.CLASS_NAME_REQUEST_DATA_TRACER + ".put(" + TomcatProfilerConstant.REQ_DATA_TYPE_DB_CREATE_STATEMENT + "); }");
}
private static void updateGetInstanceMethod(ClassPool classPool, CtClass cc) throws Exception {
@@ -76,72 +48,13 @@ public class MySQLConnectionImplModifier extends AbstractModifier {
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);
CtMethod method = cc.getDeclaredMethod("getInstance", params);
logger.info("Changing getInstance method ");
serviceMethod.insertBefore(getGetInstanceMethodBeforeInsertCode());
serviceMethod.insertAfter(getGetInstanceMethodAfterInsertCode());
}
private static String getGetInstanceMethodBeforeInsertCode() {
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();
sb.append("{");
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();
method.insertAfter("{" + TomcatProfilerConstant.CLASS_NAME_REQUEST_DATA_TRACER + ".putConnection(" + TomcatProfilerConstant.REQ_DATA_TYPE_DB_GET_CONNECTION + ",$5); }");
}
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());
}
private static String getCloseMethodBeforeInsertCode() {
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();
sb.append("{");
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();
CtMethod method = cc.getDeclaredMethod("close", null);
method.insertAfter("{" + TomcatProfilerConstant.CLASS_NAME_REQUEST_DATA_TRACER + ".put(" + TomcatProfilerConstant.REQ_DATA_TYPE_DB_CLOSE_CONNECTION + "); }");
}
}