使用字节码处理框架javassist动态注入代码

官方主页：http://www.csg.is.titech.ac.jp/~chiba/javassist/

Javassist，Cglib等是一些代码增强工具，在运行时刻进行Java字节码增强，虽然速度上稍微慢一点点，但是带来的是代码的简洁，今天用Javassist进行代码增强。

代码：

TestBean.java

1. package javassist.sample;
3. public abstract class TestBean {
5.     public String field;
7.     public abstract String getM();
9.     public abstract void setF(String f);
11.     public String getF() {
12.         return this.field;
13.     }
14. }

TestByteCode.java

1. package javassist.sample;
3. import javassist.ClassPool;
4. import javassist.CtClass;
5. import javassist.CtConstructor;
6. import javassist.CtMethod;
8. public class TestByteCode {
10.     public static void main(String[] args) throws Exception {
11.         
12.         ClassPool pool = ClassPool.getDefault();
13.         CtClass pt = pool.makeClass("asdf", pool.get("javassist.sample.TestBean"));
15.         CtMethod method1 = new CtMethod(pool.get("java.lang.String"), "getM", null, pt);
16.         method1.setBody("{return /"你好/";}");
17.         pt.addMethod(method1);
19.         CtConstructor cc = new CtConstructor(null, pt);
20.         cc.setBody("this.field=/"why?/";");
21.         pt.addConstructor(cc);
23.         CtMethod method2 = new CtMethod(CtClass.voidType, "setF",
24.                 new CtClass[] { pool.get("java.lang.String") }, pt);
25.         method2.setBody("{this.field=$1;}");
26.         pt.addMethod(method2);
28.         Class<?> c = pt.toClass();
29.         TestBean bean = (TestBean) c.newInstance();
31.         System.out.println(bean.getM());
32.         System.out.println(bean.getF());
33.         bean.setF("setf");
34.         System.out.println(bean.getF());
35.     }
36. }

输出为：

你好

why?

setf

可以看到实现了动态的构造，动态实现抽象函数。在进行代码自动化项目中，能够使用字节码增强将大大提高代码的质量，和减少代码的数量。

 

再看一个例子：

A.java

1. package javassist.demo;
3. public class A {
5.     public void method() {
6.         for (int i = 0; i < 1000000; i++) {
7.         }
8.         System.out.println("method1");
9.     }
10. }

JavassistTest.java

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

资料：

http://tsaijun.spaces.live.com/blog/cns!C9F557E0FD3838C1!121.entry

http://www.blogjava.net/zyl/archive/2007/02/10/99171.html

http://www.ibm.com/developerworks/cn/java/j-dyn0916/