前言
之前我写过一篇数据源自动切换的博文，点击查看=>多数据源自动切换，里面用两种方式实现了多数据源自动切换，但是无论是自己实现还是用官方的轮子，都有一个痛点，就是在请求数据源不是默认数据源的时候，要重复的用注解来切换数据源，这样很繁琐，近期看到同事的一个实现，眼前一亮，觉得有必要单独写一篇文章来分享一下，实现方式原理还是和我上面那篇文章里的一样，不同的是用切面和约定的方式实现全自动切换数据源。
本次代码示例会简单表述优化的内容，不会有完整代码，建议在操作实现完多数据源自动切换里的代码后再看本文章。
正片开始
写一个数据源枚举类
@Getter
@AllArgsConstructor
public enum DBTypeEnum {
PRIMARY("primary"),
LOCAL("local"),
PROD("prod");
private final String value;
}
写一个切面类
@Component
@Order(value = -999)
@Slf4j
@Aspect
public class DataSourceSwitchAspect {
@Pointcut("execution(* com.along.mapper.primary..*.*(..)) || execution(* com.along.service.primary..*.*(..))")
private void primaryAspect() {
}
@Pointcut("execution(* com.along.mapper.local..*.*(..)) || execution(* com.along.service.local..*.*(..))")
private void localAspect() {
}
@Pointcut("execution(* com.along.mapper.prod..*.*(..)) || execution(* com.along.service.prod..*.*(..))")
private void prodAspect() {
}
@Before("primaryAspect()")
public void upms() {
DbContextHolder.setDbType(DBTypeEnum.PRIMARY);
}
@Before("localAspect()")
public void local() {
DbContextHolder.setDbType(DBTypeEnum.LOCAL);
}
@Before("prodAspect()")
public void prod() {
DbContextHolder.setDbType(DBTypeEnum.PROD);
}
}
看到这是不是很清晰了，用切面的方式，监控指定的文件夹，当请求调用到指定的文件夹里的类前，自动切换数据源为对应的数据源。
这样配置完之后我们就不在需要手动切换数据源，只需要开发过程中按照遵守项目约定好的规范，把mapper接口类放在指定的目录内，就无需关心数据库是怎么切换的。
这样基本就分享完了，改动很简单，主要就是提供一个思路，具体的实现大家在真实项目中可以根据实际情况自由发挥。
