AOP的自调用问题

问题描述

在spring中，自定义了一个AOP，用来切换多个数据库的数据源，当执行service中的method1中，调用当前service中的method2，（注：两个service中都添加了自定义的AOP注解）但是第二个方法的AOP注解没有生效。代码如下：

controller方法

@RequestMapping("/test")
@RestController
@Api(value = "测试控制器", tags = "测试控制器")
public class AopTestController {

    @Autowired
    private AliCollectService aliCollectService;

    @GetMapping("/01")
    public MessageBody method(Integer id) {
        aliCollectService.mothod1(id);
        return new MessageBody(EnumSystemRetCode.SUCCESS);
    }

    @GetMapping("/02")
    public MessageBody method2(String id) {
        aliCollectService.mothod2(id);
        return new MessageBody(EnumSystemRetCode.SUCCESS);
    }
}

service方法

    @CurrentDataSource(EnumDataSource.DS3)
    public void mothod1(Integer id) {
        String dataSource = DataSourceContextHolder.getDataSource();
        System.out.println("切换之前的数据源:  "+dataSource);
        String name = Thread.currentThread().getName();
        System.out.println("主线程:  "+name);
        mothod2("");
//        new Thread(() -> {
//        }).start();;
    }

    @CurrentDataSource(EnumDataSource.DS2)
    public void mothod2(String upItemId) {
        String name = Thread.currentThread().getName();
        System.out.println("新建线程:  "+name);
        String dataSource = DataSourceContextHolder.getDataSource();
        System.out.println("切换之后的数据源:  "+dataSource);
        System.out.println(upItemId);
    }

控制台打印如下：

切换之前的数据源:  ds3
主线程:  http-nio-9201-exec-1
新建线程:  http-nio-9201-exec-1
切换之后的数据源:  ds3

解决思路

刚开始以为是当前线程无法进行aop的再次切入，新建了一个线程去调用方法2，代码没什么变化，就是新建了一个Thread，将方法2放到线程中执行

 @CurrentDataSource(EnumDataSource.DS3)
    public void mothod1(Integer id) {
        String dataSource = DataSourceContextHolder.getDataSource();
        System.out.println("切换之前的数据源:  "+dataSource);
        String name = Thread.currentThread().getName();
        System.out.println("主线程:  "+name);
        new Thread(() -> {
            mothod2("");
        }).start();
    }

    @CurrentDataSource(EnumDataSource.DS2)
    public void mothod2(String upItemId) {
        String name = Thread.currentThread().getName();
        System.out.println("新建线程:  "+name);
        String dataSource = DataSourceContextHolder.getDataSource();
        System.out.println("切换之后的数据源:  "+dataSource);
        System.out.println(upItemId);
    }

执行结果：

切换之前的数据源:  ds3
主线程:  http-nio-9201-exec-1
新建线程:  Thread-38
切换之后的数据源:  ds3

显然，虽然线程切换了，但是结果并没有什么变化，数据源没有成功切换，所以aop的方法就没有执行到；
然后我将method2单独在controller中进行调用，结果发现aop正常执行了，恍然大悟，spring中的aop就是通过代理对象调用，只有这种调用方式，才能够在真正的对象的执行前后，能够让代理对象也执行相关代码，才能起到切面的作用。而对于使用this的方式调用，这种只是自调用，并不会使用代理对象进行调用，也就无法执行切面类。

解决方案

像我之前第一种情况，只是this的自调用，所以aop不会被执行，既然这样，那么把调用方的类搞成代理类不就可以了，最简单的方法，将method2放到一个单独的service中，然后交给spring去管理，也就生成了一个代理类。

新建service

@Service
public class Method02Service {
    @CurrentDataSource(EnumDataSource.DS2)
    public void mothod2(String upItemId) {
        String dataSource = DataSourceContextHolder.getDataSource();
        System.out.println("切换之后的数据源:  "+dataSource);
        System.out.println(upItemId);
    }
}

然后调用：

  @Autowired
    private Method02Service method02Service;
    @CurrentDataSource(EnumDataSource.DS3)
    public void mothod1(Integer id) {
        String dataSource = DataSourceContextHolder.getDataSource();
        System.out.println("切换之前的数据源:  "+dataSource);
        method02Service.mothod2("");
    }

执行结果：

切换之前的数据源:  ds3
切换之后的数据源:  ds2

问题得到解决，当然还有其他方式，例如获取到当前的代理类，通过代理类去调用下面的方法。

修改method1中的调用方式如下：

 @CurrentDataSource(EnumDataSource.DS3)
    public void mothod1(Integer id) {
        String dataSource = DataSourceContextHolder.getDataSource();
        System.out.println("切换之前的数据源:  "+dataSource);
        // 这里不使用this.loadFactor而是使用AopContext.currentProxy()调用，目的是解决AOP代理不支持方法自调用的问题
        if (AopContext.currentProxy() instanceof AliCollectService) {
            ((AliCollectService) AopContext.currentProxy()).mothod2("");
            return;
        } else {
            // 部分实现没有被代理过，则直接进行自调用即可
            mothod2("");
            return ;
        }
    }

    @CurrentDataSource(EnumDataSource.DS2)
    public void mothod2(String upItemId) {
        String dataSource = DataSourceContextHolder.getDataSource();
        System.out.println("切换之后的数据源:  "+dataSource);
    }

结果：

切换之前的数据源:  ds3
切换之后的数据源:  ds2

如果是第二种方式，要在springboot启动类添加 @EnableAspectJAutoProxy(exposeProxy = true)注解，表示开启AOP代理自动配置。如果是配置文件的话，需要修改配置文件
<aop:aspectj-autoproxy proxy-target-class=“true” expose-proxy=“true”/>

以上，解决了spring aop不支持自调用的问题。