Java之javassist实现自动代理-yellowcong

Javaassist是通过操作class字节码文件来修改类的，如果需要Javaassist实现AOP，需要用到Javaassist的ProxyFactory、ProxyObject、MethodHandler三个类，通过这种方法就可以so easy的完成了，我们还可以自己写一个AOP的框架，通过Javassit同cglib一样，可以不依赖于接口的方式来实现代理操作

简单实现代码

这个简单实现，是直接使用的代码的api,没有做类的设计，只是简单将如何做Aop的主要代码写出来了，下面的我写了如何通过父类代理，来简化代理操作。

需要代理的对象

package com.yellowcong.test;public class TestDemo{    private String name;    public String getName() {        return name;    }    public void setName(String name) {        this.name = name;    }    public void say(String name) {        System.out.println("hello"+name);    }}

测试类

直接调用Javassist的api来做的，没有做封装的操作，我们可以自己建立一个拦截器，类似于JDK,然后调用Javassist的方法来解决动态代理，而且Javassist的功能强大，还可以动态编辑class文件

package com.yellowcong.test;import java.lang.reflect.Method;import javassist.util.proxy.MethodHandler;import javassist.util.proxy.ProxyFactory;import javassist.util.proxy.ProxyObject;/****作者:yellowcong*日期:2017/09/01*時間:8:23:56*描述:*/public class Demo8 {    public static void main(String[] args) throws Exception, IllegalAccessException {        //代理对象        ProxyFactory factory = new ProxyFactory();        //设定需要代理的类        factory.setSuperclass(TestDemo.class);        //创建class        Class<?> clazz = factory.createClass();        //实例化对象        TestDemo testDemo = (com.yellowcong.test.TestDemo) clazz.newInstance();        //设置代理对象        ((ProxyObject)testDemo).setHandler(new MethodHandler() {            /**             * @param obj 原来的对象              * @param method  这个是原来类的方法             * @param process JavaAssist获取的进程，是修改后的类方法             * @param args 参数             */            @Override            public Object invoke(Object obj, Method method, Method process, Object[] args) throws Throwable {                Object result = null;                System.out.println(method.getName()+"\t"+process.getName()+"执行前");                result = process.invoke(obj, args);                System.out.println("执行后");                return result;            }        });        testDemo.say("dou");    }}

自己实现AOP框架

定义代理类

代理类需要实现接口MethodHandler，里面有一个invoke的方法，同时，在getProxy类中，写了需要将代理类绑定到拦截器的类

/** * 定义接口 * @author zhangrw * @data 2017/09/01 */class MethodInterceptor implements MethodHandler{    private Object target ;    public MethodInterceptor(Object target) {        this.target = target ;    }    /**     * 执行代理方法     * @param obj     * @param method     * @param process     * @param args     * @return     * @throws Throwable     */    @Override    public Object invoke(Object obj, Method method, Method process, Object[] args) throws Throwable {        Object result = null;        System.out.println(method.getName()+"\t"+process.getName()+"执行前");        result = process.invoke(obj, args);        System.out.println("执行后");        return result;    }    /**     * 获取代理对象     * @return     * @throws Exception      * @throws InstantiationException      */    Object getProxy() throws Exception {        //代理对象        ProxyFactory factory = new ProxyFactory();        //设定需要代理的类        factory.setSuperclass(this.target.getClass());        //创建class        Class<?> clazz = factory.createClass();        //实例化对象 , 这个对象的字节类已经修改了， 是ProxyObject 的子类了        Object targetObj =  clazz.newInstance();        //设定代理的对象，我们设定为这个类，就可以了        ((ProxyObject)targetObj).setHandler(this);        return targetObj;    }}

子拦截器

定义的子拦截器，自需要覆写里面的invoke的方法就可以了，其他的方法默认就可以了

/** * 继承自共同的MethodInterceptor ，然后复写里面的invoke 方法 * @author zhangrw * @data 2017/09/01 */class MyInterceptor extends MethodInterceptor{    public MyInterceptor(Object target) {        super(target);    }    /**     * 执行代理方法     * @param obj     * @param method     * @param process     * @param args     * @return     * @throws Throwable     */    @Override    public Object invoke(Object obj, Method method, Method process, Object[] args) throws Throwable {        Object result = null;        System.out.println(method.getName()+"\t"+process.getName()+"执行前,extend");        result = process.invoke(obj, args);        System.out.println("执行后,Extend");        return result;    }}

调用代理对象

代理对象中，我们可以定义父类代理对象后，在使用子类代理对象，这样就可以节省一些重复代码的编写

/****作者:yellowcong*日期:2017/09/01*時間:8:23:56*描述:*/public class Demo9 {    public static void main(String[] args) throws Exception, IllegalAccessException {        MethodInterceptor parentProxy = new MethodInterceptor(new TestDemo());        TestDemo demo2 = (TestDemo) parentProxy.getProxy();        demo2.say("parent");        //获取代理对象        MyInterceptor proxy = new MyInterceptor(new TestDemo());        TestDemo demo = (TestDemo) proxy.getProxy();        demo.say("dobui");    }}