如何实现一个简易版的 Spring - 如何实现 AOP(中)

前言

在上篇 如何实现 AOP(上) 介绍了 AOP 技术出现的原因和一些重要的概念,在我们自己实现之前有必要先了解一下 AOP 底层到底是如何运作的,所以这篇再来看看 AOP 实现所依赖的一些核心基础技术。AOP 是使用动态代理字节码生成技术来实现的,在运行期(注意:不是编译期!)为目标对象生成代理对象,然后将横切逻辑织入到生成的代理对象中,最后系统使用的是带有横切逻辑的代理对象,而不是被代理对象,由代理对象转发到被代理对象。

代理模式

动态代理的根源是设计模式中的代理模式,代理模式在 GoF 中的描述如下:

Provide a surrogate or placeholder for another object to control access to it.

从其定义可以看出,代理模式主要是为了控制对象的访问,通常也会拥有被代理者的所有功能。通过代理模式我们可以在不改变被代理类的情况下,通过引入代理类来给被代理类添加一些功能,此时脑海里飘过计算机科学界中一句著名的话(P.S. 基础知识很重要啊,可以参见 这篇):

计算机科学的任何一个问题,都可以通过增加一个中间层来解决。

代理模式其实在现实生活中也经常会接触到,比如在一线城市租房时大部分都是找的租房中介去看的房子、谈价格以及签合同,是因为房子的房东已经把房子全权托管给了中介处理了,这里的租房中介其实就是充当了代理模式中的代理对象的解决,而真正的被代理对象(目标对象)其实是房子的房东,而和我们打交道都是租房中介(代理对象)。代理模式类图结构如下图所示:

proxy-pattern.png

图中各个部分的含义如下:

  • ISubject 是被代理对象所能提供能力的抽象。
  • SubjectImpl 是被代理的具体实现类。
  • SubjectProxy 是代理的实现类,通常该类持有 ISubject 接口的实例。
  • Client 是访问者的抽象角色,要访问 ISubject 类型的资源。

可以看到 SubjectImplSubjectProxy 都实现了相同的接口 ISubject,在代理对象 SubjectProxy 内持有 ISubject 的引用,当 Client 访问 doOperation() 时,代理对象将请求转发给被代理对象,单单从这个过程来看,代理对象如果只是为了转发请求,是不是有点多此一举了?再结合代理模式的定义思考一下,在转发之前(后者之后)不就可以添加一些访问控制了吗。

static-proxy.png

在代理对象(SubjectProxy)将请求转发给被代理对象(SubejctImpl)之前或者之后都是根据需要添加一些处理逻辑,而不需要修改被代理对象的具体实现逻辑,假设 SubjectImpl 是我们系统中 Joinpoint 所在的对象,此时 SubjectImpl 就是我们的目标对象了,只需要为这个目标对象创建一个代理对象,然后将横切逻辑添加到代理对象中,对外暴露出创建出来的代理对象就可以将将横切逻辑和原来的逻辑融合在一起了。

动态代理

到目前为止,一切都是那么美好,当只为同一个目标对象类型添加横切逻辑时,只需要创建一个代理对象即可,但是在 Joinpoint 相同而目标对象类型不同时,需要为每个不同的目标对象类型都单独创建一个代理对象,而这些代理对象的横切逻辑其实都是一样的,根据 DRY原则,需要寻找另一种技术来解决这个问题。
JDK 1.3 引入的 动态代理机制 可以为指定的接口在运行期动态的去生成代理对象,使用这个动态代理机制可以解决上述问题,这样我们就可以不事先为每个原始类创建代理类,而是在运行时动态生成代理类。在 Java 中,使用动态代理是比较简单的,它本身就已经使用反射实现了动态代理的语法,主要是由一个类 Proxy 和一个接口 InvocationHandler 组成,使用动态代理机制实现前文示例如下:

/**
 * @author mghio
 * @since 2021-05-29
 */
public interface Subject {
  void doOperation();
}

public class SubjectImpl implements Subject {
  @Override
  public void doOperation() {
    System.out.println("SubjectImpl doOperation...");
  }
}

public class JDKInvocationHandler implements InvocationHandler {
  private final Subject target;
  public JDKInvocationHandler(Subject target) {
    this.target = target;
  }
  @Override
  public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
    // add pre process logic if necessary
    
    System.out.println("Proxy before JDKInvocationHandler doOperation...");
    Object result = method.invoke(target, args);
    System.out.println("Proxy after JDKInvocationHandler doOperation...");

    // add post process logic if necessary
    return result;
  }
}

public class Client {
  public static void main(String[] args) {
    // 将生成的代理类保存在根目录下(com/sun/proxy/XXX.class)
    System.setProperty("sun.misc.ProxyGenerator.saveGeneratedFiles", "true");
    Subject target = new SubjectImpl();
    Subject proxy = (Subject) Proxy.newProxyInstance(SubjectImpl.class.getClassLoader(),
        new Class[]{Subject.class}, new JDKInvocationHandler(target));
    proxy.doOperation();
  }
}

由以上代码可知,使用 JDK 的动态代理只要 3 步:

  1. 定义公共接口 Subject,创建被代理对象 SubjectImpl
  2. 创建被代理对象的处理对象 JDKInvocationHandler,持有目标对象 Subject 的引用
  3. 使用 JDKProxy 类的静态方法 newProxyInstance 创建代理对象

通过设置 sun.misc.ProxyGenerator.saveGeneratedFiles 属性,可以将动态生成的代理类保存在项目根目录下,运行上面的示例代码生成的代理类如下:

public final class $Proxy0 extends Proxy implements Subject {
  private static Method m1;
  private static Method m2;
  private static Method m3;
  private static Method m0;

  static {
    try {
      m1 = Class.forName("java.lang.Object").getMethod("equals", Class.forName("java.lang.Object"));
      m2 = Class.forName("java.lang.Object").getMethod("toString");
      m3 = Class.forName("cn.mghio.designpattern.proxy.dynamicproxy.Subject").getMethod("doOperation");
      m0 = Class.forName("java.lang.Object").getMethod("hashCode");
    } catch (NoSuchMethodException var2) {
      throw new NoSuchMethodError(var2.getMessage());
    } catch (ClassNotFoundException var3) {
      throw new NoClassDefFoundError(var3.getMessage());
    }
  }

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

  public final void doOperation() throws  {
    try {
      super.h.invoke(this, m3, (Object[])null);
    } catch (RuntimeException | Error var2) {
      throw var2;
    } catch (Throwable var3) {
      throw new UndeclaredThrowableException(var3);
    }
  }

  // omit toString、equals、hashCode method...

}

从动态生成的代理类的代码可以看出,JDK 动态代理生成的代理类是继承 JDK 中提供的 Proxy 和实现被代理类所实现的接口。进一步可以从这个实现方式得出两点:1. 使用 JDK 动态代理时为什么只能使用接口引用指向代理,而不能使用被代理的具体类引用指向代理;2. 被代理类必须实现接口,因为 JDK 动态代理生成的代理类必须继承自 Proxy,而 Java 不支持多重继承,所以只能通过实现接口的方式。

jdk-dynamic-proxy.png

在默认情况下,Spring AOP 发现了目标对象实现了接口,会使用 JDK 动态代理机制为其动态生成代理对象,虽然提倡面向接口编程,但是也有目标对象没有实现接口的场景,当被代理的目标对象没有实现接口时就无法使用 JDK 动态代理了,那么这种情况下就需要使用第三方工具来帮忙了。

字节码生成技术

当目标对象没有实现接口时,可以通过动态字节码生成来继承目标对象来动态生成相应的子类,在生成的子类中重写父类目标对象的行为,然后将横切逻辑放在子类,在系统中使用目标对象的子类,最终的效果是代理模式是一样的,CGLIB 动态字节码生成类库(它本身其实也是一个抽象层,更底层是 ASM)可以动态生成和修改一个类的字节码。当以上示例代码目标对象未实现接口时修改为 CGLIB 动态生成字节码方式实现如下:

/**
 * @author mghio
 * @since 2021-05-29
 */
public class RealSubject {
  public void doOperation() {
    System.out.println("RealSubject doOperation...");
  }
}

public class CglibMethodInterceptor implements MethodInterceptor {
  @Override
  public Object intercept(Object o, Method method, Object[] args, MethodProxy methodProxy)
      throws Throwable {
    // add pre process logic if necessary

    System.out.println("Cglib before RealSubject doOperation...");
    Object result = methodProxy.invokeSuper(o, args);
    System.out.println("Cglib after RealSubject doOperation...");

    // add post process logic if necessary
    return result;
  }
}

public class Client {
  public static void main(String[] args) {
    // 将生成的动态代理类保存到文件中
    System.setProperty(DebuggingClassWriter.DEBUG_LOCATION_PROPERTY,
        "/Users/mghio/IdeaProjects/designpattern/cglib/proxy/");
    // 1. 创建 Enhancer 对象
    Enhancer enhancer = new Enhancer();
    // 2. 设置被代理类
    enhancer.setSuperclass(RealSubject.class);
    // 3. 设置回调对象(实现 MethodInterceptor 接口)
    enhancer.setCallback(new CglibMethodInterceptor());
    // 4. 创建代理对象
    RealSubject proxy = (RealSubject) enhancer.create();
    proxy.doOperation();
  }
}

使用 CGLIB 生成代理对象需要 4 步:

  1. 创建 Enhancer 对象,动态生成字节码的绝大部分逻辑都是在这个类中完成的。
  2. 设置被代理类(目标对象),生成的代理类会继承自这个接口,也就意味着这个类不能是 final 类型的。
  3. 设置回调对象(实现 MethodInterceptor 接口),在这里根据需要添加横切逻辑。
  4. 调用 Enhanercreate() 方法创建代理对象。

在设置 DebuggingClassWriter.DEBUG_LOCATION_PROPERTY 属性后,反编译已保存的动态生成代理类如下:

cglib-generate.png

从反编译后的代码可以看出 CGLIB 生成的代理类是通过继承被代理类 RealSubject 实现 Factory 接口实现的,要能被继承也就要求被代理类不能是 final 类型的。看到这里你可能会问:既然 JDK 动态代理要求被代理类实现接口,而 CGLIB 动态字节码生成要求不能是 final 类,那对于那些没有实现接口同时还是 final 类,要怎么动态代理呢?好问题,这个就留给你自己去思考了。

总结

本文简要介绍了 Spring AOP 实现所依赖的核心基础技术,从动态代理的根源代理模式到动态代理和动态字节码生成技术,为下篇动手实现简易版的 AOP 打下基础,在了解了所依赖的基础技术后,在具体实现时就会更加丝滑,动态代理动态字节码生成对比如下:

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

推荐阅读更多精彩内容