自定义实现Java动态代理

转自：https://www.cnblogs.com/rjzheng/p/8750265.html

一 借助JDK的API实现：

1.先创建一个接口，并实现它

public interface Person {
    void eat();
}

public class PersonImpl implements Person {
    @Override
    public void eat() {
        System.out.println("eat............");
    }
}

2.实现InvocationHandler：每一个动态代理类都要实现这个接口

public class PersonInvocationHandler implements InvocationHandler {

    //我们要代理的真实对象
    private Object obj;

    public PersonInvocationHandler(Object obj) {
        this.obj = obj;
    }

    /**
     * Object proxy:
     * 1. 可以使用反射获取代理对象的信息（也就是proxy.getClass().getName()）。
     * 2. 可以将代理对象返回以进行连续调用，这就是proxy存在的目的，因为this并不是代理对象。
     */
    @Override
    public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
        System.out.println("before eat.......");
        //当代理对象调用真实对象的方法时，其会自动的跳转到代理对象关联的handler对象的invoke方法来进行调用
        method.invoke(obj, args);
        System.out.println("after eat.......");
        return null;
    }
}

3.测试

public class JdkTest {
    public static void main(String[] args) {
        PersonInvocationHandler personInvocationHandler = new PersonInvocationHandler(new PersonImpl());

        Person personProxy = (Person) Proxy.newProxyInstance(PersonImpl.class.getClassLoader(),
                PersonImpl.class.getInterfaces(), personInvocationHandler);

        personProxy.eat();
    }
}

返回结果：

 二 自定义动态代理

1.自定义InvocationHandler

public interface MyInvocationHandler {
    public Object invoke(Object proxy, Method method, Object[] args) throws Throwable;
}

2.实现自定义的MyInvocationHandler

public class PersonInvocationHandler implements MyInvocationHandler {

    //我们要代理的真实对象
    private Object obj;

    public PersonInvocationHandler(Object obj) {
        this.obj = obj;
    }

    /**
     * Object proxy:
     * 1. 可以使用反射获取代理对象的信息（也就是proxy.getClass().getName()）。
     * 2. 可以将代理对象返回以进行连续调用，这就是proxy存在的目的，因为this并不是代理对象。
     */
    @Override
    public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
        System.out.println("before eat.......");
        //当代理对象调用真实对象的方法时，其会自动的跳转到代理对象关联的handler对象的invoke方法来进行调用
        method.invoke(obj, args);
        System.out.println("after eat.......");
        return null;
    }
}

3.自定义ClassLoader

public class MyClassLoader extends ClassLoader{

    private File classPathFile;

    public MyClassLoader() {
        String classPath = MyClassLoader.class.getResource("").getPath();
        this.classPathFile = new File(classPath);
    }

    public Class<?> findClass(String name) {
        String className = MyClassLoader.class.getPackage().getName() + "." + name;
        if (classPathFile != null){
            File file = new File(classPathFile, name + ".class");
            FileInputStream inputStream = null;
            ByteArrayOutputStream outputStream = null;
            try {
                inputStream = new FileInputStream(file);
                outputStream = new ByteArrayOutputStream();
                byte[] buf = new byte[1024];
                int len ;
                while ((len = inputStream.read(buf)) != -1){
                    outputStream.write(buf, 0, len);
                }
                return defineClass(className, outputStream.toByteArray(), 0, outputStream.size());
            }catch (Exception e){
                e.printStackTrace();
            }finally {
                if(null!=inputStream){
                    try {
                        inputStream.close();
                    } catch (IOException e) {
                        // TODO Auto-generated catch block
                        e.printStackTrace();
                    }
                }
                if(null!=outputStream){
                    try {
                        outputStream.close();
                    } catch (IOException e) {
                        // TODO Auto-generated catch block
                        e.printStackTrace();
                    }
                }
            }
        }
        return null;
    }
}

4.自定义Proxy代理类

　　主要流程：1.将源码输入到Java文件

　　　　　　　2.将Java文件编译成class文件

　　　　　　　3.将class加载进jvm

　　　　　　　4.返回代理类对象

public class MyProxy {
    public static final String ln = "
";

    public static Object newProxyInstance(MyClassLoader classLoader,
                                          Class<?>[] interfaces, MyInvocationHandler handler) {
        try {
            //1.java源码
            String src = generateSrc(interfaces);
            //2.将源码输出到Java文件中
            String filePath = MyProxy.class.getResource("").getPath();
            System.out.println(filePath);
            File file = new File(filePath + "$Proxy0.java");
            FileWriter fw = new FileWriter(file);
            fw.write(src);
            fw.flush();
            fw.close();
            //3.将Java文件编译成class文件
            JavaCompiler javaCompiler = ToolProvider.getSystemJavaCompiler();
            StandardJavaFileManager manager = javaCompiler.getStandardFileManager(null, null, null);
            Iterable iterable = manager.getJavaFileObjects(file);
            JavaCompiler.CompilationTask task = javaCompiler.getTask(null, manager, null, null,null, iterable);
            task.call();
            manager.close();
            //4.将class加载进jvm
            Class proxyClass = classLoader.findClass("$Proxy0");
            file.delete();
            //5.返回代理类对象
            Constructor constructor = proxyClass.getConstructor(MyInvocationHandler.class);
            return constructor.newInstance(handler);
        }catch (Exception e){
            e.printStackTrace();
        }
        return null;
    }

    //Java源码
    private static String generateSrc(Class<?>[] interfaces) {
        // TODO Auto-generated method stub
        StringBuffer sb = new StringBuffer();
        sb.append("package xin.hangzhi.jdk.proxy.demo.custom;" + ln);
        sb.append("import java.lang.reflect.Method;" + ln);
        sb.append("public class $Proxy0 implements " + interfaces[0].getName() + "{" + ln);
        sb.append("private MyInvocationHandler h;"+ln);
        sb.append("public $Proxy0(MyInvocationHandler h) { " + ln);
        sb.append("this.h = h;"+ln);
        sb.append("}" + ln);
        for (Method m : interfaces[0].getMethods()) {
            sb.append("public " + m.getReturnType().getName() + " "
                    + m.getName() + "() {" + ln);
            sb.append("try{" + ln);
            sb.append("Method m = " + interfaces[0].getName()
                    + ".class.getMethod("" + m.getName()
                    + "",new Class[]{});" + ln);
            sb.append("this.h.invoke(this,m,null);" + ln);
            sb.append("}catch(Throwable e){" + ln);
            sb.append("e.printStackTrace();" + ln);
            sb.append("}"+ln);
            sb.append("}"+ln);
        }
        sb.append("}" + ln);
        return sb.toString();
    }
}

5.测试结果