使用spring怎么實(shí)現(xiàn)動(dòng)態(tài)切換

本篇文章為大家展示了使用spring怎么實(shí)現(xiàn)動(dòng)態(tài)切換，內(nèi)容簡(jiǎn)明扼要并且容易理解，絕對(duì)能使你眼前一亮，通過(guò)這篇文章的詳細(xì)介紹希望你能有所收獲。

targetDataSources 就是我們的多個(gè)數(shù)據(jù)源，在初始化的時(shí)候會(huì)調(diào)用afterPropertiesSet()，去解析我們的數(shù)據(jù)源 然后 put 到 resolvedDataSources

實(shí)現(xiàn)了 DataSource 的 getConnection(); 我們看看 determineTargetDataSource(); 做了什么

通過(guò)下面的 determineCurrentLookupKey();（這個(gè)方法需要我們實(shí)現(xiàn)） 返回一個(gè)key，然后從 resolvedDataSources （其實(shí)也就是 targetDataSources） 中 get 一個(gè)數(shù)據(jù)源，實(shí)現(xiàn)了每次調(diào)用 getConnection(); 打開(kāi)連接 切換數(shù)據(jù)源，如果想動(dòng)態(tài)添加的話 只需要重新 set targetDataSources 再調(diào)用 afterPropertiesSet() 即可

Talk is cheap. Show me the code

我使用的springboot版本為 1.5.x，下面是核心代碼

完整代碼：https://gitee.com/yintianwen7/spring-dynamic-datasource (本地下載)

/**
 * 多數(shù)據(jù)源配置
 * 
 * @author Taven
 *
 */
@Configuration
@MapperScan("com.gitee.taven.mapper")
public class DataSourceConfigurer {

 /**
  * DataSource 自動(dòng)配置并注冊(cè)
  *
  * @return data source
  */
 @Bean("db0")
 @Primary
 @ConfigurationProperties(prefix = "datasource.db0")
 public DataSource dataSource0() {
  return DruidDataSourceBuilder.create().build();
 }

 /**
  * DataSource 自動(dòng)配置并注冊(cè)
  *
  * @return data source
  */
 @Bean("db1")
 @ConfigurationProperties(prefix = "datasource.db1")
 public DataSource dataSource1() {
  return DruidDataSourceBuilder.create().build();
 }

 /**
  * 注冊(cè)動(dòng)態(tài)數(shù)據(jù)源
  * 
  * @return
  */
 @Bean("dynamicDataSource")
 public DataSource dynamicDataSource() {
  DynamicRoutingDataSource dynamicRoutingDataSource = new DynamicRoutingDataSource();
  Map<Object, Object> dataSourceMap = new HashMap<>();
  dataSourceMap.put("dynamic_db0", dataSource0());
  dataSourceMap.put("dynamic_db1", dataSource1());
  dynamicRoutingDataSource.setDefaultTargetDataSource(dataSource0());// 設(shè)置默認(rèn)數(shù)據(jù)源
  dynamicRoutingDataSource.setTargetDataSources(dataSourceMap);
  return dynamicRoutingDataSource;
 }

 /**
  * Sql session factory bean.
  * Here to config datasource for SqlSessionFactory
  * <p>
  * You need to add @{@code @ConfigurationProperties(prefix = "mybatis")}, if you are using *.xml file,
  * the {@code 'mybatis.type-aliases-package'} and {@code 'mybatis.mapper-locations'} should be set in
  * {@code 'application.properties'} file, or there will appear invalid bond statement exception
  *
  * @return the sql session factory bean
  */
 @Bean
 @ConfigurationProperties(prefix = "mybatis")
 public SqlSessionFactoryBean sqlSessionFactoryBean() {
  SqlSessionFactoryBean sqlSessionFactoryBean = new SqlSessionFactoryBean();
  // 必須將動(dòng)態(tài)數(shù)據(jù)源添加到 sqlSessionFactoryBean
  sqlSessionFactoryBean.setDataSource(dynamicDataSource());
  return sqlSessionFactoryBean;
 }

 /**
  * 事務(wù)管理器
  *
  * @return the platform transaction manager
  */
 @Bean
 public PlatformTransactionManager transactionManager() {
  return new DataSourceTransactionManager(dynamicDataSource());
 }
}

通過(guò) ThreadLocal 獲取線程安全的數(shù)據(jù)源 key

package com.gitee.taven.config;

public class DynamicDataSourceContextHolder {

 private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>() {
  @Override
  protected String initialValue() {
   return "dynamic_db0";
  }
 };

 /**
  * To switch DataSource
  *
  * @param key the key
  */
 public static void setDataSourceKey(String key) {
  contextHolder.set(key);
 }

 /**
  * Get current DataSource
  *
  * @return data source key
  */
 public static String getDataSourceKey() {
  return contextHolder.get();
 }

 /**
  * To set DataSource as default
  */
 public static void clearDataSourceKey() {
  contextHolder.remove();
 }
}

動(dòng)態(tài) 添加、切換數(shù)據(jù)源

/**
 * 動(dòng)態(tài)數(shù)據(jù)源
 * 
 * @author Taven
 *
 */
public class DynamicRoutingDataSource extends AbstractRoutingDataSource {

 private final Logger logger = LoggerFactory.getLogger(getClass());

 private static Map<Object, Object> targetDataSources = new HashMap<>();
 
 /**
  * 設(shè)置當(dāng)前數(shù)據(jù)源
  *
  * @return
  */
 @Override
 protected Object determineCurrentLookupKey() {
  logger.info("Current DataSource is [{}]", DynamicDataSourceContextHolder.getDataSourceKey());
  return DynamicDataSourceContextHolder.getDataSourceKey();
 }
 
 @Override
 public void setTargetDataSources(Map<Object, Object> targetDataSources) {
  super.setTargetDataSources(targetDataSources);
  DynamicRoutingDataSource.targetDataSources = targetDataSources;
 }
 
 /**
  * 是否存在當(dāng)前key的 DataSource
  * 
  * @param key
  * @return 存在返回 true, 不存在返回 false
  */
 public static boolean isExistDataSource(String key) {
  return targetDataSources.containsKey(key);
 }
 
 /**
  * 動(dòng)態(tài)增加數(shù)據(jù)源
  * 
  * @param map 數(shù)據(jù)源屬性
  * @return
  */
 public synchronized boolean addDataSource(Map<String, String> map) {
  try {
   Connection connection = null;
   // 排除連接不上的錯(cuò)誤
   try { 
    Class.forName(map.get(DruidDataSourceFactory.PROP_DRIVERCLASSNAME));
    connection = DriverManager.getConnection(
      map.get(DruidDataSourceFactory.PROP_URL), 
      map.get(DruidDataSourceFactory.PROP_USERNAME),
      map.get(DruidDataSourceFactory.PROP_PASSWORD));
    System.out.println(connection.isClosed());
   } catch (Exception e) {
    return false;
   } finally {
    if (connection != null && !connection.isClosed()) 
     connection.close();
   }
   String database = map.get("database");//獲取要添加的數(shù)據(jù)庫(kù)名
   if (StringUtils.isBlank(database)) return false;
   if (DynamicRoutingDataSource.isExistDataSource(database)) return true; 
   DruidDataSource druidDataSource = (DruidDataSource) DruidDataSourceFactory.createDataSource(map);
   druidDataSource.init();
   Map<Object, Object> targetMap = DynamicRoutingDataSource.targetDataSources;
   targetMap.put(database, druidDataSource);
   // 當(dāng)前 targetDataSources 與 父類 targetDataSources 為同一對(duì)象 所以不需要set
//   this.setTargetDataSources(targetMap);
   this.afterPropertiesSet();
   logger.info("dataSource {} has been added", database);
  } catch (Exception e) {
   logger.error(e.getMessage());
   return false;
  }
  return true;
 } 
}

上述內(nèi)容就是使用spring怎么實(shí)現(xiàn)動(dòng)態(tài)切換，你們學(xué)到知識(shí)或技能了嗎？如果還想學(xué)到更多技能或者豐富自己的知識(shí)儲(chǔ)備，歡迎關(guān)注億速云行業(yè)資訊頻道。