手撕动态代理,手动编写

对于动态代理生疏的同学可以先看看,这篇文章:https://www.jianshu.com/p/6729457d0509
动态代理的核心就是通过字节码动态生成代理类,思路如下:
1.手动生成动态代理类的java文件
2.通过代码动态编译手动生成的动态代理类
3.将编译好的动态代理类加载到虚拟机中

代码如下

目标类

package com.swh.design.proxy.handwritingproxy;

/**
 * 被代理类接口
 */
public interface Person {

    void findJob();
    // 新增方法
    void findLove();

}
package com.swh.design.proxy.handwritingproxy;

/**
 * 原始类
 */
public class XiaoMing implements Person {

    @Override
    public void findJob() {
        System.out.println("---小明面试---");
    }

    // 新增方法实现
    @Override
    public void findLove() {
        System.out.println("---坠入爱河---");
    }
}

代理类的创建

package com.swh.design.proxy.handwritingproxy;

import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Method;
import java.lang.reflect.Proxy;

/**
 * 代理类
 */
public class JdkProxy  implements MyselfInvocationHandler {

    private Person person;

    public JdkProxy(Person person) {
        this.person = person;
    }


    public Object newProxy(){
        return MyselfProxy.newProxyInstance(new MyselfClassLoader(),person.getClass().getInterfaces(),this);
    }

    @Override
    public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
        System.out.println("---在被代理类执行之前干点坏事---");
        Object invoke = method.invoke(person, args);
        System.out.println("---在被代理类执行之后干点坏事---");
        return invoke;
    }
}

package com.swh.design.proxy.handwritingproxy;

import java.lang.reflect.Method;

public interface MyselfInvocationHandler {

    public Object invoke(Object proxy, Method method, Object[] args) throws Throwable;

}

package com.swh.design.proxy.handwritingproxy;

import java.io.*;

/**
 *  类加载器用于加载类文件
 */
public class MyselfClassLoader extends ClassLoader {

    private File classPathFile;

    public MyselfClassLoader() {
        String path = MyselfClassLoader.class.getResource("").getPath();
        this.classPathFile = new File(path);

    }

    /**
     *   这个方法获取编译好的动态代理.class文件 并加载到虚拟机中
     * @param name
     * @return
     * @throws ClassNotFoundException
     */
    @Override
    protected Class<?> findClass(String name) throws ClassNotFoundException {
        if (classPathFile == null) return null;
        String className = MyselfClassLoader.class.getPackage().getName() + "." + name;

        File file = new File(classPathFile, name + ".class");
        //File file1 = new File("E:\\design_patterns\\design_patterns\\src\\main\\java\\com\\swh\\design\\proxy\\handwritingproxy\\Proxy$0.java");
        FileInputStream inIo = null;
        ByteArrayOutputStream out = null;
        try {
            if (file.exists()) {
                inIo = new FileInputStream(file);
                out = new ByteArrayOutputStream();
                byte[] bytes = new byte[1024];
                int len;
                while ((len = inIo.read(bytes)) != -1) {
                    out.write(bytes, 0, len);
                }
                // 把.class文件加载到虚拟机中,并返回Class对象
                return defineClass(className, out.toByteArray(), 0, out.size());
            }
        } catch (FileNotFoundException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }finally {
            if(inIo!=null){
                try {
                    inIo.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
            if(out!=null){
                try {
                    out.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }

        return null;
    }
}

package com.swh.design.proxy.handwritingproxy;

import javax.tools.JavaCompiler;
import javax.tools.JavaFileObject;
import javax.tools.StandardJavaFileManager;
import javax.tools.ToolProvider;
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.lang.reflect.Constructor;
import java.lang.reflect.InvocationHandler;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;

public class MyselfProxy {

    private static final String WRAP = "\r\n";


    public static Object newProxyInstance(MyselfClassLoader loader,
                                          Class<?>[] interfaces,
                                          MyselfInvocationHandler h) {
        //手动生成动态代理类java文件的代码字符串
        String generateProxy = generateProxy(interfaces);
        // 获取当前文件夹的路径
        String path = MyselfProxy.class.getResource("").getPath() + "Proxy$0.java";
        FileWriter fileWriter = null;
        StandardJavaFileManager standardFileManager =  null;
        try {
            // 生成java文件
            fileWriter = new FileWriter(path);
            fileWriter.write(generateProxy);
            fileWriter.flush();

            // 编译java文件
            JavaCompiler systemJavaCompiler = ToolProvider.getSystemJavaCompiler();
            standardFileManager = systemJavaCompiler.getStandardFileManager(null, null, null);
            Iterable<? extends JavaFileObject> javaFileObjects = standardFileManager.getJavaFileObjects(path);
            JavaCompiler.CompilationTask task = systemJavaCompiler.getTask(null, standardFileManager, null, null, null, javaFileObjects);
            task.call();
            //把编译好的java文件加载到JVM中
            Class<?> proxy$0 = loader.findClass("Proxy$0");
            Constructor<?> constructor = proxy$0.getConstructor(MyselfInvocationHandler.class);
            //删除虚拟代理类
            File file = new File(path);
            file.delete();
            //  通过反射创建动态代理类
            return constructor.newInstance(h);

        } catch (Exception e) {
            e.printStackTrace();
        }  finally {
            if(fileWriter!=null){
                try {
                    fileWriter.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
            if(standardFileManager!=null){
                try {
                    standardFileManager.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }


        return null;
    }

    // 生成动态代理类的java文件
    private static String generateProxy(Class<?>[] interfaces) {

        StringBuilder spliceProxy = new StringBuilder();

        spliceProxy.append("package com.swh.design.proxy.handwritingproxy;" + WRAP + WRAP);

        spliceProxy.append("import java.lang.reflect.Method;"+WRAP);
        // 实现目标类的接口是为了生成动态代理类后能够使用目标类来接收,并可以使用目标类来调用目标类的方法
        spliceProxy.append("public class Proxy$0 implements " + interfaces[0].getName() + " {" + WRAP);
        spliceProxy.append("private MyselfInvocationHandler h;" + WRAP);
        spliceProxy.append("public Proxy$0(MyselfInvocationHandler h){" + WRAP);
        spliceProxy.append("this.h=h;}" + WRAP);


        Method[] methods = interfaces[0].getMethods();

        for (Method method : methods) { // 循环处理目标类的中方法 进行代理,并实现接口中的方法进行重写
            spliceProxy.append("public " + method.getReturnType().getName() + " " + method.getName() + "(){" + WRAP);
            spliceProxy.append("try{"+WRAP);
            spliceProxy.append("Method m=" + interfaces[0].getName() + ".class.getMethod(\"" + method.getName() + "\",new Class[]{});" + WRAP);
            // 调用JdkProxy类的invoke 方法 ,invoke方法加上额外功能后调用目标类的方法
            spliceProxy.append("this.h.invoke(this,m,null);" + WRAP);
            spliceProxy.append("}catch (Throwable e) {"+WRAP);
            spliceProxy.append("e.printStackTrace();"+WRAP);
            spliceProxy.append("}"+WRAP);
            spliceProxy.append("}" + WRAP);


        }
        spliceProxy.append("}" + WRAP);


        return spliceProxy.toString();


    }

}
package com.swh.design.proxy.handwritingproxy;

public class JdkClient {

    public static void main(String[] args) {
        JdkProxy jdkProxy = new JdkProxy(new XiaoMing());
        // 获取到动态代理类
        Person o = (Person) jdkProxy.newProxy();
        // 调用动态代理方法
        o.findLove();

    }
}
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 213,417评论 6 492
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 90,921评论 3 387
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 158,850评论 0 349
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,945评论 1 285
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,069评论 6 385
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,188评论 1 291
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,239评论 3 412
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,994评论 0 268
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,409评论 1 304
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,735评论 2 327
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,898评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,578评论 4 336
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,205评论 3 317
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,916评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,156评论 1 267
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,722评论 2 363
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,781评论 2 351

推荐阅读更多精彩内容