最近用javassist修改了mybatis中获取连接池中的连接的逻辑做了一点修改;
javassist是在项目编译成class文件之后, javassist对class文件中的方法进行修改,那这一段修改class文件逻辑应该加在什么地方呢?有大神帮忙看一下吗, 谢谢
Javassist提供了两个层面的API: 1)java代码层; 2)字节码层。通过java代码层,开发者即使不熟悉字节码,也可以方便快捷的完成字节码的修改。
示例代码:修改.class中的方法,并注入统计方法的执行时间等代码
/**
* @Test
*
* Use Package javassist
* Refer to the Javassist Core API. [http://www.javassist.org/html/javassist/package-summary.html]
*
*/
public void updateGetUserInfoMethod() throws Exception {
// 创建javassist的类池ClassPool对象
ClassPool pool = new ClassPool();
pool.appendSystemPath();
// 定义类
/*
* CtClass提供了检查类数据(如字段和方法)以及在类中添加新字段、方法和构造函数、以及改变类、父类和接口的方法。
* 但是请注意,Javassist 并未提供删除类中字段、方法或者构造函数的任何方法。
*/
CtClass userServiceClass = pool.get("com.ty.javaagent.UserServiceImpl");
// 需要修改的方法
CtMethod method = userServiceClass.getDeclaredMethod("getUserInfo");
// 修改原有的方法
method.setName("getUserInfo$agent");
// 创建新的方法,复制原来的方法
CtMethod newMethod = CtNewMethod.copy(method, "getUserInfo", userServiceClass, null);
// 注入的代码
StringBuffer body = new StringBuffer();
body.append("{\nlong start = System.currentTimeMillis();\n");
// 调用原有代码,类似于method();($$)表示所有的参数
body.append("getUserInfo$agent($$);\n");
body.append("System.out.println(\" take \" +\n (System.currentTimeMillis()-start) + " + "\" ms.\");\n");
body.append("}");
newMethod.setBody(body.toString());
// 增加新方法
userServiceClass.addMethod(newMethod);
UserServiceImpl userServiceImpl = (UserServiceImpl) userServiceClass.toClass().newInstance();
userServiceImpl.getUserInfo();
}