17、AopContext.currentProxy() 类内方法调用切入
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 {
@Override
public void someMethod() {
System.out.println("someMethod...");
this.anotherMethod();
}
@Override
public void someMethodWithProxy() {
System.out.println("someMethodWithProxy...");
MyBean myBean = (MyBean) AopContext.currentProxy();
myBean.anotherMethod();
}
@Override
public 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 {
@Resource
private SomeInterface someInterface;
public static void main(String[] args) {
SpringApplication.run(DemoApplication.class, args);
}
@Bean
public CommandLineRunner commandLineRunner() {
return args -> {
System.out.println("===== someMethod ======");
someInterface.someMethod();
System.out.println("===== someMethodWithProxy ======");
someInterface.someMethodWithProxy();
System.out.println("===== end ======");
};
}
}
2.5 执行结果
执行结果如下,可以看到通过 AopContext.currentProxy()
获取当前类的代理后再调用,可以正常进行 AOP 增强。
整理完毕,完结撒花~ 🌻
参考地址:
1、 AopContext.currentProxy(),https://blog.csdn.net/qq_29860591/article/details/108728150;