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不支持自调用的问题。