SpringBoot Mybatis動態(tài)數(shù)據(jù)源切換方案實現(xiàn)過程

更新時間：2020年04月17日 10:01:21 作者：經(jīng)典雞翅

這篇文章主要介紹了SpringBoot+Mybatis實現(xiàn)動態(tài)數(shù)據(jù)源切換方案過程,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下

背景

最近讓我做一個大數(shù)據(jù)的系統(tǒng)，分析了一下，麻煩的地方就是多數(shù)據(jù)源切換抽取數(shù)據(jù)?？紤]到可以跨服務器跨數(shù)據(jù)庫抽數(shù)，再整理數(shù)據(jù)，就配置了這個動態(tài)數(shù)據(jù)源的解決方案。在此分享給大家。

實現(xiàn)方案

數(shù)據(jù)庫配置文件

我們項目使用的是yml形式的配置文件，采用的是hikari的數(shù)據(jù)庫連接池。第一步我們自然是配置多個數(shù)據(jù)庫源頭。
我們找到spring的datasource，在下方配置三個數(shù)據(jù)源。

spring:
 application:
 name: dynamicDatasource
 datasource:
 test1:
  driver-class-name: com.mysql.jdbc.Driver
  url: jdbc:mysql://127.0.0.1:3306/test1?serverTimezone=GMT%2B8&characterEncoding=utf-8&useSSL=false
  username: root
  password: 123456
 test2:
  driver-class-name: com.mysql.jdbc.Driver
  url: jdbc:mysql://127.0.0.1:3306/test2?serverTimezone=GMT%2B8&characterEncoding=utf-8&useSSL=false
  username: root
  password: 123456
 test3:
  driver-class-name: com.mysql.jdbc.Driver
  url: jdbc:mysql://127.0.0.1:3306/test3?serverTimezone=GMT%2B8&characterEncoding=utf-8&useSSL=false
  username: root
  password: 123456
 hikari:
  leak-detection-threshold: 2000

定義數(shù)據(jù)源實體類

我們可以建立個datasourceBean文件夾專門管理數(shù)據(jù)源的實體類。

我們這里要建立三個實體類。分別對應test1，test2，test3

@Configuration
public class Test1DataSourceBean {

 @Value("${spring.datasource.test1.driver-class-name}")
 private String test1Driver;

 @Value("${spring.datasource.test1.url}")
 private String test1Url;

 @Value("${spring.datasource.test1.username}")
 private String test1Username;

 @Value("${spring.datasource.test1.password}")
 private String test1Password;

 @Bean(name="test1DataSource")
 public DataSource test1DataSource() throws Exception{
  HikariDataSource dataSource = new HikariDataSource();
  dataSource.setDriverClassName(test1Driver);
  dataSource.setJdbcUrl(test1Url);
  dataSource.setUsername(test1Username);
  dataSource.setPassword(test1Password);
  return dataSource;
 }
}

@Configuration
public class Test2DataSourceBean {

 @Value("${spring.datasource.test2.driver-class-name}")
 private String test2Driver;

 @Value("${spring.datasource.test2.url}")
 private String test2Url;

 @Value("${spring.datasource.test2.username}")
 private String test2Username;

 @Value("${spring.datasource.test2.password}")
 private String test2Password;

 @Bean(name="test2DataSource")
 public DataSource test2DataSource() throws Exception{
  HikariDataSource dataSource = new HikariDataSource();
  dataSource.setDriverClassName(test2Driver);
  dataSource.setJdbcUrl(test2Url);
  dataSource.setUsername(test2Username);
  dataSource.setPassword(test2Password);
  return dataSource;
 }
}

@Configuration
public class Test3DataSourceBean {

 @Value("${spring.datasource.test3.driver-class-name}")
 private String test3Driver;

 @Value("${spring.datasource.test3.url}")
 private String test3Url;

 @Value("${spring.datasource.test3.username}")
 private String test3Username;

 @Value("${spring.datasource.test3.password}")
 private String test3Password;

 @Bean(name="test3DataSource")
 public DataSource test3DataSource() throws Exception{
  HikariDataSource dataSource = new HikariDataSource();
  dataSource.setDriverClassName(test3Driver);
  dataSource.setJdbcUrl(test3Url);
  dataSource.setUsername(test3Username);
  dataSource.setPassword(test3Password);
  return dataSource;
 }
}

定義一個枚舉類管理數(shù)據(jù)源

public enum DatabaseType {

 test1("test1", "test1"),
 test2("test2", "test2"),
 test3("test3","test3");

 private String name;
 private String value;

 DatabaseType(String name, String value){
  this.name = name;
  this.value = value;
 }

 public String getName(){
  return name;
 }

 public String getValue(){
  return value;
 }
}

定義一個線程安全的數(shù)據(jù)源容器

public class DatabaseContextHolder {
 private static final ThreadLocal<DatabaseType> contextHolder = new ThreadLocal<>();
 public static void setDatabaseType(DatabaseType type){
  contextHolder.set(type);
 }
 public static DatabaseType getDatabaseType(){
  return contextHolder.get();
 }
}

定義動態(tài)數(shù)據(jù)源

public class DynamicDataSource extends AbstractRoutingDataSource{
 protected Object determineCurrentLookupKey() {
  return DatabaseContextHolder.getDatabaseType();
 }
}

mybatis配置類

網(wǎng)上的很多文章配置出來都會產(chǎn)生數(shù)據(jù)源循環(huán)依賴的問題，這里解決了這個問題。

@Configuration
@MapperScan(basePackages="cn.test.jichi", sqlSessionFactoryRef="sessionFactory")
public class MybatisConfig {

 /**
  * @Description:設置動態(tài)數(shù)據(jù)源
  */
 @Bean(name="dynamicDataSource")
 @Primary
 public DynamicDataSource DataSource(
   @Qualifier("test1DataSource") DataSource test1DataSource,
   @Qualifier("test2DataSource") DataSource test2DataSource,
   @Qualifier("test3DataSource") DataSource test3DataSource){
  Map<Object, Object> targetDataSource = new HashMap<>();
  targetDataSource.put(DatabaseType.test1, test1DataSource);
  targetDataSource.put(DatabaseType.test2, test2DataSource);
  targetDataSource.put(DatabaseType.test3, test3DataSource);
  DynamicDataSource dataSource = new DynamicDataSource();
  dataSource.setTargetDataSources(targetDataSource);
  dataSource.setDefaultTargetDataSource(test1DataSource);
  return dataSource;
 }

 /**
  * @Description:根據(jù)動態(tài)數(shù)據(jù)源創(chuàng)建sessionFactory
  */
 @Bean(name="sessionFactory")
 public SqlSessionFactory sessionFactory(
   @Qualifier("test1DataSource") DataSource test1DataSource,
   @Qualifier("test2DataSource") DataSource test2DataSource,
   @Qualifier("test3DataSource") DataSource test3DataSource) throws Exception{
  SqlSessionFactoryBean sessionFactoryBean = new SqlSessionFactoryBean();
  //構(gòu)造方法，解決動態(tài)數(shù)據(jù)源循環(huán)依賴問題。
  sessionFactoryBean.setDataSource(this.DataSource(test1DataSource,test2DataSource, test3DataSource));
  return sessionFactoryBean.getObject();
 }
}

示例

 public void testDymnaicDatasource(){
  //不切換數(shù)據(jù)源默認是自己的。
  System.out.println("-----默認數(shù)據(jù)源");
  DemoEntity totalCount = demoMapper.getTotalCount();
  String nameCount1 = totalCount.getNameCount();
  String ageCount2 = totalCount.getAgeCount();
  System.out.println("nameCount:"+nameCount1);
  System.out.println("ageCount:"+ageCount2);
  //數(shù)據(jù)源切換為branch
  System.out.println("-----數(shù)據(jù)源為test2");
  DynamicDataSourceUtils.chooseBranchDataSource();
  Integer nameCount = demoMapper.getNameCount();
  Integer ageCount = demoMapper.getAgeCount();
  System.out.println("nameCount:"+nameCount);
  System.out.println("ageCount:"+ageCount);
  //數(shù)據(jù)源為basic
  System.out.println("-----數(shù)據(jù)源為test3");
  DynamicDataSourceUtils.chooseBasicDataSource();
  Integer ageCount1 = demoMapper.getAgeCount();
  System.out.println("ageCount:"+ageCount1);

 }

總結(jié)

至此實現(xiàn)了多數(shù)據(jù)源的動態(tài)切換?？梢栽谕粋€方法里面進行操作多個數(shù)據(jù)源。

以上就是本文的全部內(nèi)容，希望對大家的學習有所幫助，也希望大家多多支持腳本之家。

您可能感興趣的文章:

java 取出文本文件中空行的實例代碼
這篇文章介紹了java 取出文本文件中空行的實例代碼，有需要的朋友可以參考一下
2013-09-09
一場由Java中Integer引發(fā)的踩坑實戰(zhàn)
Java中的數(shù)據(jù)類型分為基本數(shù)據(jù)類型和復雜數(shù)據(jù)類型int是前者而integer是后者(也就是一個類),下面這篇文章主要給大家介紹了關于由Java中Integer引發(fā)的踩坑實戰(zhàn),需要的朋友可以參考下
2022-11-11
淺析Java數(shù)據(jù)庫操作工具包jOOQ的使用
jOOQ?是一個輕量級的?Java?ORM（對象關系映射）框架,可用來構(gòu)建復雜的?SQL?查詢,這篇文章主要來和大家介紹一下jOOQ的使用,需要的可以參考下
2024-04-04
java開發(fā)使用BigDecimal避坑四則
這篇文章主要為大家介紹了java開發(fā)使用BigDecimal的避坑四則，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪
2022-07-07
SpringBoot使用Mybatis注解實現(xiàn)分頁動態(tài)sql開發(fā)教程
這篇文章主要為大家介紹了SpringBoot使用Mybatis注解實現(xiàn)分頁及動態(tài)sql開發(fā)教程，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪
2022-03-03
java實現(xiàn)的海盜算法優(yōu)化版
這篇文章主要介紹了java實現(xiàn)的海盜算法優(yōu)化版,結(jié)合實例形式分析了java海盜算法的具體實現(xiàn)技巧,需要的朋友可以參考下
2017-07-07
IDEA如何自動生成serialVersionUID的設置
這篇文章主要介紹了IDEA如何自動生成 serialVersionUID 的設置，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧
2020-09-09
maven?setting.xml文件配置禪定之旅
這篇文章主要為大家介紹了maven?setting.xml文件禪定之旅，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪
2023-05-05
Spring Boot配置讀取實現(xiàn)方法解析
這篇文章主要介紹了Spring Boot配置讀取實現(xiàn)方法解析,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下
2020-08-08
Java如何獲取真實請求IP
這篇文章主要介紹了Java如何獲取真實請求IP問題,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教
2023-08-08