這篇文章將為大家詳細講解有關(guān)Spring Boot數(shù)據(jù)源加載及其多數(shù)據(jù)源的示例分析，小編覺得挺實用的，因此分享給大家做個參考，希望大家閱讀完這篇文章后可以有所收獲。

專注于為中小企業(yè)提供成都網(wǎng)站制作、網(wǎng)站建設(shè)服務,電腦端+手機端+微信端的三站合一,更高效的管理,為中小企業(yè)曲陽免費做網(wǎng)站提供優(yōu)質(zhì)的服務。我們立足成都，凝聚了一批互聯(lián)網(wǎng)行業(yè)人才，有力地推動了上1000+企業(yè)的穩(wěn)健成長，幫助中小企業(yè)通過網(wǎng)站建設(shè)實現(xiàn)規(guī)模擴充和轉(zhuǎn)變。

業(yè)務需求

• 提供所有微服務數(shù)據(jù)源的圖形化維護功能

• 代碼生成可以根據(jù)選擇的數(shù)據(jù)源加載表等源信息

• 數(shù)據(jù)源管理要支持動態(tài)配置，實時生效

附錄效果圖

實現(xiàn)思路

本文提供方法僅供類似簡單業(yè)務場景，在生產(chǎn)環(huán)境和復雜的業(yè)務場景 請使用分庫分表的中間件（例如mycat）或者框架 sharding-sphere （一直在用）等

先來看Spring 默認的數(shù)據(jù)源注入策略，如下代碼默認的事務管理器在初始化時回去加載數(shù)據(jù)源實現(xiàn)。這里就是我們動態(tài)數(shù)據(jù)源的入口

// 默認的事務管理器
ppublic class DataSourceTransactionManager extends AbstractPlatformTransactionManager
  implements ResourceTransactionManager, InitializingBean {
 
 // 啟動時候注入一個數(shù)據(jù)源
 public void setDataSource(@Nullable DataSource dataSource) {
  if (dataSource instanceof TransactionAwareDataSourceProxy) {
   this.dataSource = ((TransactionAwareDataSourceProxy) dataSource).getTargetDataSource();
  }
  else {
   this.dataSource = dataSource;
  }
 }
」

通過注入一個新的DataSourceTransactionManager 實現(xiàn)，并且給它設(shè)置多個 DataSource 來實現(xiàn)多數(shù)據(jù)源實現(xiàn)

看下Spring 默認提供的路由數(shù)據(jù)源字段

public abstract class AbstractRoutingDataSource extends AbstractDataSource implements InitializingBean {
 
 // 用戶設(shè)置的全部的數(shù)據(jù)源配置
 @Nullable
 private Map<Object, Object> targetDataSources;
 // 為空默認的數(shù)據(jù)源配置
 @Nullable
 private Object defaultTargetDataSource;
 
 // 路由鍵查找實現(xiàn)
 private DataSourceLookup dataSourceLookup = new JndiDataSourceLookup();
 
 // 最終有效的數(shù)據(jù)源配置（一般清空對應上邊用戶的設(shè)置）
 @Nullable
 private Map<Object, DataSource> resolvedDataSources;
}

開始動手

實現(xiàn)AbstractRoutingDataSource，定一個動態(tài)數(shù)據(jù)源實現(xiàn)，只需要實現(xiàn)他的路由key 查找方法即可。

這里的路由key 對應其實是resolvedDataSources Map 的key喲

@Slf4j
public class DynamicDataSource extends AbstractRoutingDataSource {

 /**
  * 指定路由Key,這里很簡單 獲取 threadLocal 中目標key 即可
  *
  * @return
  */
 @Override
 protected Object determineCurrentLookupKey() {
  return DynamicDataSourceContextHolder.getDataSourceType();
 }
}

把我們動態(tài)數(shù)據(jù)源實現(xiàn)注入到Spring 的事務管理器,去數(shù)據(jù)庫查詢出來全部的數(shù)據(jù)源信息，定義一個個具體的數(shù)據(jù)源實現(xiàn) 我這里使用的HikariDataSource 給他賦值等等

@Slf4j
@Configuration
@AllArgsConstructor
public class DynamicDataSourceConfig implements TransactionManagementConfigurer {
 private final Map<Object, Object> dataSourceMap = new HashMap<>(8);
 private final DataSourceProperties dataSourceProperties;

 @Bean("dynamicDataSource")
 public DynamicDataSource dataSource() {
  JdbcTemplate(dds).queryForList(DataSourceConstant.QUERY_DS_SQL);
  log.info("開始 -> 初始化動態(tài)數(shù)據(jù)源");
  Optional.of(dbList).ifPresent(list -> list.forEach(db -> {
   log.info("數(shù)據(jù)源:{}", db.get(DataSourceConstant.DS_NAME));
   HikariDataSource ds = new HikariDataSource();
   dataSourceMap.put(db.get(DataSourceConstant.DS_ROUTE_KEY), ds);
  }));
  
  DynamicDataSource ds = new DynamicDataSource();
  ds.setTargetDataSources(dataSourceMap);
  return ds;
 }

 @Bean
 public PlatformTransactionManager txManager() {
  return new DataSourceTransactionManager(dataSource());
 }

 @Override
 public PlatformTransactionManager annotationDrivenTransactionManager() {
  return txManager();
 }

}

怎么使用

只需要根據(jù)用戶前臺選擇的數(shù)據(jù)源key ，在業(yè)務類保存到TTL 即可，會自動根據(jù)選擇路由數(shù)據(jù)源

DynamicDataSourceContextHolder.setDataSourceType（key）

這里當然也可以根據(jù)AOP 自定義注解等實現(xiàn)。

如何動態(tài)數(shù)據(jù)源動態(tài)配置

上邊其實已經(jīng)完成了 我們想要的需求功能，但是有什么問題呢？
我們在數(shù)據(jù)源管理面維護了數(shù)據(jù)源，動態(tài)去修改這個 dataSourceMap 其實是無效的，不能做到實時刷新

我們來看下 AbstractRoutingDataSource 的加載map 數(shù)據(jù)源的源碼，只有在初始化的時候調(diào)用 afterPropertiesSet 去初始數(shù)據(jù)源map.

那我們只要獲取當前的DynamicDataSource bean 手動調(diào)用afterPropertiesSet 即可。

整個代碼如下

public class DynamicDataSourceConfig implements TransactionManagementConfigurer {
 private final Map<Object, Object> dataSourceMap = new HashMap<>(8);
 private final DataSourceProperties dataSourceProperties;
 private final StringEncryptor stringEncryptor;

 @Bean("dynamicDataSource")
 public DynamicDataSource dataSource() {
  DynamicDataSource ds = new DynamicDataSource();
  HikariDataSource cads = new HikariDataSource();
  cads.setJdbcUrl(dataSourceProperties.getUrl());
  cads.setDriverClassName(dataSourceProperties.getDriverClassName());
  cads.setUsername(dataSourceProperties.getUsername());
  cads.setPassword(dataSourceProperties.getPassword());
  ds.setDefaultTargetDataSource(cads);
  dataSourceMap.put(0, cads);
  ds.setTargetDataSources(dataSourceMap);
  return ds;
 }

 /**
  * 組裝默認配置的數(shù)據(jù)源，查詢數(shù)據(jù)庫配置
  */
 @PostConstruct
 public void init() {
  DriverManagerDataSource dds = new DriverManagerDataSource();
  dds.setUrl(dataSourceProperties.getUrl());
  dds.setDriverClassName(dataSourceProperties.getDriverClassName());
  dds.setUsername(dataSourceProperties.getUsername());
  dds.setPassword(dataSourceProperties.getPassword());

  List<Map<String, Object>> dbList = new JdbcTemplate(dds).queryForList(DataSourceConstant.QUERY_DS_SQL);
  log.info("開始 -> 初始化動態(tài)數(shù)據(jù)源");
  Optional.of(dbList).ifPresent(list -> list.forEach(db -> {
   log.info("數(shù)據(jù)源:{}", db.get(DataSourceConstant.DS_NAME));
   HikariDataSource ds = new HikariDataSource();
   ds.setJdbcUrl(String.valueOf(db.get(DataSourceConstant.DS_JDBC_URL)));
   ds.setDriverClassName(Driver.class.getName());
   ds.setUsername((String) db.get(DataSourceConstant.DS_USER_NAME));

   String decPwd = stringEncryptor.decrypt((String) db.get(DataSourceConstant.DS_USER_PWD));
   ds.setPassword(decPwd);
   dataSourceMap.put(db.get(DataSourceConstant.DS_ROUTE_KEY), ds);
  }));

  log.info("完畢 -> 初始化動態(tài)數(shù)據(jù)源,共計 {} 條", dataSourceMap.size());
 }

 /**
  * 重新加載數(shù)據(jù)源配置
  */
 public Boolean reload() {
  init();
  DynamicDataSource dataSource = dataSource();
  dataSource.setTargetDataSources(dataSourceMap);
  dataSource.afterPropertiesSet();
  return Boolean.FALSE;
 }


 @Bean
 public PlatformTransactionManager txManager() {
  return new DataSourceTransactionManager(dataSource());
 }

 @Override
 public PlatformTransactionManager annotationDrivenTransactionManager() {
  return txManager();
 }

關(guān)于“Spring Boot數(shù)據(jù)源加載及其多數(shù)據(jù)源的示例分析”這篇文章就分享到這里了，希望以上內(nèi)容可以對大家有一定的幫助，使各位可以學到更多知識，如果覺得文章不錯，請把它分享出去讓更多的人看到。

新聞名稱：SpringBoot數(shù)據(jù)源加載及其多數(shù)據(jù)源的示例分析
網(wǎng)站地址：http://chinadenli.net/article36/gogcsg.html

成都網(wǎng)站建設(shè)公司_創(chuàng)新互聯(lián)，為您提供網(wǎng)頁設(shè)計公司、企業(yè)網(wǎng)站制作、做網(wǎng)站、面包屑導航、網(wǎng)站設(shè)計、App設(shè)計

廣告

聲明：本網(wǎng)站發(fā)布的內(nèi)容（圖片、視頻和文字）以用戶投稿、用戶轉(zhuǎn)載內(nèi)容為主，如果涉及侵權(quán)請盡快告知，我們將會在第一時間刪除。文章觀點不代表本網(wǎng)站立場，如需處理請聯(lián)系客服。電話：028-86922220；郵箱：631063699@qq.com。內(nèi)容未經(jīng)允許不得轉(zhuǎn)載，或轉(zhuǎn)載時需注明來源： 創(chuàng)新互聯(lián)