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

news2024/10/5 13:51:16

一、简介

背景：

在之前 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 ======");
        };
    }
}