Spring(17) AopContext.currentProxy() 类内方法调用切入

一、简介

背景：

在之前 Spring 的 AOP 用法中，只有代理的类才会被切入。例如：我们在 Controller 层调用 Service 的方式时，是可以被切入的。但是如果我们在 Service 层 A 方法中，调用 B 方法，切点切的是 B 方法，那么这时候是不会被切入的。

解决方案如标题所示，可以使用 AopContext.currentProxy() 来解决。在 A 方法中使用如下方式来调用 B 方法，这样一来，就能切入了：

import org.springframework.aop.framework.AopContext;((Service) AopContext.currentProxy()).B();

二、代码示例

2.1 接口类

首先定义一个接口，其中前两个方法内部通过不同方式调用了第三个方法。

SomeInterface.java

public interface SomeInterface {/** 调用方，普通调用 */void someMethod();/** 调用方，获取代理调用 */void someMethodWithProxy();/** 被调用方 */void anotherMethod();
}

2.2 接口实现类

然后我们按照上面描述的关系来编写实现类。

MyBean.java

import com.demo.module.test.SomeInterface;
import org.springframework.aop.framework.AopContext;
import org.springframework.stereotype.Component;@Component
public class MyBean implements SomeInterface {@Overridepublic void someMethod() {System.out.println("someMethod...");this.anotherMethod();}@Overridepublic void someMethodWithProxy() {System.out.println("someMethodWithProxy...");MyBean myBean = (MyBean) AopContext.currentProxy();myBean.anotherMethod();}@Overridepublic void anotherMethod() {System.out.println("anotherMethod...");}
}

2.3 AOP切面类

这里实现了一个切面来增强被调用的方法 anotherMethod()。

SomeAspect.java

import org.aspectj.lang.annotation.After;
import org.aspectj.lang.annotation.Aspect;
import org.springframework.stereotype.Component;@Aspect
@Component
public class SomeAspect {@After("execution(* com.demo.impl.MyBean.anotherMethod(..))")public void afterAnotherMethod() {System.out.println("after anotherMethod");}
}

2.4 启动类（测试）

注意，这里需要在启动类或者配置类中添加如下注解，不然无法使用 AopContext.currentProxy() 方法来获取代理类。

@EnableAspectJAutoProxy(exposeProxy = true) // 开启 Spring 注解 AOP 配置的支持

不添加会出现如下报错：java.lang.IllegalStateException: Cannot find current proxy: Set ‘exposeProxy’ property on Advised to ‘true’ to make it available…

Caused by: java.lang.IllegalStateException: Cannot find current proxy: Set 'exposeProxy' property on Advised to 'true' to make it available, and ensure that AopContext.currentProxy() is invoked in the same thread as the AOP invocation context.at org.springframework.aop.framework.AopContext.currentProxy(AopContext.java:69) ~[spring-aop-5.2.6.RELEASE.jar:5.2.6.RELEASE]at com.demo.module.test.impl.MyBean.someMethodWithProxy(MyBean.java:26) ~[classes/:na]

此外，我们在启动类中注入了一个 CommandLineRunner 类，用于启动后立马执行测试代码。

DemoApplication.java

import com.demo.module.test.SomeInterface;
import org.springframework.boot.CommandLineRunner;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.EnableAspectJAutoProxy;import javax.annotation.Resource;@EnableAspectJAutoProxy(exposeProxy = true) // 开启 Spring 注解 AOP 配置的支持
@SpringBootApplication
public class DemoApplication {@Resourceprivate SomeInterface someInterface;public static void main(String[] args) {SpringApplication.run(DemoApplication.class, args);}@Beanpublic CommandLineRunner commandLineRunner() {return args -> {System.out.println("===== someMethod ======");someInterface.someMethod();System.out.println("===== someMethodWithProxy ======");someInterface.someMethodWithProxy();System.out.println("===== end ======");};}
}