JDK動態代理過程原理及手寫實現詳解
JDK動態代理的過程
JDK動態代理采用字節重組,重新生成對象來替代原始對象,以達到動態代理的目的。
JDK中有一個規范,在ClassPath下隻要是$開頭的.class文件,一般都是自動生成的。
要實現JDK動態代理生成對象,首先得弄清楚JDK動態代理的過程。
1.獲取被代理對象的引用,並且使用反射獲取它的所有接口。
2.JDK動態代理類重新生成一個新的類,同時新的類要實現被代理類實現的所有接口。
3.動態生成Java代碼,新添加的業務邏輯方法由一定的邏輯代碼調用。
4.編譯新生成的Java代碼(.class文件)。
5.重新加載到VM中運行。
手寫實現JDK動態代理
JDK動態代理功能非常強大, 接下來就模仿JDK動態代理實現一個屬於自己的動態代理。
創建MyInvocationHandler接口
參考JDK動態代理的InvocationHandler
接口,創建屬於自己的MyInvocationHandler
接口
public interface MyInvocationHandler { Object invoke(Object proxy, Method method, Object[] args) throws Throwable; }
創建MyClassLoader類加載器
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) { 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; } }
創建代理類
創建的代理類是整個JDK動態代理的核心
public class MyProxy { // 回車、換行符 public static final String ln = "\r\n"; /** * 重新生成一個新的類,並實現被代理類實現的所有接口 * * @param classLoader 類加載器 * @param interfaces 被代理類實現的所有接口 * @param invocationHandler * @return 返回字節碼重組以後的新的代理對象 */ public static Object newProxyInstance(MyClassLoader classLoader, Class<?>[] interfaces, MyInvocationHandler invocationHandler) { try { // 動態生成源代碼.java文件 String sourceCode = generateSourceCode(interfaces); // 將源代碼寫入到磁盤中 String filePath = MyProxy.class.getResource("").getPath(); File f = new File(filePath + "$Proxy0.java"); FileWriter fw = new FileWriter(f); fw.write(sourceCode); fw.flush(); fw.close(); // 把生成的.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(); // 編譯生成的.class文件加載到JVM中來 Class proxyClass = classLoader.findClass("$Proxy0"); Constructor c = proxyClass.getConstructor(MyInvocationHandler.class); //刪除生成的.java文件 f.delete(); // 返回字節碼重組以後的新的代理對象 return c.newInstance(invocationHandler); } catch (Exception e) { e.printStackTrace(); } return null; } /** * 動態生成源代碼.java文件 * * @param interfaces 被代理類實現的所有接口 * @return .java文件的源代碼 */ private static String generateSourceCode(Class<?>[] interfaces) { StringBuffer sb = new StringBuffer(); sb.append(MyProxy.class.getPackage() + ";" + ln); sb.append("import " + interfaces[0].getName() + ";" + ln); sb.append("import java.lang.reflect.*;" + ln); sb.append("public class $Proxy0 implements " + interfaces[0].getName() + "{" + ln); sb.append("MyInvocationHandler invocationHandler;" + ln); sb.append("public $Proxy0(MyInvocationHandler invocationHandler) { " + ln); sb.append("this.invocationHandler = invocationHandler;"); sb.append("}" + ln); for (Method m : interfaces[0].getMethods()) { Class<?>[] params = m.getParameterTypes(); StringBuffer paramNames = new StringBuffer(); StringBuffer paramValues = new StringBuffer(); 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 + ") {" + ln); sb.append("try{" + ln); sb.append("Method m = " + interfaces[0].getName() + ".class.getMethod(\"" + m.getName() + "\",new Class[]{" + paramClasses + "});" + ln); sb.append((hasReturnValue(m.getReturnType()) ? "return " : "") + getCaseCode("this.invocationHandler.invoke(this,m,new Object[]{" + paramClasses + "})", m.getReturnType()) + ";" + ln); sb.append("}catch(Error ex) { }"); sb.append("catch(Throwable e){" + ln); sb.append("throw new UndeclaredThrowableException(e);" + ln); sb.append("}"); sb.append(getReturnEmptyCode(m.getReturnType())); sb.append("}"); } sb.append("}" + ln); return sb.toString(); } /** * 定義返回類型 */ private static Map<Class, Class> mappings = new HashMap<Class, Class>(); /** * 初始化一些返回類型 */ static { mappings.put(int.class, Integer.class); mappings.put(Integer.class, Integer.class); mappings.put(double.class, Double.class); mappings.put(Double.class, Double.class); } private static String getReturnEmptyCode(Class<?> returnClass) { if (mappings.containsKey(returnClass)) { if (returnClass.equals(int.class) || returnClass.equals(Integer.class)) { return "return 0;"; } else if (returnClass.equals(double.class) || returnClass.equals(Double.class)) { return "return 0.0;"; } else { return "return 0;"; } } else if (returnClass == void.class) { return ""; } else { return "return null;"; } } /** * 判斷返回值類型 * * @param code * @param returnClass * @return */ private static String getCaseCode(String code, Class<?> returnClass) { if (mappings.containsKey(returnClass)) { // ((java.lang.Double) this.invocationHandler.invoke(this, m, new Object[]{})).doubleValue(); String re = "((" + mappings.get(returnClass).getName() + ")" + code + ")." + returnClass.getSimpleName().toLowerCase() + "Value()"; return re; } return code; } /** * 判斷代理接口的方法的返回值是否為void * * @param clazz 方法的返回值類型 * @return */ private static boolean hasReturnValue(Class<?> clazz) { return clazz != void.class; } /** * 參數首字母小寫 * * @param src * @return */ private static String toLowerFirstCase(String src) { char[] chars = src.toCharArray(); if (chars[0] >= 'A' && chars[0] <= 'Z') { chars[0] += 32; } return String.valueOf(chars); } /** * 首字母大寫 * * @param src * @return */ private static String toUpperFirstCase(String src) { char[] chars = src.toCharArray(); if (chars[0] >= 'a' && chars[0] <= 'z') { chars[0] -= 32; } return String.valueOf(chars); } }
使用自定義動態代理類
創建接口
public interface IUser { void shopping(); Double expenses(); }
創建被代理接口
public class User implements IUser { @Override public void shopping() { System.out.println("user shopping...."); } @Override public Double expenses() { return 50.5; } }
創建代理接口
public class UseProxy implements MyInvocationHandler { private Object target; public Object myJDKProxy(Object target){ this.target = target; Class<?> clazz = target.getClass(); return MyProxy.newProxyInstance(new MyClassLoader(),clazz.getInterfaces(),this); } public Object invoke(Object proxy, Method method, Object[] args) throws Throwable { System.out.println("代理user,執行shopping()開始..."); Object result = method.invoke(this.target, args); System.out.println("代理user,執行shopping()結束..."); return result; } }
客戶端調用
public static void main(String[] args) { UseProxy useProxy = new UseProxy(); IUser user = (IUser) useProxy.myJDKProxy(new User()); user.shopping(); System.out.println(user.expenses()); }
執行結果
代理user,執行shopping()開始…
user shopping….
代理user,執行shopping()結束…
——————————–
代理user,執行shopping()開始…
代理user,執行shopping()結束…
——————————–
50.5
生成源代碼
查看生產的Java文件源代碼
package cn.ybzy.demo.proxy.proxy; import cn.ybzy.demo.proxy.client.IUser; import java.lang.reflect.*; public class $Proxy0 implements cn.ybzy.demo.proxy.client.IUser{ MyInvocationHandler invocationHandler; public $Proxy0(MyInvocationHandler invocationHandler) { this.invocationHandler = invocationHandler;} public java.lang.Double expenses() { try{ Method m = cn.ybzy.demo.proxy.client.IUser.class.getMethod("expenses",new Class[]{}); return ((java.lang.Double)this.invocationHandler.invoke(this,m,new Object[]{})).doubleValue(); }catch(Error ex) { }catch(Throwable e){ throw new UndeclaredThrowableException(e); }return 0.0;}public void shopping() { try{ Method m = cn.ybzy.demo.proxy.client.IUser.class.getMethod("shopping",new Class[]{}); this.invocationHandler.invoke(this,m,new Object[]{}); }catch(Error ex) { }catch(Throwable e){ throw new UndeclaredThrowableException(e); }}}
以上就是JDK動態代理過程原理及手寫實現詳解的詳細內容,更多關於JDK動態代理過程的資料請關註WalkonNet其它相關文章!
推薦閱讀:
- Java開發反射機制的實戰經驗總結
- 在Java Spring框架中使用的設計模式有哪些
- Map與JavaBean相互轉換的工具類
- java代理模式(jdk proxy)
- 詳解Java中的反射機制和動態代理