javassist学习

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

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

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

	public class JavassistClass{
		private String name="default";
		public JavassistClass(){
			name="me";
		}
	        public String getName() {
			return name;
		}
		public void setName(String name) {
			this.name = name;
		}
		public void execute(){
			System.out.println(name);
			System.out.println("execute ok");
		}
	}

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

import java.lang.reflect.Method;
import java.lang.reflect.Modifier;

import javassist.ClassPool;
import javassist.CtClass;
import javassist.CtConstructor;
import javassist.CtField;
import javassist.CtField.Initializer;
import javassist.CtMethod;
import javassist.CtNewMethod;

public class JavassistLearn{
	
	
	public static void main(String[] args) throws Exception{
		ClassPool cp=ClassPool.getDefault();
		CtClass ctClass=cp.makeClass("com.slovef.JavassistClass");
		
		StringBuffer body=null;
		//参数  1：属性类型  2：属性名称  3：所属类CtClass
		CtField ctField=new CtField(cp.get("java.lang.String"), "name", ctClass);
		ctField.setModifiers(Modifier.PRIVATE);
		//设置name属性的get set方法
		ctClass.addMethod(CtNewMethod.setter("setName", ctField));
		ctClass.addMethod(CtNewMethod.getter("getName", ctField));
		ctClass.addField(ctField, Initializer.constant("default"));
		
		//参数  1：参数类型   2：所属类CtClass
		CtConstructor ctConstructor=new CtConstructor(new CtClass[]{}, ctClass);
		body=new StringBuffer();
		body.append("{\n name=\"me\";\n}");
		ctConstructor.setBody(body.toString());
		ctClass.addConstructor(ctConstructor);
		
		//参数：  1：返回类型  2：方法名称  3：传入参数类型  4：所属类CtClass
		CtMethod ctMethod=new CtMethod(CtClass.voidType,"execute",new CtClass[]{},ctClass);
		ctMethod.setModifiers(Modifier.PUBLIC);
		body=new StringBuffer();
		body.append("{\n System.out.println(name);");
		body.append("\n System.out.println(\"execute ok\");");
		body.append("\n return ;");
		body.append("\n}");
		ctMethod.setBody(body.toString());
		ctClass.addMethod(ctMethod);
		Class<?> c=ctClass.toClass();
		Object o=c.newInstance();
		Method method=o.getClass().getMethod("execute", new Class[]{});
		//调用字节码生成类的execute方法
		method.invoke(o, new Object[]{});
	}

}

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

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

package com.taobao.relationrecommend.web.util;

import java.lang.reflect.Method;

import javassist.util.proxy.MethodFilter;
import javassist.util.proxy.MethodHandler;
import javassist.util.proxy.ProxyFactory;

public class JavassistLearn{
	
	
	public static void main(String[] args) throws Exception{
		ProxyFactory factory=new ProxyFactory();
		//设置父类，ProxyFactory将会动态生成一个类，继承该父类
		factory.setSuperclass(JavassistClass.class);
		//设置过滤器，判断哪些方法调用需要被拦截
		factory.setFilter(new MethodFilter() {
			@Override
			public boolean isHandled(Method m) {
				if(m.getName().equals("getName")){
					return true;
				}
				return false;
			}
		});
		//设置拦截处理
		factory.setHandler(new MethodHandler() {
			@Override
			public Object invoke(Object self, Method thisMethod, Method proceed,
					Object[] args) throws Throwable {
				//拦截后前置处理，改写name属性的内容
				//实际情况可根据需求修改
				JavassistClass o=(JavassistClass) self;
				o.setName("haha");
				return proceed.invoke(self, args);
			}
		});
		
		Class<?> c=factory.createClass();
		JavassistClass object=(JavassistClass) c.newInstance();
		System.out.println(object.getName());
		
	}

}