springboot+mybatis 实现 动态数据源

应用场景

  单个应用里面同时需要操作多个库的表，且库不固定；
  这是以前做的一个 数据解析 服务场景下的一个功能，因为每次解析的数据量很大，会生成大量的表，所以每次的解析生成的数据都会单独创建一个来存储；
  当然在其他 比如 多个数据源 实现读写分离 等场景下也可能会遇到这样的问题；

实现方案

  这里使用的方案是参考的 AbstractRoutingDataSource，重写 DataSource，并在获取 connection 的时候实现数据源的自动切换，从而实现使用一个 SqlSessionFactory 同时操作多个 数据源 DataSource 并自由切换的功能；

具体方案

  首先，定义一个数据源的上下文，既切换数据源的工具，这里使用 ThreadLocal 存储当前线程当前阶段使用的数据源 key，实现线程之间的数据源隔离；

public final class DatabaseContext {
   
    private static final ThreadLocal<String> CUR_DB = new ThreadLocal<>();
    static  {
   
        /** 默认使用的数据源 */
        CUR_DB.set("main");
    }
    /**
     * 切换 数据源
     * @param dbName
     */
    public static void switchDb(String dbName) {
   
        CUR_DB.set(dbName);
    }
    /**
     * 获取当前 数据源
     * @return
     */
    public static String getCurDb() {
   
        return CUR_DB.get();
    }
}

  然后自定义 可自由 切换 DataSource 的 DataSource，其实 spring-jdbc 提供了一个 AbstractRoutingDataSource 类是实现数据源的动态切换，但 AbstractRoutingDataSource 的数据源必须是固定的几个库，所以这里就进行了自定义实现；

/**
 * 动态数据源
 */
public class MultiDataSource extends AbstractDataSource implements InitializingBean {
   
    /**
     * 其他的动态数据源，统一起来方便管理
     */
    private static final Map<String, DataSource> DATA_SOURCE_MAP = new ConcurrentHashMap<>();
	/** 主数据源 */
    private DataSource defaultTargetDataSource;

    @Autowired
    private DataSourceConfig