javassist学习

最近学习了javassist的使用方法。 
javassist是一个字节码类库，可以用他来动态生成类，动态修改类等等。

下面是如果用javassist来动态创建一个类的demol

我们需要创建的目标类，如下：

Java代码  

1. public class JavassistClass{
2. private String name="default";
3. public JavassistClass(){
4. name="me";
5. }
6. public String getName() {
7. return name;
8. }
9. public void setName(String name) {
10. this.name = name;
11. }
12. public void execute(){
13. System.out.println(name);
14. System.out.println("execute ok");
15. }
16. }

如下为用javassist来创建类的代码：

Java代码  

1. import java.lang.reflect.Method;
2. import java.lang.reflect.Modifier;
3. import javassist.ClassPool;
4. import javassist.CtClass;
5. import javassist.CtConstructor;
6. import javassist.CtField;
7. import javassist.CtField.Initializer;
8. import javassist.CtMethod;
9. import javassist.CtNewMethod;
10. public class JavassistLearn{
11. public static void main(String[] args) throws Exception{
12. ClassPool cp=ClassPool.getDefault();
13. CtClass ctClass=cp.makeClass("com.slovef.JavassistClass");
14. StringBuffer body=null;
15. //参数  1：属性类型  2：属性名称  3：所属类CtClass
16. CtField ctField=new CtField(cp.get("java.lang.String"), "name", ctClass);
17. ctField.setModifiers(Modifier.PRIVATE);
18. //设置name属性的get set方法
19. ctClass.addMethod(CtNewMethod.setter("setName", ctField));
20. ctClass.addMethod(CtNewMethod.getter("getName", ctField));
21. ctClass.addField(ctField, Initializer.constant("default"));
22. //参数  1：参数类型   2：所属类CtClass
23. CtConstructor ctConstructor=new CtConstructor(new CtClass[]{}, ctClass);
24. body=new StringBuffer();
25. body.append("{\n name=\"me\";\n}");
26. ctConstructor.setBody(body.toString());
27. ctClass.addConstructor(ctConstructor);
28. //参数：  1：返回类型  2：方法名称  3：传入参数类型  4：所属类CtClass
29. CtMethod ctMethod=new CtMethod(CtClass.voidType,"execute",new CtClass[]{},ctClass);
30. ctMethod.setModifiers(Modifier.PUBLIC);
31. body=new StringBuffer();
32. body.append("{\n System.out.println(name);");
33. body.append("\n System.out.println(\"execute ok\");");
34. body.append("\n return ;");
35. body.append("\n}");
36. ctMethod.setBody(body.toString());
37. ctClass.addMethod(ctMethod);
38. Class<?> c=ctClass.toClass();
39. Object o=c.newInstance();
40. Method method=o.getClass().getMethod("execute", new Class[]{});
41. //调用字节码生成类的execute方法
42. method.invoke(o, new Object[]{});
43. }
44. }

javassist还有一个比较常见的用途是AOP，比如对一些类统一加权限过滤，加日志监控等等。 
下面示例如何使用javassist来进行AOP拦截处理

我们对上面例子的JavassistClass类的getName()方法进行拦截前置处理

Java代码  

1. package com.taobao.relationrecommend.web.util;
2. import java.lang.reflect.Method;
3. import javassist.util.proxy.MethodFilter;
4. import javassist.util.proxy.MethodHandler;
5. import javassist.util.proxy.ProxyFactory;
6. public class JavassistLearn{
7. public static void main(String[] args) throws Exception{
8. ProxyFactory factory=new ProxyFactory();
9. //设置父类，ProxyFactory将会动态生成一个类，继承该父类
10. factory.setSuperclass(JavassistClass.class);
11. //设置过滤器，判断哪些方法调用需要被拦截
12. factory.setFilter(new MethodFilter() {
13. @Override
14. public boolean isHandled(Method m) {
15. if(m.getName().equals("getName")){
16. return true;
17. }
18. return false;
19. }
20. });
21. //设置拦截处理
22. factory.setHandler(new MethodHandler() {
23. @Override
24. public Object invoke(Object self, Method thisMethod, Method proceed,
25. Object[] args) throws Throwable {
26. //拦截后前置处理，改写name属性的内容
27. //实际情况可根据需求修改
28. JavassistClass o=(JavassistClass) self;
29. o.setName("haha");
30. return proceed.invoke(self, args);
31. }
32. });
33. Class<?> c=factory.createClass();
34. JavassistClass object=(JavassistClass) c.newInstance();
35. System.out.println(object.getName());
36. }
37. }