模拟jdk的代理手动生成代理类

1.代理类
public class MyProxy {

public static final String ln = "\r\n";

//MyInvocationHandler这里输入代理类自己
public static Object newProxyInstance(MyClassLoader classLoader, Class<?> [] interfaces, MyInvocationHandler h){
   try {
       //1、动态生成源代码.java文件
       String src = generateSrc(interfaces);
       System.out.println(src);
       //2、Java文件输出磁盘
       String filePath = MyProxy.class.getResource("").getPath();
       //filepath =
       System.out.println(filePath);
       File f = new File(filePath + "$Proxy0.java");
       FileWriter fw = new FileWriter(f);
       fw.write(src);
       fw.flush();
       fw.close();
       //3、把生成的.java文件编译成.class文件
       JavaCompiler compiler = ToolProvider.getSystemJavaCompiler();
       StandardJavaFileManager manage = compiler.getStandardFileManager(null,null,null);
       Iterable iterable = manage.getJavaFileObjects(f);
       JavaCompiler.CompilationTask task = compiler.getTask(null,manage,null,null,null,iterable);
       task.call();
       manage.close();
       //4、编译生成的.class文件加载到JVM中来
      Class proxyClass =  classLoader.findClass("$Proxy0");
      Constructor c = proxyClass.getConstructor(MyInvocationHandler.class);
      f.delete();
      //5、返回字节码重组以后的新的代理对象
       return c.newInstance(h);
   }catch (Exception e){
       e.printStackTrace();
   }
    return null;
}

private static String generateSrc(Class<?>[] interfaces){
        StringBuffer sb = new StringBuffer();
        //使用建造者的链式写法
        sb.append(MyProxy.class.getPackage() + ";" + ln)
          .append("import " + interfaces[0].getName() + ";" + ln)
          .append("import java.lang.reflect.*;" + ln)
          .append("public class $Proxy0 implements " + interfaces[0].getName() + " {" + ln)
          //构造器注入代理类 MyInvocationHandler
          .append("MyInvocationHandler h;" + ln)
          .append("public $Proxy0(MyInvocationHandler h) { " + ln)
          .append("this.h = h;" +ln)
          .append("}" + ln);
  /**
   * 看下这个类中的方法
   */
  for (int i = 0; i < interfaces[0].getMethods().length; i++) {
    //public abstract void com.dxy.yesas2020goals.study.partone.disignpattern.proxypattern.dynamicproxy.myproxy.client.IPerson.findLove(java.lang.Integer)
    System.out.println("---test -----" + interfaces[0].getMethods()[0]);
  }
  for (Method m : interfaces[0].getMethods()){
                Class<?>[] params = m.getParameterTypes();
                //paramNames接受参数值名字本例中为height
                StringBuffer paramNames = new StringBuffer();
                //paramValues接受参数值本例中为180
                StringBuffer paramValues = new StringBuffer();
                 //paramClasses接受参数值的类型本例中为 java.lang.Interger
                StringBuffer paramClasses = new StringBuffer();

                for (int i = 0; i < params.length; i++) {
                    Class clazz = params[i];
                    String type = clazz.getName();
                    String paramName = toLowerFirstCase(clazz.getSimpleName());
                    paramNames.append(type + " " +  paramName);
                    paramValues.append(paramName);
                    paramClasses.append(clazz.getName() + ".class");
                    if(i > 0 && i < params.length-1){
                        paramNames.append(",");
                        paramClasses.append(",");
                        paramValues.append(",");
                    }
                }
                //拼接类中的方法
                sb.append("public " + m.getReturnType().getName() + " " + m.getName() + "(" + paramNames.toString() + ") {" + ln)
                    .append("try { " + ln)
                    .append(" Method m = " + interfaces[0].getName() + ".class.getMethod(\"" + m.getName() + "\",new Class[]{" + paramClasses.toString() + "});" + ln)
                    .append((hasReturnValue(m.getReturnType()) ? "return " : "") + getCaseCode("this.h.invoke(this,m,new Object[]{" + paramValues + "})",m.getReturnType()) + ";" + ln)
                    .append("}catch(Error _ex) { }" + ln)
                    .append("catch(Throwable e){" + ln)
                    .append("throw new UndeclaredThrowableException(e);" + ln)
                    .append("}"+ln)
                    .append(getReturnEmptyCode(m.getReturnType()))
                .append("}");
            }
        sb.append("}" + ln);
        return sb.toString();
}


private static Map<Class,Class> mappings = new HashMap<Class, Class>();
static {
    mappings.put(int.class,Integer.class);
}

private static String getReturnEmptyCode(Class<?> returnClass){
    if(mappings.containsKey(returnClass)){
        return "return 0;";
    }else if(returnClass == void.class){
        return "";
    }else {
        return "return null;";
    }
}

private static String getCaseCode(String code,Class<?> returnClass){
    if(mappings.containsKey(returnClass)){
        return "((" + mappings.get(returnClass).getName() +  ")" + code + ")." + returnClass.getSimpleName() + "Value()";
    }
    return code;
}

private static boolean hasReturnValue(Class<?> clazz){
    return clazz != void.class;
}

private static String toLowerFirstCase(String src){
    char [] chars = src.toCharArray();
    chars[0] += 32;
    return String.valueOf(chars);
}

}

2. 类加载自己的器
public class MyClassLoader extends ClassLoader {

private File classPathFile;
public MyClassLoader(){
    //获取类所在路径
    String classPath = MyClassLoader.class.getResource("").getPath();
    this.classPathFile = new File(classPath);
}

@Override
protected Class<?> findClass(String name) throws ClassNotFoundException {
    //获取类的全限定名
    String className = MyClassLoader.class.getPackage().getName() + "." + name;
    if(classPathFile  != null){
        File classFile = new File(classPathFile,name.replaceAll("\\.","/") + ".class");
        if(classFile.exists()){
            FileInputStream in = null;
            ByteArrayOutputStream out = null;
            try{
                in = new FileInputStream(classFile);
                out = new ByteArrayOutputStream();
                byte [] buff = new byte[1024];
                int len;
                while ((len = in.read(buff)) != -1){
                    out.write(buff,0,len);
                }
                return defineClass(className,out.toByteArray(),0,out.size());
            }catch (Exception e){
                e.printStackTrace();
            }
        }
    }
    return null;
}

}
3.MyInvocationHandler

//代理类需要实现的接口
public interface MyInvocationHandler {
    public Object invoke(Object proxy, Method method, Object[] args)
            throws Throwable;
}