mirror of
https://github.com/wahyd4/pinpoint.git
synced 2026-08-14 15:26:31 +10:00
[유치수] [NOBTS] change project type.
git-svn-id: http://svn.bds.nhncorp.com/pe/hippo-tomcat-profiler/trunk@431 84d0f5b1-2673-498c-a247-62c4ff18d310
This commit is contained in:
@@ -0,0 +1,55 @@
|
||||
package com.profiler.modifier.db.mysql;
|
||||
|
||||
import javassist.ClassPool;
|
||||
import javassist.CtClass;
|
||||
import javassist.CtMethod;
|
||||
|
||||
import com.profiler.config.TomcatProfilerConstant;
|
||||
import com.profiler.modifier.AbstractModifier;
|
||||
|
||||
public class MySQLStatementModifier extends AbstractModifier{
|
||||
public static byte[] modify(ClassPool classPool,ClassLoader classLoader,String javassistClassName,byte[] classFileBuffer) {
|
||||
log("MySQLStatementModifier modifing");
|
||||
// printClassInfo(javassistClassName);
|
||||
return changeMethod(classPool,classLoader,javassistClassName,classFileBuffer);
|
||||
}
|
||||
private static 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;
|
||||
} catch (Exception e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
return null;
|
||||
}
|
||||
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);
|
||||
log("*** Changing executeQuery method ");
|
||||
// serviceMethod.insertBefore(getExecuteQueryMethodBeforeInsertCode());
|
||||
serviceMethod.insertAfter(getExecuteQueryMethodAfterInsertCode());
|
||||
}
|
||||
@SuppressWarnings("unused")
|
||||
private static String getExecuteQueryMethodBeforeInsertCode() {
|
||||
StringBuilder sb=new StringBuilder();
|
||||
// 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("{");
|
||||
// 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();
|
||||
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user