mirror of
https://github.com/wahyd4/pinpoint.git
synced 2026-08-26 13:17:15 +10:00
[유치수] [NOBTS] refactoring
git-svn-id: http://svn.bds.nhncorp.com/pe/hippo-tomcat-profiler/trunk@471 84d0f5b1-2673-498c-a247-62c4ff18d310
This commit is contained in:
@@ -13,20 +13,21 @@ public class MySQLResultSetModifier extends AbstractModifier {
|
||||
private static final Logger logger = Logger.getLogger(MySQLResultSetModifier.class);
|
||||
|
||||
public byte[] modify(ClassPool classPool, ClassLoader classLoader, String javassistClassName, byte[] classFileBuffer) {
|
||||
logger.info("ResultSetImpl modifing. %s", javassistClassName);
|
||||
checkLibrary(classPool, javassistClassName, classLoader);
|
||||
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);
|
||||
|
||||
updateNextMethod(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();
|
||||
@@ -35,35 +36,12 @@ public class MySQLResultSetModifier extends AbstractModifier {
|
||||
}
|
||||
|
||||
private static void updateNextMethod(ClassPool classPool, CtClass cc) throws Exception {
|
||||
CtMethod serviceMethod1 = cc.getDeclaredMethod("next", null);
|
||||
|
||||
logger.info("*** Changing next() method");
|
||||
|
||||
serviceMethod1.insertBefore(getNextMethodBeforeInsertCode());
|
||||
}
|
||||
|
||||
private static String getNextMethodBeforeInsertCode() {
|
||||
StringBuilder sb = new StringBuilder();
|
||||
sb.append("{");
|
||||
sb.append(TomcatProfilerConstant.CLASS_NAME_REQUEST_DATA_TRACER + ".updateFetchCount();");
|
||||
sb.append("}");
|
||||
return sb.toString();
|
||||
CtMethod method = cc.getDeclaredMethod("next", null);
|
||||
method.insertBefore("{" + TomcatProfilerConstant.CLASS_NAME_REQUEST_DATA_TRACER + ".updateFetchCount(); }");
|
||||
}
|
||||
|
||||
private static void updateCloseMethod(ClassPool classPool, CtClass cc) throws Exception {
|
||||
CtMethod serviceMethod1 = cc.getDeclaredMethod("close", null);
|
||||
|
||||
logger.info("Changing close() method");
|
||||
|
||||
serviceMethod1.insertBefore(getCloseMethodBeforeInsertCode());
|
||||
}
|
||||
|
||||
private static String getCloseMethodBeforeInsertCode() {
|
||||
StringBuilder sb = new StringBuilder();
|
||||
sb.append("{");
|
||||
sb.append(TomcatProfilerConstant.CLASS_NAME_REQUEST_DATA_TRACER + ".addResultSetData();");
|
||||
sb.append("}");
|
||||
|
||||
return sb.toString();
|
||||
CtMethod method = cc.getDeclaredMethod("close", null);
|
||||
method.insertBefore("{" + TomcatProfilerConstant.CLASS_NAME_REQUEST_DATA_TRACER + ".addResultSetData(); }");
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user