使用javassist动态注入代码

    关于java字节码的处理,目前有很多工具,如bcel,asm。不过这些都需要直接跟虚拟机指令打交道。如果你不想了解虚拟机指令,可以采用javassist。javassist是jboss的一个子项目,其主要的优点,在于简单,而且快速。直接使用java编码的形式,而不需要了解虚拟机指令,就能动态改变类的结构,或者动态生成类。
    下面通过一个简单的例子,通过javassist来实现如何动态注入代码。
    假设,存在类A,如下:

java 代码

  1. public class A { 
  2. public void method() { 
  3. for (int i = 0; i < 1000000; i++) { 
  4.         } 
  5.         System.out.println("method1"); 
  6.     } 

测试类B如下:

java 代码

  1. public class B { 
  2. public static void main(String[] args) { 
  3.         A a = new A(); 
  4.         a.method();    
  5.     } 

现在想统计一下method的执行时间,
默认的实现是修改method:

java 代码

  1. public void method() { 
  2. long start = System.currentTimeMillis(); 
  3. for (int i = 0; i < 1000000; i++) { 
  4.         } 
  5.         System.out.println("method1"); 
  6. long end = System.currentTimeMillis(); 
  7.         System.out.println(end - start); 
  8.     } 

如果A的方法很多,统计方法的执行时间的代码就会相应的增加。为了减少工作量,通过动态注入代码的形式来实现。
修改B的main方法:

java 代码

  1. public static void main(String[] args) throws Exception { 
  2. //用于取得字节码类,必须在当前的classpath中,使用全称
  3.       CtClass ctClass = ClassPool.getDefault().get("org.esoft.A"); 
  4. //需要修改的方法名称
  5.       String mname = "method";        
  6.       CtMethod mold = ctClass.getDeclaredMethod(mname); 
  7. //修改原有的方法名称
  8.       String nname = mname + "$impl"; 
  9.       mold.setName(nname); 
  10. //创建新的方法,复制原来的方法
  11.       CtMethod mnew = CtNewMethod.copy(mold, mname, ctClass, null); 
  12. //主要的注入代码
  13.       StringBuffer body = new StringBuffer(); 
  14.       body.append("{\nlong start = System.currentTimeMillis();\n"); 
  15. //调用原有代码,类似于method();($$)表示所有的参数
  16.       body.append(nname + "($$);\n"); 
  17.       body.append("System.out.println(\"Call to method "
  18.                   + mname 
  19.                   + " took \" +\n (System.currentTimeMillis()-start) + "
  20.                   + "\" ms.\");\n"); 
  21.       body.append("}"); 
  22. //替换新方法
  23.       mnew.setBody(body.toString()); 
  24. //增加新方法
  25.       ctClass.addMethod(mnew); 
  26. //类已经更改,注意不能使用A a=new A();,因为在同一个classloader中,不允许装载同一个类两次
  27.       A a=(A)ctClass.toClass().newInstance(); 
  28.       a.method(); 
  29.   } 

这只是简单的一个应用。javassist还提供了很多的功能,用于更改类结构。有兴趣的可以参考相关文档

 

转自http://www.javaeye.com/topic/53740

posted @ 2010-05-02 18:10  心梦帆影  阅读(557)  评论(0编辑  收藏  举报