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

[유치수] [NOBTS] modified

Browse Source 
git-svn-id: http://svn.bds.nhncorp.com/pe/hippo-tomcat-profiler/trunk@564 84d0f5b1-2673-498c-a247-62c4ff18d310
This commit is contained in:
Chisu Yu
2012-08-30 08:47:51 +00:00
parent 0cf2c2c5e1
commit 01bf8eb108
8 changed files with 170 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/main/java/com/profiler/modifier/db/mysql/MySQLStatementModifier.java
+10 -47
View File
@@ -5,15 +5,10 @@ import java.util.logging.Level;
import java.util.logging.Logger;
import javassist.ByteArrayClassPath;
import javassist.CtClass;
import javassist.CtMethod;
import com.profiler.config.TomcatProfilerConstant;
import com.profiler.interceptor.Interceptor;
import com.profiler.interceptor.bci.ByteCodeInstrumentor;
import com.profiler.interceptor.bci.InstrumentClass;
import com.profiler.modifier.AbstractModifier;
import com.profiler.trace.DatabaseRequestTracer;
public class MySQLStatementModifier extends AbstractModifier {
@@ -31,54 +26,22 @@ public class MySQLStatementModifier extends AbstractModifier {
if (logger.isLoggable(Level.INFO)) {
logger.info("Modifing. " + javassistClassName);
}
// checkLibrary(classLoader, javassistClassName);
Interceptor interceptor = newInterceptor(classLoader, protectedDomain, "com.profiler.modifier.db.mysql.interceptors.ExecuteQueryMethodInterceptor");
if (interceptor == null) {
return null;
}
byteCodeInstrumentor.checkLibrary(classLoader, javassistClassName);
classPool.insertClassPath(new ByteArrayClassPath(javassistClassName, classFileBuffer));
InstrumentClass aClass = byteCodeInstrumentor.getClass(javassistClassName);
aClass.addInterceptor("executeQuery", new String[] { "java.lang.String" }, interceptor);
boolean instrumented = aClass.addInterceptor("executeQuery", new String[] { "java.lang.String" }, newInterceptor(classLoader, protectedDomain, "com.profiler.modifier.db.mysql.interceptors.ExecuteQueryMethodInterceptor"));
instrumented &= aClass.addInterceptor("executeUpdate", new String[] { "java.lang.String", "boolean", "boolean" }, newInterceptor(classLoader, protectedDomain, "com.profiler.modifier.db.mysql.interceptors.ExecuteUpdateMethodInterceptor"));
System.out.println("instrumented=" + instrumented);
if (!instrumented) {
return null;
}
return aClass.toBytecode();
// return changeMethod(javassistClassName, classFileBuffer);
}
@Deprecated
private byte[] changeMethod(String javassistClassName, byte[] classfileBuffer) {
try {
CtClass cc = classPool.get(javassistClassName);
updateExecuteQueryMethod(cc);
printClassConvertComplete(javassistClassName);
return cc.toBytecode();
} catch (Exception e) {
if (logger.isLoggable(Level.WARNING)) {
logger.log(Level.WARNING, e.getMessage(), e);
}
}
return null;
}
@Deprecated
private void updateExecuteQueryMethod(CtClass cc) throws Exception {
CtClass[] params = new CtClass[1];
params[0] = classPool.getCtClass("java.lang.String");
CtMethod method = cc.getDeclaredMethod("executeQuery", params);
StringBuilder sb = new StringBuilder();
sb.append("{");
sb.append(DatabaseRequestTracer.FQCN + ".putSqlQuery(" + TomcatProfilerConstant.REQ_DATA_TYPE_DB_QUERY + ",$1);");
sb.append(DatabaseRequestTracer.FQCN + ".put(" + TomcatProfilerConstant.REQ_DATA_TYPE_DB_EXECUTE_QUERY + ");");
sb.append("}");
method.insertAfter(sb.toString());
}
}