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] add mysql interceptor

Browse Source 
git-svn-id: http://svn.bds.nhncorp.com/pe/hippo-tomcat-profiler/trunk@556 84d0f5b1-2673-498c-a247-62c4ff18d310
This commit is contained in:
Chisu Yu
2012-08-29 06:04:59 +00:00
parent c703ac84c9
commit 34dd7ef06d
8 changed files with 203 additions and 106 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/main/java/com/profiler/modifier/db/mysql/MySQLStatementModifier.java
+59 -43
View File
@@ -1,17 +1,16 @@
package com.profiler.modifier.db.mysql;
import com.profiler.interceptor.bci.ByteCodeInstrumentor;
import javassist.CtClass;
import javassist.CtMethod;
import com.profiler.config.TomcatProfilerConstant;
import com.profiler.modifier.AbstractModifier;
import com.profiler.trace.DatabaseRequestTracer;
import java.security.ProtectionDomain;
import java.util.logging.Level;
import java.util.logging.Logger;
import javassist.ByteArrayClassPath;
import com.profiler.interceptor.Interceptor;
import com.profiler.interceptor.bci.ByteCodeInstrumentor;
import com.profiler.interceptor.bci.InstrumentClass;
import com.profiler.modifier.AbstractModifier;
public class MySQLStatementModifier extends AbstractModifier {
private final Logger logger = Logger.getLogger(MySQLStatementModifier.class.getName());
@@ -23,43 +22,60 @@ public class MySQLStatementModifier extends AbstractModifier {
public String getTargetClass() {
return "com/mysql/jdbc/StatementImpl";
}
public byte[] modify(ClassLoader classLoader, String javassistClassName, ProtectionDomain protectedDomain, byte[] classFileBuffer) {
if (logger.isLoggable(Level.INFO)){
logger.info("Modifing. " + javassistClassName);
}
checkLibrary(classLoader, javassistClassName);
return changeMethod(javassistClassName, classFileBuffer);
}
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);
}
if (logger.isLoggable(Level.INFO)) {
logger.info("Modifing. " + javassistClassName);
}
return null;
// 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);
return aClass.toBytecode();
// return changeMethod(javassistClassName, classFileBuffer);
}
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());
}
// 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;
// }
//
// 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());
// }
}