注册

Java | JDK 动态代理的原理其实很简单

前言



  • 代理模式(Proxy Pattern)也称委托模式(Delegate Pattern),是一种结构型设计模式,也是一项基础设计技巧;
  • 其中,动态代理有很多有意思的应用场景,比如 AOP、日志框架、全局性异常处理、事务处理等。这篇文章,我们主要讨论最基本的 JDK 动态代理。



目录





前置知识


这篇文章的内容会涉及以下前置 / 相关知识,贴心的我都帮你准备好了,请享用~





1. 概述



  • 什么是代理 (模式)? 代理模式 (Proxy Pattern) 也称委托模式 (Deletage Pattern),属于结构型设计模式,也是一项基本的设计技巧。通常,代理模式用于处理两种问题:

    • 1、控制对基础对象的访问
    • 2、在访问基础对象时增加额外功能


这是两种非常朴素的场景,正因如此,我们常常会觉得其它设计模式中存在代理模式的影子。UML 类图和时序图如下:





  • 代理的基本分类: 静态代理 + 动态代理,分类的标准是 “代理关系是否在编译期确定;


  • 动态代理的实现方式: JDK、CGLIB、Javassist、ASM





2. 静态代理


2.1 静态代理的定义


静态代理是指代理关系在编译期确定的代理模式。使用静态代理时,通常的做法是为每个业务类抽象一个接口,对应地创建一个代理类。举个例子,需要给网络请求增加日志打印:


1、定义基础接口
public interface HttpApi {
String get(String url);
}

2、网络请求的真正实现
public class RealModule implements HttpApi {
@Override
public String get(String url) {
return "result";
}
}

3、代理类
public class Proxy implements HttpApi {
private HttpApi target;

Proxy(HttpApi target) {
this.target = target;
}

@Override
public String get(String url) {
// 扩展的功能
Log.i("http-statistic", url);
// 访问基础对象
return target.get(url);
}
}

2.2 静态代理的缺点



  • 1、重复性: 需要代理的业务或方法越多,重复的模板代码越多;
  • 2、脆弱性: 一旦改动基础接口,代理类也需要同步修改(因为代理类也实现了基础接口)。



3. 动态代理


3.1 动态代理的定义


动态代理是指代理关系在运行时确定的代理模式。需要注意,JDK 动态代理并不等价于动态代理,前者只是动态代理的实现之一,其它实现方案还有:CGLIB 动态代理、Javassist 动态代理和 ASM 动态代理等。因为代理类在编译前不存在,代理关系到运行时才能确定,因此称为动态代理。


3.2 JDK 动态代理示例


我们今天主要讨论JDK 动态代理(Dymanic Proxy API),它是 JDK1.3 中引入的特性,核心 API 是 Proxy 类和 InvocationHandler 接口。它的原理是利用反射机制在运行时生成代理类的字节码。


我们继续用打印日志的例子,使用动态代理时:


public class ProxyFactory {
public static HttpApi getProxy(HttpApi target) {
return (HttpApi) Proxy.newProxyInstance(
target.getClass().getClassLoader(),
target.getClass().getInterfaces(),
new LogHandler(target));
}

private static class LogHandler implements InvocationHandler {
private HttpApi target;

LogHandler(HttpApi target) {
this.target = target;
}
// method底层的方法无参数时,args为空或者长度为0
@Override
public Object invoke(Object proxy, Method method, @Nullable Object[] args)
throws Throwable
{
// 扩展的功能
Log.i("http-statistic", (String) args[0]);
// 访问基础对象
return method.invoke(target, args);
}
}
}

如果需要兼容多个业务接口,可以使用泛型:


public class ProxyFactory {
@SuppressWarnings("unchecked")
public static T getProxy(T target) {
return (T) Proxy.newProxyInstance(
target.getClass().getClassLoader(),
target.getClass().getInterfaces(),
new LogHandler(target));
}

private static class LogHandler implements InvocationHandler {
// 同上
}
}

客户端调用:


HttpAPi proxy = ProxyFactory.getProxy(target);
OtherHttpApi proxy = ProxyFactory.getProxy(otherTarget);

通过泛型参数传递不同的类型,客户端可以按需实例化不同类型的代理对象。基础接口的所有方法都统一到 InvocationHandler#invoke() 处理。静态代理的两个缺点都得到解决:



  • 1、重复性:即使有多个基础业务需要代理,也不需要编写过多重复的模板代码;
  • 2、脆弱性:当基础接口变更时,同步改动代理并不是必须的。

3.3 静态代理 & 动态代理对比



  • 共同点:两种代理模式实现都在不改动基础对象的前提下,对基础对象进行访问控制和扩展,符合开闭原则。
  • 不同点:静态代理存在重复性和脆弱性的缺点;而动态代理(搭配泛型参数)可以实现了一个代理同时处理 N 种基础接口,一定程度上规避了静态代理的缺点。从原理上讲,静态代理的代理类 Class 文件在编译期生成,而动态代理的代理类 Class 文件在运行时生成,代理类在 coding 阶段并不存在,代理关系直到运行时才确定。



4. JDK 动态代理源码分析


这一节,我们来分析 JDK 动态代理的源码,核心类是 Proxy,主要分析 Proxy 如何生成代理类,以及如何将方法调用统一分发到 InvocationHandler 接口。


4.1 API 概述


Proxy 类主要包括以下 API:


Proxy描述
getProxyClass(ClassLoader, Class...) : Class获取实现目标接口的代理类 Class 对象
newProxyInstance(ClassLoader,Class[],InvocationHandler) : Object获取实现目标接口的代理对象
isProxyClass(Class) : boolean判断一个 Class 对象是否属于代理类
getInvocationHandler(Object) : InvocationHandler获取代理对象内部的 InvocationHandler

4.2 核心源码


Proxy.java


1、获取代理类 Class 对象
public static Class getProxyClass(ClassLoader loader,Class... interfaces){
final Class[] intfs = interfaces.clone();
...
1.1 获得代理类 Class 对象
return getProxyClass0(loader, intfs);
}

2、实例化代理类对象
public static Object newProxyInstance(ClassLoader loader,Class[] interfaces,InvocationHandler h){
...
final Class[] intfs = interfaces.clone();
2.1 获得代理类 Class对象
Class cl = getProxyClass0(loader, intfs);
...
2.2 获得代理类构造器 (接收一个 InvocationHandler 参数)
// private static final Class[] constructorParams = { InvocationHandler.class };
final Constructor cons = cl.getConstructor(constructorParams);
final InvocationHandler ih = h;
...
2.3 反射创建实例
return newInstance(cons, ih);
}

可以看到,实例化代理对象也需要先通过 getProxyClass0(...) 获取代理类 Class 对象,而 newProxyInstance(...) 随后会获取参数为 InvocationHandler 的构造函数实例化一个代理类对象。


我们先看下代理类 Class 对象是如何获取的:


Proxy.java


-> 1.12.1 获得代理类 Class对象
private static Class getProxyClass0(ClassLoader loader,Class... interfaces) {
...
从缓存中获取代理类,如果缓存未命中,则通过ProxyClassFactory生成代理类
return proxyClassCache.get(loader, interfaces);
}

private static final class ProxyClassFactory implements BiFunction[], Class>{

3.1 代理类命名前缀
private static final String proxyClassNamePrefix = "$Proxy";

3.2 代理类命名后缀,从 0 递增(原子 Long)
private static final AtomicLong nextUniqueNumber = new AtomicLong();

@Override
public Class apply(ClassLoader loader, Class[] interfaces)
{
Map, Boolean> interfaceSet = new IdentityHashMap<>(interfaces.length);
3.3 参数校验
for (Class intf : interfaces) {
// 验证参数 interfaces 和 ClassLoder 中加载的是同一个类
// 验证参数 interfaces 是接口类型
// 验证参数 interfaces 中没有重复项
// 否则抛出 IllegalArgumentException
}
// 验证所有non-public接口来自同一个包

3.4(一般地)代理类包名
// public static final String PROXY_PACKAGE = "com.sun.proxy";
String proxyPkg = ReflectUtil.PROXY_PACKAGE + ".";

3.5 代理类的全限定名
long num = nextUniqueNumber.getAndIncrement();
String proxyName = proxyPkg + proxyClassNamePrefix + num;

3.6 生成字节码数据
byte[] proxyClassFile = ProxyGenerator.generateProxyClass(proxyName, interfaces);

3.7 从字节码生成 Class 对象
return defineClass0(loader, proxyName,proxyClassFile, 0, proxyClassFile.length);
}
}

-> 3.6 生成字节码数据
public static byte[] generateProxyClass(final String var0, Class[] var1) {
ProxyGenerator var2 = new ProxyGenerator(var0, var1);
...
final byte[] var3 = var2.generateClassFile();
return var3;
}
<?>,>

ProxyGenerator.java


private byte[] generateClassFile() {
3.6.1 只代理Object的hashCode、equals和toString
this.addProxyMethod(hashCodeMethod, Object.class);
this.addProxyMethod(equalsMethod, Object.class);
this.addProxyMethod(toStringMethod, Object.class);

3.6.2 代理接口的每个方法
...
for(var1 = 0; var1 < this.interfaces.length; ++var1) {
...
}

3.6.3 添加带有 InvocationHandler 参数的构造器
this.methods.add(this.generateConstructor());
var7 = this.proxyMethods.values().iterator();
while(var7.hasNext()) {
...
3.6.4 在每个代理的方法中调用InvocationHandler#invoke()
}

3.6.5 输出字节流
ByteArrayOutputStream var9 = new ByteArrayOutputStream();
DataOutputStream var10 = new DataOutputStream(var9);
...
return var9.toByteArray();
}

以上代码已经非常简化了,主要关注核心流程:JDK 动态代理生成的代理类命名为 com.sun.proxy$Proxy[从0开始的数字](例如:com.sun.proxy$Proxy0),这个类继承自 java.lang.reflect.Proxy。其内部还有一个参数为 InvocationHandler 的构造器,对于代理接口的方法调用都会分发到 InvocationHandler#invoke()。


UML 类图如下,需要注意图中红色箭头,表示代理类和 HttpApi 接口的代理关系在运行时才确定:




提示: Android 系统中生成字节码和从字节码生成 Class 对象的步骤都是 native 方法:



  • private static native Class generateProxy(…)
  • 对应的native方法:dalvik/vm/native/java_lang_reflect_Proxy.cpp


4.3 查看代理类源码


可以看到,ProxyGenerator#generateProxyClass() 其实是一个静态 public 方法,所以我们直接调用,并将代理类 Class 的字节流写入磁盘文件,使用 IntelliJ IDEA 的反编译功能查看源代码。


输出字节码:


byte[] classFile = ProxyGenerator.generateProxyClass("$proxy0",new Class[]{HttpApi.class});
// 直接写入项目路径下,方便使用IntelliJ IDEA的反编译功能
String path = "/Users/pengxurui/IdeaProjects/untitled/src/proxy/HttpApi.class";
try(FileOutputStream fos = new FileOutputStream(path)){
fos.write(classFile);
fos.flush();
System.out.println("success");
} catch (Exception e){
e.printStackTrace();
System.out.println("fail");
}

反编译结果:


public final class $proxy0 extends Proxy implements HttpApi {
//反射的元数据Method存储起来,避免重复创建
private static Method m1;
private static Method m2;
private static Method m3;
private static Method m0;

public $proxy0(InvocationHandler var1) throws {
super(var1);
}

/**
* Object#hashCode()
* Object#equals(Object)
* Object#toString()
*/


// 实现了HttpApi接口
public final String get() throws {
try {
//转发到Invocation#invoke()
return (String)super.h.invoke(this, m3, (Object[])null);
} catch (RuntimeException | Error var2) {
throw var2;
} catch (Throwable var3) {
throw new UndeclaredThrowableException(var3);
}
}

static {
try {
//Object#hashCode()
//Object#equals(Object)
//Object#toString()
m3 = Class.forName("HttpApi").getMethod("get");
} catch (NoSuchMethodException var2) {
throw new NoSuchMethodError(var2.getMessage());
} catch (ClassNotFoundException var3) {
throw new NoClassDefFoundError(var3.getMessage());
}
}
}

4.4 常见误区



  • 基础对象必须实现基础接口,否则不能使用动态代理

这个想法可能来自于一些没有实现任何接口的类,因此就没有办法得到接口的Class对象作为Proxy#newProxyInstance() 的参数,这确实会带来一些麻烦,举个例子:


package com.domain;
public interface HttpApi {
String get();
}

// 另一个包的non-public接口
package com.domain.inner;
/**non-public**/interface OtherHttpApi{
String get();
}

package com.domain.inner;
// OtherHttpApiImpl类没有实现HttpApi接口或者没有实现任何接口
public class OtherHttpApiImpl /**extends OtherHttpApi**/{
public String get() {
return "result";
}
}

// Client:
HttpApi api = (HttpApi) Proxy.newProxyInstance(...}, new InvocationHandler() {
OtherHttpApiImpl impl = new OtherHttpApiImpl();

@Override
public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
// TODO:扩展的新功能
// IllegalArgumentException: object is not an instance of declaring class
return method.invoke(impl,args);
}
});
api.get();

在这个例子里,OtherHttpApiImpl 类因为历史原因没有实现 HttpApi 接口,虽然方法签名与 HttpApi 接口的方法签名完全相同,但是遗憾,无法完成代理。也有补救的办法,找到 HttpApi 接口中签名相同的 Method,使用这个 Method 来转发调用。例如:


HttpApi api = (HttpApi) Proxy.newProxyInstance(...}, new InvocationHandler() {
OtherHttpApiImpl impl = new OtherHttpApiImpl();

@Override
public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
// TODO:扩展的新功能
if (method.getDeclaringClass() != impl.getClass()) {
// 找到相同签名的方法
Method realMethod = impl.getClass().getDeclaredMethod(method.getName(), method.getParameterTypes());
return realMethod.invoke(impl, args);
}else{
return method.invoke(impl,args);
}
}
});



5. 总结


今天,我们讨论了静态代理和动态代理两种代理模式,静态代理在设计模式中随处可见,但存在重复性和脆弱性的缺点,动态代理的代理关系在运行时确定,可以实现一个代理处理 N 种基础接口,一定程度上规避了静态代理的缺点。在我们熟悉的一个网络请求框架中,就充分利用了动态代理的特性,你知道是在说哪个框架吗?




参考资料







作者:彭丑丑
链接:https://juejin.cn/post/6974018412158664734
来源:掘金
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。

0 个评论

要回复文章请先登录注册