spring boot + mybatis實現(xiàn)動態(tài)切換數(shù)據(jù)源實例代碼

更新時間：2018年10月30日 15:20:38 作者：擼碼識途

這篇文章主要給大家介紹了關于spring boot + mybatis實現(xiàn)動態(tài)切換數(shù)據(jù)源的相關資料，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧

前言

前幾天有個需求，需要使用不同的數(shù)據(jù)源，例如某業(yè)務要用A數(shù)據(jù)源，另一個業(yè)務要用B數(shù)據(jù)源。我上網(wǎng)收集了一些資料整合了一下，雖然最后這個需求不了了之了，但是多數(shù)據(jù)源動態(tài)切換還是蠻好用的，所以記錄一下，或許以后有用呢？或者自己感興趣又想玩呢！

下面話不多說了，隨著小編來一起看看詳細的介紹吧

方法如下：

1.加個依賴

 <dependency>
  <groupId>org.mybatis.spring.boot</groupId>
  <artifactId>mybatis-spring-boot-starter</artifactId>
　　　　　　　<version>1.3.1</version>
 </dependency>

2.application.properties配置文件

#主從數(shù)據(jù)庫
master.db.driverClassName=com.mysql.jdbc.Driver
master.db.url=jdbc:mysql://localhost:3306/cbd?characterEncoding=UTF-8&useUnicode=true&useSSL=false
master.db.username=root
master.db.password=admin
slave.db.driverClassName=com.mysql.jdbc.Driver
slave.db.url=jdbc:mysql://localhost:3306/cbd_test?characterEncoding=UTF-8&useUnicode=true&useSSL=false
slave.db.username=root
slave.db.password=admin

mybatis.config-location= classpath:config/mybatis-config.xml
mybatis.mapper-locations=classpath:mapper/**/*.xml

3.禁用springboot默認加載數(shù)據(jù)源配置

@SpringBootApplication(exclude = {DataSourceAutoConfiguration.class})
public class Application {
 
 public static void main(String[] args) throws Exception {
  SpringApplication.run(Application.class, args);
 }
}

4.數(shù)據(jù)源配置類

/**
 * 主數(shù)據(jù)源
 */
@Configuration
@ConfigurationProperties(prefix = "master.db")
public class MasterDataSourceConfig {
 private String url;
 private String username;
 private String password;
 private String driverClassName;
}

/**
 * 從數(shù)據(jù)源配置
 */
@Configuration
@ConfigurationProperties(prefix = "slave.db")
public class SlaveDataSourceConfig {
 private String url;
 private String username;
 private String password;
 private String driverClassName;
}

/**
 * 數(shù)據(jù)源配置類
 */
@Configuration
public class DataSourceComponent {

 @Resource
 private MasterDataSourceConfig masterDataSourceConfig;

 @Resource
 private SlaveDataSourceConfig slaveDataSourceConfig;


  @Bean(name = "master")
 public DataSource masterDataSource() {
  DataSource dataSource = new DataSource();
  dataSource.setUrl(masterDataSourceConfig.getUrl());
  dataSource.setUsername(masterDataSourceConfig.getUsername());
  dataSource.setPassword(masterDataSourceConfig.getPassword());
  dataSource.setDriverClassName(masterDataSourceConfig.getDriverClassName());
  return dataSource;
 }

 @Bean(name = "slave")
 public DataSource slaveDataSource() {
  DataSource dataSource = new DataSource();
  dataSource.setUrl(slaveDataSourceConfig.getUrl());
  dataSource.setUsername(slaveDataSourceConfig.getUsername());
  dataSource.setPassword(slaveDataSourceConfig.getPassword());
  dataSource.setDriverClassName(slaveDataSourceConfig.getDriverClassName());
  return dataSource;
 }

 @Primary//不加這個會報錯。
 @Bean(name = "multiDataSource")
 public MultiRouteDataSource exampleRouteDataSource() {
  MultiRouteDataSource multiDataSource = new MultiRouteDataSource();
  Map<Object, Object> targetDataSources = new HashMap<>();
  targetDataSources.put("master", masterDataSource());
  targetDataSources.put("slave", slaveDataSource());
  multiDataSource.setTargetDataSources(targetDataSources);
  multiDataSource.setDefaultTargetDataSource(masterDataSource());
  return multiDataSource;
 }
}

5.數(shù)據(jù)源上下文

/**
 * 數(shù)據(jù)源上下文
 */
public class DataSourceContext {
 private static final ThreadLocal<String> contextHolder = new ThreadLocal<>();

 public static void setDataSource(String value) {
  contextHolder.set(value);
 }

 public static String getDataSource() {
  return contextHolder.get();
 }

 public static void clearDataSource() {
  contextHolder.remove();
 }
}

6.DataSource路由類

/*
* 重寫的函數(shù)決定了最后選擇的DataSource
*/
public class MultiRouteDataSource extends AbstractRoutingDataSource {

 @Override
 protected Object determineCurrentLookupKey() {
  //通過綁定線程的數(shù)據(jù)源上下文實現(xiàn)多數(shù)據(jù)源的動態(tài)切換,有興趣的可以去查閱資料或源碼
  return DataSourceContext.getDataSource();
 }

}

7.使用，修改上下文中的數(shù)據(jù)源就可以切換自己想要使用的數(shù)據(jù)源了。

 public UserVO findUser(String username) {
  DataSourceContext.setDataSource("slave");
  UserVO userVO = userMapper.findByVO(username);
  System.out.println(userVO.getName()+"=====================");
  return null;
 }

這種是在業(yè)務中使用代碼設置數(shù)據(jù)源的方式，也可以使用AOP+注解的方式實現(xiàn)控制，方法多多！

總結

以上就是這篇文章的全部內(nèi)容了，希望本文的內(nèi)容對大家的學習或者工作具有一定的參考學習價值，如果有疑問大家可以留言交流，謝謝大家對腳本之家的支持。

您可能感興趣的文章: