mirror of
https://github.com/wahyd4/pinpoint.git
synced 2026-08-27 05:36:39 +10:00
[유치수] [NOBTS] refactoring
git-svn-id: http://svn.bds.nhncorp.com/pe/hippo-tomcat-profiler/trunk@473 84d0f5b1-2673-498c-a247-62c4ff18d310
This commit is contained in:
+15
-11
@@ -13,18 +13,22 @@ public class CubridPreparedStatementModifier extends AbstractModifier {
|
||||
|
||||
private static final Logger logger = Logger.getLogger(CubridPreparedStatementModifier.class);
|
||||
|
||||
public byte[] modify(ClassPool classPool, ClassLoader classLoader, String javassistClassName, byte[] classFileBuffer) {
|
||||
logger.info("Modifing. %s", javassistClassName);
|
||||
checkLibrary(classPool, javassistClassName, classLoader);
|
||||
return changeMethod(classPool, classLoader, javassistClassName, classFileBuffer);
|
||||
public CubridPreparedStatementModifier(ClassPool classPool) {
|
||||
super(classPool);
|
||||
}
|
||||
|
||||
private byte[] changeMethod(ClassPool classPool, ClassLoader classLoader, String javassistClassName, byte[] classfileBuffer) {
|
||||
public byte[] modify(ClassLoader classLoader, String javassistClassName, byte[] classFileBuffer) {
|
||||
logger.info("Modifing. %s", javassistClassName);
|
||||
checkLibrary(classLoader, javassistClassName);
|
||||
return changeMethod(javassistClassName, classFileBuffer);
|
||||
}
|
||||
|
||||
private byte[] changeMethod(String javassistClassName, byte[] classfileBuffer) {
|
||||
try {
|
||||
CtClass cc = classPool.get(javassistClassName);
|
||||
|
||||
updateExecuteQueryMethod(classPool, cc);
|
||||
updateConstructor(classPool, cc);
|
||||
updateExecuteQueryMethod(cc);
|
||||
updateConstructor(cc);
|
||||
|
||||
printClassConvertComplete(javassistClassName);
|
||||
|
||||
@@ -36,12 +40,12 @@ public class CubridPreparedStatementModifier extends AbstractModifier {
|
||||
return null;
|
||||
}
|
||||
|
||||
private static void updateConstructor(ClassPool classPool, CtClass cc) throws Exception {
|
||||
private static void updateConstructor(CtClass cc) throws Exception {
|
||||
CtConstructor[] constructorList = cc.getConstructors();
|
||||
|
||||
|
||||
for (CtConstructor constructor : constructorList) {
|
||||
CtClass params[] = constructor.getParameterTypes();
|
||||
|
||||
|
||||
if (params.length > 2) {
|
||||
StringBuilder sb = new StringBuilder();
|
||||
sb.append("{");
|
||||
@@ -54,7 +58,7 @@ public class CubridPreparedStatementModifier extends AbstractModifier {
|
||||
}
|
||||
}
|
||||
|
||||
private static void updateExecuteQueryMethod(ClassPool classPool, CtClass cc) throws Exception {
|
||||
private static void updateExecuteQueryMethod(CtClass cc) throws Exception {
|
||||
CtMethod serviceMethod = cc.getDeclaredMethod("execute", null);
|
||||
serviceMethod.insertAfter("{" + TomcatProfilerConstant.CLASS_NAME_REQUEST_DATA_TRACER + ".put(" + TomcatProfilerConstant.REQ_DATA_TYPE_DB_EXECUTE_QUERY + "); }");
|
||||
}
|
||||
|
||||
@@ -12,18 +12,22 @@ public class CubridResultSetModifier extends AbstractModifier {
|
||||
|
||||
private static final Logger logger = Logger.getLogger(CubridResultSetModifier.class);
|
||||
|
||||
public byte[] modify(ClassPool classPool, ClassLoader classLoader, String javassistClassName, byte[] classFileBuffer) {
|
||||
logger.info("Modifing. %s", javassistClassName);
|
||||
checkLibrary(classPool, javassistClassName, classLoader);
|
||||
return changeMethod(classPool, classLoader, javassistClassName, classFileBuffer);
|
||||
public CubridResultSetModifier(ClassPool classPool) {
|
||||
super(classPool);
|
||||
}
|
||||
|
||||
private byte[] changeMethod(ClassPool classPool, ClassLoader classLoader, String javassistClassName, byte[] classfileBuffer) {
|
||||
public byte[] modify(ClassLoader classLoader, String javassistClassName, byte[] classFileBuffer) {
|
||||
logger.info("Modifing. %s", javassistClassName);
|
||||
checkLibrary(classLoader, javassistClassName);
|
||||
return changeMethod(javassistClassName, classFileBuffer);
|
||||
}
|
||||
|
||||
private byte[] changeMethod(String javassistClassName, byte[] classfileBuffer) {
|
||||
try {
|
||||
CtClass cc = classPool.get(javassistClassName);
|
||||
|
||||
updateNextMethod(classPool, cc);
|
||||
updateCloseMethod(classPool, cc);
|
||||
updateNextMethod(cc);
|
||||
updateCloseMethod(cc);
|
||||
|
||||
printClassConvertComplete(javassistClassName);
|
||||
|
||||
@@ -35,12 +39,12 @@ public class CubridResultSetModifier extends AbstractModifier {
|
||||
return null;
|
||||
}
|
||||
|
||||
private static void updateNextMethod(ClassPool classPool, CtClass cc) throws Exception {
|
||||
private static void updateNextMethod(CtClass cc) throws Exception {
|
||||
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 {
|
||||
private static void updateCloseMethod(CtClass cc) throws Exception {
|
||||
CtMethod method = cc.getDeclaredMethod("close", null);
|
||||
method.insertBefore("{" + TomcatProfilerConstant.CLASS_NAME_REQUEST_DATA_TRACER + ".addResultSetData(); } ");
|
||||
}
|
||||
|
||||
@@ -12,17 +12,21 @@ public class CubridStatementModifier extends AbstractModifier {
|
||||
|
||||
private static final Logger logger = Logger.getLogger(CubridStatementModifier.class);
|
||||
|
||||
public byte[] modify(ClassPool classPool, ClassLoader classLoader, String javassistClassName, byte[] classFileBuffer) {
|
||||
logger.info("Modifing. %s", javassistClassName);
|
||||
checkLibrary(classPool, javassistClassName, classLoader);
|
||||
return changeMethod(classPool, classLoader, javassistClassName, classFileBuffer);
|
||||
public CubridStatementModifier(ClassPool classPool) {
|
||||
super(classPool);
|
||||
}
|
||||
|
||||
private byte[] changeMethod(ClassPool classPool, ClassLoader classLoader, String javassistClassName, byte[] classfileBuffer) {
|
||||
public byte[] modify(ClassLoader classLoader, String javassistClassName, byte[] classFileBuffer) {
|
||||
logger.info("Modifing. %s", javassistClassName);
|
||||
checkLibrary(classLoader, javassistClassName);
|
||||
return changeMethod(javassistClassName, classFileBuffer);
|
||||
}
|
||||
|
||||
private byte[] changeMethod(String javassistClassName, byte[] classfileBuffer) {
|
||||
try {
|
||||
CtClass cc = classPool.get(javassistClassName);
|
||||
|
||||
updateExecuteQueryMethod(classPool, cc);
|
||||
updateExecuteQueryMethod(cc);
|
||||
|
||||
printClassConvertComplete(javassistClassName);
|
||||
|
||||
@@ -34,7 +38,7 @@ public class CubridStatementModifier extends AbstractModifier {
|
||||
return null;
|
||||
}
|
||||
|
||||
private static void updateExecuteQueryMethod(ClassPool classPool, CtClass cc) throws Exception {
|
||||
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);
|
||||
|
||||
@@ -12,17 +12,21 @@ public class CubridUStatementModifier extends AbstractModifier {
|
||||
|
||||
private static final Logger logger = Logger.getLogger(CubridUStatementModifier.class);
|
||||
|
||||
public byte[] modify(ClassPool classPool, ClassLoader classLoader, String javassistClassName, byte[] classFileBuffer) {
|
||||
logger.info("Modifing. %s", javassistClassName);
|
||||
checkLibrary(classPool, javassistClassName, classLoader);
|
||||
return changeMethod(classPool, classLoader, javassistClassName, classFileBuffer);
|
||||
public CubridUStatementModifier(ClassPool classPool) {
|
||||
super(classPool);
|
||||
}
|
||||
|
||||
private byte[] changeMethod(ClassPool classPool, ClassLoader classLoader, String javassistClassName, byte[] classfileBuffer) {
|
||||
public byte[] modify(ClassLoader classLoader, String javassistClassName, byte[] classFileBuffer) {
|
||||
logger.info("Modifing. %s", javassistClassName);
|
||||
checkLibrary(classLoader, javassistClassName);
|
||||
return changeMethod(javassistClassName, classFileBuffer);
|
||||
}
|
||||
|
||||
private byte[] changeMethod(String javassistClassName, byte[] classfileBuffer) {
|
||||
try {
|
||||
CtClass cc = classPool.get(javassistClassName);
|
||||
|
||||
updateBindValueMethod(classPool, cc);
|
||||
updateBindValueMethod(cc);
|
||||
|
||||
printClassConvertComplete(javassistClassName);
|
||||
|
||||
@@ -34,7 +38,7 @@ public class CubridUStatementModifier extends AbstractModifier {
|
||||
return null;
|
||||
}
|
||||
|
||||
private static void updateBindValueMethod(ClassPool classPool, CtClass cc) throws Exception {
|
||||
private void updateBindValueMethod(CtClass cc) throws Exception {
|
||||
CtClass[] params1 = new CtClass[3];
|
||||
params1[0] = classPool.getCtClass("int");
|
||||
params1[1] = classPool.getCtClass("byte");
|
||||
|
||||
@@ -12,17 +12,21 @@ public class DBCPBasicDataSourceModifier extends AbstractModifier {
|
||||
|
||||
private static final Logger logger = Logger.getLogger(DBCPBasicDataSourceModifier.class);
|
||||
|
||||
public byte[] modify(ClassPool classPool, ClassLoader classLoader, String javassistClassName, byte[] classFileBuffer) {
|
||||
logger.info("Modifing. %s", javassistClassName);
|
||||
checkLibrary(classPool, javassistClassName, classLoader);
|
||||
return changeMethod(classPool, classLoader, javassistClassName, classFileBuffer);
|
||||
public DBCPBasicDataSourceModifier(ClassPool classPool) {
|
||||
super(classPool);
|
||||
}
|
||||
|
||||
private byte[] changeMethod(ClassPool classPool, ClassLoader classLoader, String javassistClassName, byte[] classfileBuffer) {
|
||||
public byte[] modify(ClassLoader classLoader, String javassistClassName, byte[] classFileBuffer) {
|
||||
logger.info("Modifing. %s", javassistClassName);
|
||||
checkLibrary(classLoader, javassistClassName);
|
||||
return changeMethod(javassistClassName, classFileBuffer);
|
||||
}
|
||||
|
||||
private byte[] changeMethod(String javassistClassName, byte[] classfileBuffer) {
|
||||
try {
|
||||
CtClass cc = classPool.get(javassistClassName);
|
||||
|
||||
updateGetConnectionMethod(classPool, cc);
|
||||
updateGetConnectionMethod(cc);
|
||||
|
||||
printClassConvertComplete(javassistClassName);
|
||||
|
||||
@@ -34,7 +38,7 @@ public class DBCPBasicDataSourceModifier extends AbstractModifier {
|
||||
return null;
|
||||
}
|
||||
|
||||
private static void updateGetConnectionMethod(ClassPool classPool, CtClass cc) throws Exception {
|
||||
private void updateGetConnectionMethod(CtClass cc) throws Exception {
|
||||
CtMethod method = cc.getDeclaredMethod("getConnection", null);
|
||||
method.insertAfter("{" + TomcatProfilerConstant.CLASS_NAME_REQUEST_DATA_TRACER + ".putConnection(" + TomcatProfilerConstant.REQ_DATA_TYPE_DB_GET_CONNECTION + ",$0.getUrl()); }");
|
||||
}
|
||||
|
||||
@@ -12,17 +12,21 @@ public class DBCPPoolModifier extends AbstractModifier {
|
||||
|
||||
private static final Logger logger = Logger.getLogger(DBCPPoolModifier.class);
|
||||
|
||||
public byte[] modify(ClassPool classPool, ClassLoader classLoader, String javassistClassName, byte[] classFileBuffer) {
|
||||
logger.info("Modifing. %s", javassistClassName);
|
||||
checkLibrary(classPool, javassistClassName, classLoader);
|
||||
return changeMethod(classPool, classLoader, javassistClassName, classFileBuffer);
|
||||
public DBCPPoolModifier(ClassPool classPool) {
|
||||
super(classPool);
|
||||
}
|
||||
|
||||
private byte[] changeMethod(ClassPool classPool, ClassLoader classLoader, String javassistClassName, byte[] classfileBuffer) {
|
||||
public byte[] modify(ClassLoader classLoader, String javassistClassName, byte[] classFileBuffer) {
|
||||
logger.info("Modifing. %s", javassistClassName);
|
||||
checkLibrary(classLoader, javassistClassName);
|
||||
return changeMethod(javassistClassName, classFileBuffer);
|
||||
}
|
||||
|
||||
private byte[] changeMethod(String javassistClassName, byte[] classfileBuffer) {
|
||||
try {
|
||||
CtClass cc = classPool.get(javassistClassName);
|
||||
|
||||
updateCloseMethod(classPool, cc);
|
||||
updateCloseMethod(cc);
|
||||
|
||||
printClassConvertComplete(javassistClassName);
|
||||
|
||||
@@ -34,7 +38,7 @@ public class DBCPPoolModifier extends AbstractModifier {
|
||||
return null;
|
||||
}
|
||||
|
||||
private static void updateCloseMethod(ClassPool classPool, CtClass cc) throws Exception {
|
||||
private void updateCloseMethod(CtClass cc) throws Exception {
|
||||
CtMethod method = cc.getDeclaredMethod("close", null);
|
||||
method.insertAfter("{" + TomcatProfilerConstant.CLASS_NAME_REQUEST_DATA_TRACER + ".put(" + TomcatProfilerConstant.REQ_DATA_TYPE_DB_CLOSE_CONNECTION + "); }");
|
||||
}
|
||||
|
||||
@@ -12,18 +12,22 @@ public class MSSQLConnectionModifier extends AbstractModifier {
|
||||
|
||||
private static final Logger logger = Logger.getLogger(MSSQLConnectionModifier.class);
|
||||
|
||||
public byte[] modify(ClassPool classPool, ClassLoader classLoader, String javassistClassName, byte[] classFileBuffer) {
|
||||
logger.info("Modifing. %s", javassistClassName);
|
||||
checkLibrary(classPool, javassistClassName, classLoader);
|
||||
return changeMethods(classPool, classLoader, javassistClassName, classFileBuffer);
|
||||
public MSSQLConnectionModifier(ClassPool classPool) {
|
||||
super(classPool);
|
||||
}
|
||||
|
||||
private byte[] changeMethods(ClassPool classPool, ClassLoader classLoader, String javassistClassName, byte[] classfileBuffer) {
|
||||
public byte[] modify(ClassLoader classLoader, String javassistClassName, byte[] classFileBuffer) {
|
||||
logger.info("Modifing. %s", javassistClassName);
|
||||
checkLibrary(classLoader, javassistClassName);
|
||||
return changeMethods(javassistClassName, classFileBuffer);
|
||||
}
|
||||
|
||||
private byte[] changeMethods(String javassistClassName, byte[] classfileBuffer) {
|
||||
try {
|
||||
CtClass cc = classPool.get(javassistClassName);
|
||||
|
||||
updateCreateStatementMethod(classPool, cc);
|
||||
updateCloseMethod(classPool, cc);
|
||||
updateCreateStatementMethod(cc);
|
||||
updateCloseMethod(cc);
|
||||
|
||||
printClassConvertComplete(javassistClassName);
|
||||
|
||||
@@ -35,7 +39,7 @@ public class MSSQLConnectionModifier extends AbstractModifier {
|
||||
return null;
|
||||
}
|
||||
|
||||
private static void updateCreateStatementMethod(ClassPool classPool, CtClass cc) throws Exception {
|
||||
private void updateCreateStatementMethod(CtClass cc) throws Exception {
|
||||
CtClass[] params = new CtClass[2];
|
||||
params[0] = classPool.getCtClass("int");
|
||||
params[1] = classPool.getCtClass("int");
|
||||
@@ -44,7 +48,7 @@ public class MSSQLConnectionModifier extends AbstractModifier {
|
||||
method.insertAfter("{" + TomcatProfilerConstant.CLASS_NAME_REQUEST_DATA_TRACER + ".put(" + TomcatProfilerConstant.REQ_DATA_TYPE_DB_CREATE_STATEMENT + "); }");
|
||||
}
|
||||
|
||||
private static void updateCloseMethod(ClassPool classPool, CtClass cc) throws Exception {
|
||||
private void updateCloseMethod(CtClass cc) throws Exception {
|
||||
CtMethod method = cc.getDeclaredMethod("close", null);
|
||||
method.insertAfter("{" + TomcatProfilerConstant.CLASS_NAME_REQUEST_DATA_TRACER + ".put(" + TomcatProfilerConstant.REQ_DATA_TYPE_DB_CLOSE_CONNECTION + "); }");
|
||||
}
|
||||
|
||||
@@ -13,19 +13,23 @@ public class MSSQLPreparedStatementModifier extends AbstractModifier {
|
||||
|
||||
private static final Logger logger = Logger.getLogger(MSSQLPreparedStatementModifier.class);
|
||||
|
||||
public byte[] modify(ClassPool classPool, ClassLoader classLoader, String javassistClassName, byte[] classFileBuffer) {
|
||||
logger.info("Modifing. %s", javassistClassName);
|
||||
checkLibrary(classPool, javassistClassName, classLoader);
|
||||
return changeMethod(classPool, classLoader, javassistClassName, classFileBuffer);
|
||||
public MSSQLPreparedStatementModifier(ClassPool classPool) {
|
||||
super(classPool);
|
||||
}
|
||||
|
||||
private byte[] changeMethod(ClassPool classPool, ClassLoader classLoader, String javassistClassName, byte[] classfileBuffer) {
|
||||
public byte[] modify(ClassLoader classLoader, String javassistClassName, byte[] classFileBuffer) {
|
||||
logger.info("Modifing. %s", javassistClassName);
|
||||
checkLibrary(classLoader, javassistClassName);
|
||||
return changeMethod(javassistClassName, classFileBuffer);
|
||||
}
|
||||
|
||||
private byte[] changeMethod(String javassistClassName, byte[] classfileBuffer) {
|
||||
try {
|
||||
CtClass cc = classPool.get(javassistClassName);
|
||||
|
||||
updateSetParameterMethod(classPool, cc);
|
||||
updateExecuteQueryMethod(classPool, cc);
|
||||
updateConstructor(classPool, cc);
|
||||
updateSetParameterMethod(cc);
|
||||
updateExecuteQueryMethod(cc);
|
||||
updateConstructor(cc);
|
||||
|
||||
printClassConvertComplete(javassistClassName);
|
||||
|
||||
@@ -37,7 +41,7 @@ public class MSSQLPreparedStatementModifier extends AbstractModifier {
|
||||
return null;
|
||||
}
|
||||
|
||||
private static void updateSetParameterMethod(ClassPool classPool, CtClass cc) throws Exception {
|
||||
private void updateSetParameterMethod(CtClass cc) throws Exception {
|
||||
CtClass[] params1 = new CtClass[5];
|
||||
params1[0] = classPool.getCtClass("int");
|
||||
params1[1] = classPool.getCtClass("java.lang.Object");
|
||||
@@ -49,16 +53,16 @@ public class MSSQLPreparedStatementModifier extends AbstractModifier {
|
||||
method.insertBefore("{" + TomcatProfilerConstant.CLASS_NAME_REQUEST_DATA_TRACER + ".putSqlParam($1,$2);} ");
|
||||
}
|
||||
|
||||
private static void updateConstructor(ClassPool classPool, CtClass cc) throws Exception {
|
||||
private void updateConstructor(CtClass cc) throws Exception {
|
||||
CtConstructor[] constructorList = cc.getConstructors();
|
||||
|
||||
|
||||
if (constructorList.length == 1) {
|
||||
CtConstructor constructor = constructorList[0];
|
||||
constructor.insertAfter("{" + TomcatProfilerConstant.CLASS_NAME_REQUEST_DATA_TRACER + ".putSqlQuery(" + TomcatProfilerConstant.REQ_DATA_TYPE_DB_QUERY + ",$2); }");
|
||||
}
|
||||
}
|
||||
|
||||
private static void updateExecuteQueryMethod(ClassPool classPool, CtClass cc) throws Exception {
|
||||
private void updateExecuteQueryMethod(CtClass cc) throws Exception {
|
||||
CtMethod serviceMethod = cc.getDeclaredMethod("execute", null);
|
||||
serviceMethod.insertAfter("{" + TomcatProfilerConstant.CLASS_NAME_REQUEST_DATA_TRACER + ".put(" + TomcatProfilerConstant.REQ_DATA_TYPE_DB_EXECUTE_QUERY + "); }");
|
||||
}
|
||||
|
||||
@@ -12,18 +12,22 @@ public class MSSQLResultSetModifier extends AbstractModifier {
|
||||
|
||||
private static final Logger logger = Logger.getLogger(MSSQLResultSetModifier.class);
|
||||
|
||||
public byte[] modify(ClassPool classPool, ClassLoader classLoader, String javassistClassName, byte[] classFileBuffer) {
|
||||
logger.info("Modifing. %s", javassistClassName);
|
||||
checkLibrary(classPool, javassistClassName, classLoader);
|
||||
return changeMethod(classPool, classLoader, javassistClassName, classFileBuffer);
|
||||
public MSSQLResultSetModifier(ClassPool classPool) {
|
||||
super(classPool);
|
||||
}
|
||||
|
||||
private byte[] changeMethod(ClassPool classPool, ClassLoader classLoader, String javassistClassName, byte[] classfileBuffer) {
|
||||
public byte[] modify(ClassLoader classLoader, String javassistClassName, byte[] classFileBuffer) {
|
||||
logger.info("Modifing. %s", javassistClassName);
|
||||
checkLibrary(classLoader, javassistClassName);
|
||||
return changeMethod(javassistClassName, classFileBuffer);
|
||||
}
|
||||
|
||||
private byte[] changeMethod(String javassistClassName, byte[] classfileBuffer) {
|
||||
try {
|
||||
CtClass cc = classPool.get(javassistClassName);
|
||||
|
||||
updateNextMethod(classPool, cc);
|
||||
updateCloseMethod(classPool, cc);
|
||||
updateNextMethod(cc);
|
||||
updateCloseMethod(cc);
|
||||
|
||||
printClassConvertComplete(javassistClassName);
|
||||
|
||||
@@ -35,12 +39,12 @@ public class MSSQLResultSetModifier extends AbstractModifier {
|
||||
return null;
|
||||
}
|
||||
|
||||
private static void updateNextMethod(ClassPool classPool, CtClass cc) throws Exception {
|
||||
private void updateNextMethod(CtClass cc) throws Exception {
|
||||
CtMethod serviceMethod1 = cc.getDeclaredMethod("next", null);
|
||||
serviceMethod1.insertBefore("{" + TomcatProfilerConstant.CLASS_NAME_REQUEST_DATA_TRACER + ".updateFetchCount(); }");
|
||||
}
|
||||
|
||||
private static void updateCloseMethod(ClassPool classPool, CtClass cc) throws Exception {
|
||||
private void updateCloseMethod(CtClass cc) throws Exception {
|
||||
CtMethod serviceMethod1 = cc.getDeclaredMethod("close", null);
|
||||
serviceMethod1.insertBefore("{" + TomcatProfilerConstant.CLASS_NAME_REQUEST_DATA_TRACER + ".addResultSetData(); }");
|
||||
}
|
||||
|
||||
@@ -9,19 +9,24 @@ import com.profiler.logging.Logger;
|
||||
import com.profiler.modifier.AbstractModifier;
|
||||
|
||||
public class MSSQLStatementModifier extends AbstractModifier {
|
||||
|
||||
private static final Logger logger = Logger.getLogger(MSSQLStatementModifier.class);
|
||||
|
||||
public byte[] modify(ClassPool classPool, ClassLoader classLoader, String javassistClassName, byte[] classFileBuffer) {
|
||||
logger.info("Modifing. %s", javassistClassName);
|
||||
checkLibrary(classPool, javassistClassName, classLoader);
|
||||
return changeMethod(classPool, classLoader, javassistClassName, classFileBuffer);
|
||||
public MSSQLStatementModifier(ClassPool classPool) {
|
||||
super(classPool);
|
||||
}
|
||||
|
||||
private byte[] changeMethod(ClassPool classPool, ClassLoader classLoader, String javassistClassName, byte[] classfileBuffer) {
|
||||
public byte[] modify(ClassLoader classLoader, String javassistClassName, byte[] classFileBuffer) {
|
||||
logger.info("Modifing. %s", javassistClassName);
|
||||
checkLibrary(classLoader, javassistClassName);
|
||||
return changeMethod(javassistClassName, classFileBuffer);
|
||||
}
|
||||
|
||||
private byte[] changeMethod(String javassistClassName, byte[] classfileBuffer) {
|
||||
try {
|
||||
CtClass cc = classPool.get(javassistClassName);
|
||||
|
||||
updateExecuteQueryMethod(classPool, cc);
|
||||
updateExecuteQueryMethod(cc);
|
||||
|
||||
printClassConvertComplete(javassistClassName);
|
||||
|
||||
@@ -33,7 +38,7 @@ public class MSSQLStatementModifier extends AbstractModifier {
|
||||
return null;
|
||||
}
|
||||
|
||||
private static void updateExecuteQueryMethod(ClassPool classPool, CtClass cc) throws Exception {
|
||||
private void updateExecuteQueryMethod(CtClass cc) throws Exception {
|
||||
CtClass[] params = new CtClass[1];
|
||||
params[0] = classPool.getCtClass("java.lang.String");
|
||||
CtMethod serviceMethod = cc.getDeclaredMethod("executeQuery", params);
|
||||
|
||||
@@ -12,19 +12,23 @@ public class MySQLConnectionImplModifier extends AbstractModifier {
|
||||
|
||||
private static final Logger logger = Logger.getLogger(MySQLConnectionImplModifier.class);
|
||||
|
||||
public byte[] modify(ClassPool classPool, ClassLoader classLoader, String javassistClassName, byte[] classFileBuffer) {
|
||||
logger.info("Modifing. %s", javassistClassName);
|
||||
checkLibrary(classPool, javassistClassName, classLoader);
|
||||
return changeMethods(classPool, classLoader, javassistClassName, classFileBuffer);
|
||||
public MySQLConnectionImplModifier(ClassPool classPool) {
|
||||
super(classPool);
|
||||
}
|
||||
|
||||
private byte[] changeMethods(ClassPool classPool, ClassLoader classLoader, String javassistClassName, byte[] classfileBuffer) {
|
||||
public byte[] modify(ClassLoader classLoader, String javassistClassName, byte[] classFileBuffer) {
|
||||
logger.info("Modifing. %s", javassistClassName);
|
||||
checkLibrary(classLoader, javassistClassName);
|
||||
return changeMethods(javassistClassName, classFileBuffer);
|
||||
}
|
||||
|
||||
private byte[] changeMethods(String javassistClassName, byte[] classfileBuffer) {
|
||||
try {
|
||||
CtClass cc = classPool.get(javassistClassName);
|
||||
|
||||
updateGetInstanceMethod(classPool, cc);
|
||||
updateCreateStatementMethod(classPool, cc);
|
||||
updateCloseMethod(classPool, cc);
|
||||
updateGetInstanceMethod(cc);
|
||||
updateCreateStatementMethod(cc);
|
||||
updateCloseMethod(cc);
|
||||
|
||||
printClassConvertComplete(javassistClassName);
|
||||
|
||||
@@ -36,12 +40,12 @@ public class MySQLConnectionImplModifier extends AbstractModifier {
|
||||
return null;
|
||||
}
|
||||
|
||||
private static void updateCreateStatementMethod(ClassPool classPool, CtClass cc) throws Exception {
|
||||
private void updateCreateStatementMethod(CtClass cc) throws Exception {
|
||||
CtMethod method = cc.getDeclaredMethod("createStatement", null);
|
||||
method.insertAfter("{" + TomcatProfilerConstant.CLASS_NAME_REQUEST_DATA_TRACER + ".put(" + TomcatProfilerConstant.REQ_DATA_TYPE_DB_CREATE_STATEMENT + "); }");
|
||||
}
|
||||
|
||||
private static void updateGetInstanceMethod(ClassPool classPool, CtClass cc) throws Exception {
|
||||
private void updateGetInstanceMethod(CtClass cc) throws Exception {
|
||||
CtClass[] params = new CtClass[5];
|
||||
params[0] = classPool.getCtClass("java.lang.String");
|
||||
params[1] = classPool.getCtClass("int");
|
||||
@@ -53,7 +57,7 @@ public class MySQLConnectionImplModifier extends AbstractModifier {
|
||||
method.insertAfter("{" + TomcatProfilerConstant.CLASS_NAME_REQUEST_DATA_TRACER + ".putConnection(" + TomcatProfilerConstant.REQ_DATA_TYPE_DB_GET_CONNECTION + ",$5); }");
|
||||
}
|
||||
|
||||
private static void updateCloseMethod(ClassPool classPool, CtClass cc) throws Exception {
|
||||
private void updateCloseMethod(CtClass cc) throws Exception {
|
||||
CtMethod method = cc.getDeclaredMethod("close", null);
|
||||
method.insertAfter("{" + TomcatProfilerConstant.CLASS_NAME_REQUEST_DATA_TRACER + ".put(" + TomcatProfilerConstant.REQ_DATA_TYPE_DB_CLOSE_CONNECTION + "); }");
|
||||
}
|
||||
|
||||
@@ -10,22 +10,26 @@ import com.profiler.logging.Logger;
|
||||
import com.profiler.modifier.AbstractModifier;
|
||||
|
||||
public class MySQLPreparedStatementModifier extends AbstractModifier {
|
||||
|
||||
private static final Logger logger = Logger.getLogger(MySQLPreparedStatementModifier.class);
|
||||
|
||||
public byte[] modify(ClassPool classPool, ClassLoader classLoader, String javassistClassName, byte[] classFileBuffer) {
|
||||
logger.info("Modifing. %s", javassistClassName);
|
||||
checkLibrary(classPool, javassistClassName, classLoader);
|
||||
return changeMethod(classPool, classLoader, javassistClassName, classFileBuffer);
|
||||
public MySQLPreparedStatementModifier(ClassPool classPool) {
|
||||
super(classPool);
|
||||
}
|
||||
|
||||
private byte[] changeMethod(ClassPool classPool, ClassLoader classLoader, String javassistClassName, byte[] classfileBuffer) {
|
||||
public byte[] modify(ClassLoader classLoader, String javassistClassName, byte[] classFileBuffer) {
|
||||
logger.info("Modifing. %s", javassistClassName);
|
||||
checkLibrary(classLoader, javassistClassName);
|
||||
return changeMethod(javassistClassName, classFileBuffer);
|
||||
}
|
||||
|
||||
private byte[] changeMethod(String javassistClassName, byte[] classfileBuffer) {
|
||||
try {
|
||||
CtClass cc = classPool.get(javassistClassName);
|
||||
|
||||
updateSetInternalMethod(classPool, cc);
|
||||
updateExecuteQueryMethod(classPool, cc);
|
||||
updateConstructor(classPool, cc);
|
||||
updateSetInternalMethod(cc);
|
||||
updateExecuteQueryMethod(cc);
|
||||
updateConstructor(cc);
|
||||
|
||||
printClassConvertComplete(javassistClassName);
|
||||
|
||||
@@ -37,7 +41,7 @@ public class MySQLPreparedStatementModifier extends AbstractModifier {
|
||||
return null;
|
||||
}
|
||||
|
||||
private static void updateSetInternalMethod(ClassPool classPool, CtClass cc) throws Exception {
|
||||
private void updateSetInternalMethod(CtClass cc) throws Exception {
|
||||
CtClass[] params1 = new CtClass[2];
|
||||
params1[0] = classPool.getCtClass("int");
|
||||
params1[1] = classPool.getCtClass("java.lang.String");
|
||||
@@ -53,7 +57,7 @@ public class MySQLPreparedStatementModifier extends AbstractModifier {
|
||||
method2.insertBefore("{" + TomcatProfilerConstant.CLASS_NAME_REQUEST_DATA_TRACER + ".putSqlParam($1,$2); }");
|
||||
}
|
||||
|
||||
private static void updateConstructor(ClassPool classPool, CtClass cc) throws Exception {
|
||||
private void updateConstructor(CtClass cc) throws Exception {
|
||||
CtConstructor[] constructorList = cc.getConstructors();
|
||||
if (constructorList.length == 3) {
|
||||
for (CtConstructor constructor : constructorList) {
|
||||
@@ -65,7 +69,7 @@ public class MySQLPreparedStatementModifier extends AbstractModifier {
|
||||
}
|
||||
}
|
||||
|
||||
private static void updateExecuteQueryMethod(ClassPool classPool, CtClass cc) throws Exception {
|
||||
private void updateExecuteQueryMethod(CtClass cc) throws Exception {
|
||||
CtMethod method = cc.getDeclaredMethod("executeQuery", null);
|
||||
method.insertAfter("{" + TomcatProfilerConstant.CLASS_NAME_REQUEST_DATA_TRACER + ".put(" + TomcatProfilerConstant.REQ_DATA_TYPE_DB_EXECUTE_QUERY + "); }");
|
||||
}
|
||||
|
||||
@@ -12,18 +12,22 @@ 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("Modifing. %s", javassistClassName);
|
||||
checkLibrary(classPool, javassistClassName, classLoader);
|
||||
return changeMethod(classPool, classLoader, javassistClassName, classFileBuffer);
|
||||
public MySQLResultSetModifier(ClassPool classPool) {
|
||||
super(classPool);
|
||||
}
|
||||
|
||||
private byte[] changeMethod(ClassPool classPool, ClassLoader classLoader, String javassistClassName, byte[] classfileBuffer) {
|
||||
public byte[] modify(ClassLoader classLoader, String javassistClassName, byte[] classFileBuffer) {
|
||||
logger.info("Modifing. %s", javassistClassName);
|
||||
checkLibrary(classLoader, javassistClassName);
|
||||
return changeMethod(javassistClassName, classFileBuffer);
|
||||
}
|
||||
|
||||
private byte[] changeMethod(String javassistClassName, byte[] classfileBuffer) {
|
||||
try {
|
||||
CtClass cc = classPool.get(javassistClassName);
|
||||
|
||||
updateNextMethod(classPool, cc);
|
||||
updateCloseMethod(classPool, cc);
|
||||
updateNextMethod(cc);
|
||||
updateCloseMethod(cc);
|
||||
|
||||
printClassConvertComplete(javassistClassName);
|
||||
|
||||
@@ -35,12 +39,12 @@ public class MySQLResultSetModifier extends AbstractModifier {
|
||||
return null;
|
||||
}
|
||||
|
||||
private static void updateNextMethod(ClassPool classPool, CtClass cc) throws Exception {
|
||||
private void updateNextMethod(CtClass cc) throws Exception {
|
||||
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 {
|
||||
private void updateCloseMethod(CtClass cc) throws Exception {
|
||||
CtMethod method = cc.getDeclaredMethod("close", null);
|
||||
method.insertBefore("{" + TomcatProfilerConstant.CLASS_NAME_REQUEST_DATA_TRACER + ".addResultSetData(); }");
|
||||
}
|
||||
|
||||
@@ -12,17 +12,21 @@ 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("Modifing. %s", javassistClassName);
|
||||
checkLibrary(classPool, javassistClassName, classLoader);
|
||||
return changeMethod(classPool, classLoader, javassistClassName, classFileBuffer);
|
||||
public MySQLStatementModifier(ClassPool classPool) {
|
||||
super(classPool);
|
||||
}
|
||||
|
||||
private byte[] changeMethod(ClassPool classPool, ClassLoader classLoader, String javassistClassName, byte[] classfileBuffer) {
|
||||
public byte[] modify(ClassLoader classLoader, String javassistClassName, byte[] classFileBuffer) {
|
||||
logger.info("Modifing. %s", javassistClassName);
|
||||
checkLibrary(classLoader, javassistClassName);
|
||||
return changeMethod(javassistClassName, classFileBuffer);
|
||||
}
|
||||
|
||||
private byte[] changeMethod(String javassistClassName, byte[] classfileBuffer) {
|
||||
try {
|
||||
CtClass cc = classPool.get(javassistClassName);
|
||||
|
||||
updateExecuteQueryMethod(classPool, cc);
|
||||
updateExecuteQueryMethod(cc);
|
||||
|
||||
printClassConvertComplete(javassistClassName);
|
||||
|
||||
@@ -34,7 +38,7 @@ public class MySQLStatementModifier extends AbstractModifier {
|
||||
return null;
|
||||
}
|
||||
|
||||
private static void updateExecuteQueryMethod(ClassPool classPool, CtClass cc) throws Exception {
|
||||
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);
|
||||
|
||||
+14
-10
@@ -13,19 +13,23 @@ public class OraclePreparedStatementModifier extends AbstractModifier {
|
||||
|
||||
private static final Logger logger = Logger.getLogger(OraclePreparedStatementModifier.class);
|
||||
|
||||
public byte[] modify(ClassPool classPool, ClassLoader classLoader, String javassistClassName, byte[] classFileBuffer) {
|
||||
public OraclePreparedStatementModifier(ClassPool classPool) {
|
||||
super(classPool);
|
||||
}
|
||||
|
||||
public byte[] modify(ClassLoader classLoader, String javassistClassName, byte[] classFileBuffer) {
|
||||
logger.info("Modifing. %s", javassistClassName);
|
||||
checkLibrary(classPool, javassistClassName, classLoader);
|
||||
return changeMethod(classPool, classLoader, javassistClassName, classFileBuffer);
|
||||
checkLibrary(classLoader, javassistClassName);
|
||||
return changeMethod(javassistClassName, classFileBuffer);
|
||||
}
|
||||
|
||||
private byte[] changeMethod(ClassPool classPool, ClassLoader classLoader, String javassistClassName, byte[] classfileBuffer) {
|
||||
private byte[] changeMethod(String javassistClassName, byte[] classfileBuffer) {
|
||||
try {
|
||||
CtClass cc = classPool.get(javassistClassName);
|
||||
|
||||
updateSetInternalMethod(classPool, cc);
|
||||
updateExecuteMethod(classPool, cc);
|
||||
updateConstructor(classPool, cc);
|
||||
updateSetInternalMethod(cc);
|
||||
updateExecuteMethod(cc);
|
||||
updateConstructor(cc);
|
||||
|
||||
printClassConvertComplete(javassistClassName);
|
||||
|
||||
@@ -37,7 +41,7 @@ public class OraclePreparedStatementModifier extends AbstractModifier {
|
||||
return null;
|
||||
}
|
||||
|
||||
private static void updateSetInternalMethod(ClassPool classPool, CtClass cc) throws Exception {
|
||||
private void updateSetInternalMethod(CtClass cc) throws Exception {
|
||||
CtClass[] params1 = new CtClass[2];
|
||||
params1[0] = classPool.getCtClass("int");
|
||||
params1[1] = classPool.getCtClass("java.lang.String");
|
||||
@@ -56,7 +60,7 @@ public class OraclePreparedStatementModifier extends AbstractModifier {
|
||||
// ".putSqlParam($1,$2); {");
|
||||
}
|
||||
|
||||
private static void updateConstructor(ClassPool classPool, CtClass cc) throws Exception {
|
||||
private void updateConstructor(CtClass cc) throws Exception {
|
||||
CtConstructor[] constructorList = cc.getConstructors();
|
||||
|
||||
for (CtConstructor constructor : constructorList) {
|
||||
@@ -67,7 +71,7 @@ public class OraclePreparedStatementModifier extends AbstractModifier {
|
||||
}
|
||||
}
|
||||
|
||||
private static void updateExecuteMethod(ClassPool classPool, CtClass cc) throws Exception {
|
||||
private void updateExecuteMethod(CtClass cc) throws Exception {
|
||||
CtMethod method = cc.getDeclaredMethod("execute", null);
|
||||
method.insertAfter("{" + TomcatProfilerConstant.CLASS_NAME_REQUEST_DATA_TRACER + ".put(" + TomcatProfilerConstant.REQ_DATA_TYPE_DB_EXECUTE_QUERY + "); }");
|
||||
}
|
||||
|
||||
@@ -12,18 +12,22 @@ public class OracleResultSetModifier extends AbstractModifier {
|
||||
|
||||
private static final Logger logger = Logger.getLogger(OracleResultSetModifier.class);
|
||||
|
||||
public byte[] modify(ClassPool classPool, ClassLoader classLoader, String javassistClassName, byte[] classFileBuffer) {
|
||||
logger.info("Modifing. %s", javassistClassName);
|
||||
checkLibrary(classPool, javassistClassName, classLoader);
|
||||
return changeMethod(classPool, classLoader, javassistClassName, classFileBuffer);
|
||||
public OracleResultSetModifier(ClassPool classPool) {
|
||||
super(classPool);
|
||||
}
|
||||
|
||||
private byte[] changeMethod(ClassPool classPool, ClassLoader classLoader, String javassistClassName, byte[] classfileBuffer) {
|
||||
public byte[] modify(ClassLoader classLoader, String javassistClassName, byte[] classFileBuffer) {
|
||||
logger.info("Modifing. %s", javassistClassName);
|
||||
checkLibrary(classLoader, javassistClassName);
|
||||
return changeMethod(javassistClassName, classFileBuffer);
|
||||
}
|
||||
|
||||
private byte[] changeMethod(String javassistClassName, byte[] classfileBuffer) {
|
||||
try {
|
||||
CtClass cc = classPool.get(javassistClassName);
|
||||
|
||||
updateNextMethod(classPool, cc);
|
||||
updateCloseMethod(classPool, cc);
|
||||
updateNextMethod(cc);
|
||||
updateCloseMethod(cc);
|
||||
|
||||
printClassConvertComplete(javassistClassName);
|
||||
|
||||
@@ -35,12 +39,12 @@ public class OracleResultSetModifier extends AbstractModifier {
|
||||
return null;
|
||||
}
|
||||
|
||||
private static void updateNextMethod(ClassPool classPool, CtClass cc) throws Exception {
|
||||
private void updateNextMethod(CtClass cc) throws Exception {
|
||||
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 {
|
||||
private void updateCloseMethod(CtClass cc) throws Exception {
|
||||
CtMethod method = cc.getDeclaredMethod("close", null);
|
||||
method.insertBefore("{" + TomcatProfilerConstant.CLASS_NAME_REQUEST_DATA_TRACER + ".addResultSetData(); }");
|
||||
}
|
||||
|
||||
@@ -11,17 +11,21 @@ import com.profiler.modifier.AbstractModifier;
|
||||
public class OracleStatementModifier extends AbstractModifier {
|
||||
private static final Logger logger = Logger.getLogger(OracleStatementModifier.class);
|
||||
|
||||
public byte[] modify(ClassPool classPool, ClassLoader classLoader, String javassistClassName, byte[] classFileBuffer) {
|
||||
logger.info("Modifing. %s", javassistClassName);
|
||||
checkLibrary(classPool, javassistClassName, classLoader);
|
||||
return changeMethod(classPool, classLoader, javassistClassName, classFileBuffer);
|
||||
public OracleStatementModifier(ClassPool classPool) {
|
||||
super(classPool);
|
||||
}
|
||||
|
||||
private byte[] changeMethod(ClassPool classPool, ClassLoader classLoader, String javassistClassName, byte[] classfileBuffer) {
|
||||
public byte[] modify(ClassLoader classLoader, String javassistClassName, byte[] classFileBuffer) {
|
||||
logger.info("Modifing. %s", javassistClassName);
|
||||
checkLibrary(classLoader, javassistClassName);
|
||||
return changeMethod(javassistClassName, classFileBuffer);
|
||||
}
|
||||
|
||||
private byte[] changeMethod(String javassistClassName, byte[] classfileBuffer) {
|
||||
try {
|
||||
CtClass cc = classPool.get(javassistClassName);
|
||||
|
||||
updateExecuteQueryMethod(classPool, cc);
|
||||
updateExecuteQueryMethod(cc);
|
||||
|
||||
printClassConvertComplete(javassistClassName);
|
||||
|
||||
@@ -32,7 +36,7 @@ public class OracleStatementModifier extends AbstractModifier {
|
||||
return null;
|
||||
}
|
||||
|
||||
private static void updateExecuteQueryMethod(ClassPool classPool, CtClass cc) throws Exception {
|
||||
private void updateExecuteQueryMethod(CtClass cc) throws Exception {
|
||||
CtClass[] params = new CtClass[1];
|
||||
params[0] = classPool.getCtClass("java.lang.String");
|
||||
// CtMethod serviceMethod=cc.getDeclaredMethod("executeQuery", params);
|
||||
|
||||
Reference in New Issue
Block a user