常州网站设计公司,动漫设计专业怎么样,怎么查询企业电话,大连电商平台有哪些背景
在项目开发过程中#xff0c;我们可能会遇到一个场景#xff1a;某个类型数据源有多个数据源实例#xff0c;需要我们按照不同的请求切换到不同数据源去。 而目前绝大多数java应用都是基于Spring框架来开发#xff0c;我们很多时候相关的数据源连接都是交给了Spring框…背景
在项目开发过程中我们可能会遇到一个场景某个类型数据源有多个数据源实例需要我们按照不同的请求切换到不同数据源去。 而目前绝大多数java应用都是基于Spring框架来开发我们很多时候相关的数据源连接都是交给了Spring框架去管理这就需要Spring能够支持动态数据源切换。
方案
Spring中预留了这个接口通过AbstractRoutingDataSource能够动态切换数据源。
public abstract class AbstractRoutingDataSource extends AbstractDataSource implements InitializingBean {这是一个抽象类预留了一个抽象方法
protected abstract Object determineCurrentLookupKey();我们知道数据源一般都会提供一个getConnection方法来获取一个连接在AbstractRoutingDataSource 实现如下 Overridepublic Connection getConnection() throws SQLException {return determineTargetDataSource().getConnection();}protected DataSource determineTargetDataSource() {Assert.notNull(this.resolvedDataSources, DataSource router not initialized);Object lookupKey determineCurrentLookupKey();DataSource dataSource this.resolvedDataSources.get(lookupKey);if (dataSource null (this.lenientFallback || lookupKey null)) {dataSource this.resolvedDefaultDataSource;}if (dataSource null) {throw new IllegalStateException(Cannot determine target DataSource for lookup key [ lookupKey ]);}return dataSource;}可以看到AbstractRoutingDataSource 获取连接的主要逻辑就是通过determineCurrentLookupKey获取到一个数据源的关联key然后从resolvedDataSources中去获取。 而resolvedDataSources的初始化则放在afterPropertiesSet中 Overridepublic void afterPropertiesSet() {if (this.targetDataSources null) {throw new IllegalArgumentException(Property targetDataSources is required);}this.resolvedDataSources CollectionUtils.newHashMap(this.targetDataSources.size());this.targetDataSources.forEach((key, value) - {Object lookupKey resolveSpecifiedLookupKey(key);DataSource dataSource resolveSpecifiedDataSource(value);this.resolvedDataSources.put(lookupKey, dataSource);});if (this.defaultTargetDataSource ! null) {this.resolvedDefaultDataSource resolveSpecifiedDataSource(this.defaultTargetDataSource);}}这里起始就是通过targetDataSources中指定的数据源复制到resolvedDataSources 中去。因此如果多数源是固定的那么只需要实现determineCurrentLookupKey方法即可。但是如果多数据源不固定比如可能会有数据源的变更那么这种实现是不能够支持因为这种实现从服务启动的视乎后续数据源就不能发生变更这需要我们自己实现determineTargetDataSource. 下面是一个参考实现
public class DataSourceContextHolder {private static final ThreadLocalString DATASOURCE_CONTEXT_KEY_HOLDER new ThreadLocal();public static void switchDataSource(String key){log.info(Switch to data source: key);DATASOURCE_CONTEXT_KEY_HOLDER.set(key);}public static String getDataSourceKey(){return DATASOURCE_CONTEXT_KEY_HOLDER.get() ;}}public class DynamicDataSource extends AbstractRoutingDataSource {private MapObject, Object targetDataSources new HashMap();private MapObject, DataSource dataSources new HashMap();public DynamicDataSource (){super.setDefaultTargetDataSource(null);super.setTargetDataSources(targetDataSources);super.afterPropertiesSet();}Overrideprotected DataSource determineTargetDataSource() {Object dataSourceKey determineCurrentLookupKey();return dataSources.get(dataSourceKey);}Overrideprotected Object determineCurrentLookupKey() {return DataSourceContextHolder .getDataSourceKey();}public synchronized void addDataSource(String key, DataSource dataSource){targetDataSources.put(key,dataSource);dataSources.put(key,dataSource);log.info(add tenant dynamic dataSource for tenantId {} ,key);}
}这样我们通过DataSourceContextHolder 来调整当前线程关联的数据源。
