__BlueCat
2019-09-11 15:10
采纳率: 75%
浏览 438
已采纳

javassist修改字节码的逻辑, 应该在项目运行的哪一步执行?

最近用javassist修改了mybatis中获取连接池中的连接的逻辑做了一点修改;
javassist是在项目编译成class文件之后, javassist对class文件中的方法进行修改,那这一段修改class文件逻辑应该加在什么地方呢?有大神帮忙看一下吗, 谢谢

  • 写回答
  • 关注问题
  • 收藏
  • 邀请回答

1条回答 默认 最新

  • 无形有道 2019-09-11 21:19
    已采纳

    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();
        }
    
    打赏 评论

相关推荐 更多相似问题