多数据源配置

关于多数据源

切换数据源的场景是在开发过程中经常会遇到的，一般应用在分库分表情况，根据不同的信息将内容存储到对应的数据库，也用于主从数据库等，再简单点的例子，如：统一系统，我们有两个城市的库，test1,test2；再不修改的配置文件，不重启项目的情况下如何做到自由的切换数据，从而方便测试？

实现思路

其实实现多数据源主要也就是三个步骤

• 自定义注解 ：我们需要自定义一个注解@DataSource，未来使用再service上或者具体的方法上面，表示使用该注解的实现类或者方法使用某一数据源
• 自定义切面 ，某个类或者方法使用@DataSource时，将@DataSource注解中记录的数据源存入到ThreaTdLocal中
• 最后当Mappers执行的时候，会自动在AbstractRoutiingDataSource中查找所需要的数据源，我们需要自定义配置继承AbstractRoutiingDataSource，并返回ThreaTdLocal中保存的数据源信息即可

具体代码思路

注解DataSource

/**
 * 该注解可以加在一个service类或者方法上，通过value 属性来指定类或者方法应该用哪个数据库
 * */

@Retention(RetentionPolicy.RUNTIME)
@Target({ElementType.TYPE,ElementType.METHOD})
public @interface DataSource {
    String value();
}

定义ThreadLocal
题外话：前几天面试时被问到了threadLocald的相关问题

public class DynamicDataSourceContextHolder {
    private static ThreadLocal<String> CONTEXT_HOLDER=new ThreadLocal<>();

    public static void setDataSourceType(String dsType){
        CONTEXT_HOLDER.set(dsType);
    }

    public static String getDataSourceType(){
        return CONTEXT_HOLDER.get();
    }
    public static void clearDataSourceType(){
        CONTEXT_HOLDER.remove();
    }
}