home/pinpoint
2
0
Fork 0
mirror of https://github.com/wahyd4/pinpoint.git synced 2026-08-15 07:46:19 +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/MySQLStatementModifier.java
+9 -32
View File
@@ -13,18 +13,20 @@ public class MySQLStatementModifier extends AbstractModifier {
private static final Logger logger = Logger.getLogger(MySQLStatementModifier.class);
public byte[] modify(ClassPool classPool, ClassLoader classLoader, String javassistClassName, byte[] classFileBuffer) {
logger.info("MySQLStatementModifier modifing. %s", javassistClassName);
checkLibrary(classPool, javassistClassName, classLoader);
return changeMethod(classPool, classLoader, javassistClassName, classFileBuffer);
logger.info("Modifing. %s", javassistClassName);
checkLibrary(classPool, javassistClassName, classLoader);
return changeMethod(classPool, classLoader, javassistClassName, classFileBuffer);
}
private byte[] changeMethod(ClassPool classPool, ClassLoader classLoader, String javassistClassName, byte[] classfileBuffer) {
try {
CtClass cc = classPool.get(javassistClassName);
updateExecuteQueryMethod(classPool, cc);
byte[] newClassfileBuffer = cc.toBytecode();
printClassConvertComplete(javassistClassName);
return newClassfileBuffer;
return cc.toBytecode();
} catch (Exception e) {
logger.error(e.getMessage());
e.printStackTrace();
@@ -35,39 +37,14 @@ public class MySQLStatementModifier extends AbstractModifier {
private static void updateExecuteQueryMethod(ClassPool classPool, CtClass cc) throws Exception {
CtClass[] params = new CtClass[1];
params[0] = classPool.getCtClass("java.lang.String");
CtMethod serviceMethod = cc.getDeclaredMethod("executeQuery", params);
CtMethod method = cc.getDeclaredMethod("executeQuery", params);
logger.info("Changing executeQuery method");
serviceMethod.insertBefore(getExecuteQueryMethodBeforeInsertCode());
serviceMethod.insertAfter(getExecuteQueryMethodAfterInsertCode());
}
private static String getExecuteQueryMethodBeforeInsertCode() {
StringBuilder sb = new StringBuilder();
if (logger.isDebugEnabled()) {
sb.append("{");
sb.append("System.out.println(\"-----StatementImpl.executeQuery(String) method is called\");");
sb.append("System.out.println(\"-----Query=[\"+$1+\"]\");");
sb.append("}");
}
return sb.toString();
}
private static String getExecuteQueryMethodAfterInsertCode() {
StringBuilder sb = new StringBuilder();
sb.append("{");
if (logger.isDebugEnabled()) {
sb.append("System.out.println(\"-----StatementImpl.executeQuery(String) method is ended\");");
}
sb.append(TomcatProfilerConstant.CLASS_NAME_REQUEST_DATA_TRACER + ".putSqlQuery(" + TomcatProfilerConstant.REQ_DATA_TYPE_DB_QUERY + ",$1);");
sb.append(TomcatProfilerConstant.CLASS_NAME_REQUEST_DATA_TRACER + ".put(" + TomcatProfilerConstant.REQ_DATA_TYPE_DB_EXECUTE_QUERY + ");");
sb.append("}");
return sb.toString();
method.insertAfter(sb.toString());
}
}