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