You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
source-code-hunter/docs/Spring/AOP/JDK动态代理的实现原理解析.md

6.3 KiB

最近在看spring AOP部分的源码所以对JDK动态代理具体是如何实现的这件事产生了很高的兴趣而且能从源码上了解这个原理的话也有助于对spring-aop模块的理解。话不多说上代码。

/**
 * 一般会使用实现了 InvocationHandler 的类作为代理对象的生产工厂,
 * 并且通过持有被代理对象target来在invoke()方法中对被代理对象的目标方法进行调用和增强,
 * 这些我们都能通过下面这段代码看懂但代理对象是如何生成的invoke()方法又是如何被调用的呢?
 */
public class ProxyFactory implements InvocationHandler{
	
	private Object target = null;
	
	public Object getInstanse(Object target){
		
		this.target = target;
		return Proxy.newProxyInstance(target.getClass().getClassLoader(), 
				target.getClass().getInterfaces(), this);
	}

	@Override
	public Object invoke(Object proxy, Method method, Object[] args)
			throws Throwable {
		
		Object ret = null;
		System.out.println("前置增强");
		ret = method.invoke(target, args);
		System.out.println("后置增强");
		return ret;
	}
}

/**
 * 实现了接口MyInterface和接口的play()方法,可以作为被代理类
 */
public class TargetObject implements MyInterface {

	@Override
	public void play() {
		System.out.println("妲己,陪你玩 ~");
		
	}
}

/**
 * 测试类
 */
public class ProxyTest {

	public static void main(String[] args) {
		TargetObject target = new TargetObject();
		// ProxyFactory 实现了 InvocationHandler接口其中的 getInstanse() 方法利用 Proxy 类
		// 生成了target目标对象的代理对象并返回且ProxyFactory持有对target的引用可以在
		// invoke() 中完成对 target 相应方法的调用,以及目标方法前置后置的增强处理
		ProxyFactory proxyFactory = new ProxyFactory();
		// 这个mi就是JDK的 Proxy 类动态生成的代理类 $Proxy0 的实例,该实例中的方法都持有对
		// invoke() 方法的回调,所以当调用其方法时,就能够执行 invoke() 中的增强处理
		MyInterface mi = (MyInterface)proxyFactory.getInstanse(target);
		// 这样可以看到 mi 的 Class 到底是什么
		System.out.println(mi.getClass());
		// 这里实际上调用的就是 $Proxy0代理类中对 play() 方法的实现,结合下面的代码可以看到
		// play() 方法通过 super.h.invoke() 完成了对 InvocationHandler对象(proxyFactory)中
		// invoke()方法的回调,所以我们才能够通过 invoke() 方法实现对 target 对象方法的
		// 前置后置增强处理
		mi.play();
		// 总的来说就是在invoke()方法中完成target目标方法的调用及前置后置增强
		// JDK动态生成的代理类中对 invoke() 方法进行了回调
	}
	
	/**
	 * 将ProxyGenerator生成的动态代理类的输出到文件中利用反编译工具luyten等就可
	 * 以看到生成的代理类的源码咯,下面给出了其反编译好的代码实现
	 */
	@Test
	public void generatorSrc(){
		byte[] bytesFile = ProxyGenerator.generateProxyClass("$Proxy0", TargetObject.class.getInterfaces());
		FileOutputStream fos = null;
		try{
			String path = System.getProperty("user.dir") + "\\$Proxy0.class";
			File file = new File(path);
			fos = new FileOutputStream(file);
			fos.write(bytesFile);
			fos.flush();
		} catch (Exception e){
			e.printStackTrace();
		} finally{
			try {
				fos.close();
			} catch (IOException e) {
				// TODO Auto-generated catch block
				e.printStackTrace();
			}
		}
	}
}

/**
 * Proxy生成的代理类可以看到其继承了Proxy并且实现了被代理类的接口
 */
public final class $Proxy0 extends Proxy implements MyInterface
{
    private static Method m1;
    private static Method m0;
    private static Method m3;
    private static Method m2;
    
    static {
        try {
            $Proxy0.m1 = Class.forName("java.lang.Object").getMethod("equals", Class.forName("java.lang.Object"));
            $Proxy0.m0 = Class.forName("java.lang.Object").getMethod("hashCode", (Class<?>[])new Class[0]);
            //实例化MyInterface的play方法
            $Proxy0.m3 = Class.forName("com.shuitu.test.MyInterface").getMethod("play", (Class<?>[])new Class[0]);
            $Proxy0.m2 = Class.forName("java.lang.Object").getMethod("toString", (Class<?>[])new Class[0]);
        }
        catch (NoSuchMethodException ex) {
            throw new NoSuchMethodError(ex.getMessage());
        }
        catch (ClassNotFoundException ex2) {
            throw new NoClassDefFoundError(ex2.getMessage());
        }
    }
    
    public $Proxy0(final InvocationHandler invocationHandler) {
        super(invocationHandler);
    }
    
    public final void play() {
        try {
        	// 这个 h 其实就是我们调用 Proxy.newProxyInstance() 方法时传进去的ProxyFactory(InvocationHandler对象)
        	// 该对象的 invoke() 方法中实现了对目标对象的目标方法的增强。看到这里,利用动态代理实现方法增强的
        	// 实现原理就全部理清咯
            super.h.invoke(this, $Proxy0.m3, null);
        }
        catch (Error | RuntimeException error) {
            throw new RuntimeException();
        }
        catch (Throwable t) {
            throw new UndeclaredThrowableException(t);
        }
    }
    
    public final boolean equals(final Object o) {
        try {
            return (boolean)super.h.invoke(this, $Proxy0.m1, new Object[] { o });
        }
        catch (Error | RuntimeException error) {
            throw new RuntimeException();
        }
        catch (Throwable t) {
            throw new UndeclaredThrowableException(t);
        }
    }
    
    public final int hashCode() {
        try {
            return (int)super.h.invoke(this, $Proxy0.m0, null);
        }
        catch (Error | RuntimeException error) {
            throw new RuntimeException();
        }
        catch (Throwable t) {
            throw new UndeclaredThrowableException(t);
        }
    }
    
    public final String toString() {
        try {
            return (String)super.h.invoke(this, $Proxy0.m2, null);
        }
        catch (Error | RuntimeException error) {
            throw new RuntimeException();
        }
        catch (Throwable t) {
            throw new UndeclaredThrowableException(t);
        }
    }
}