springboot+mybatis實現動態切換資料來源

摘要： 前幾天有個需求，需要使用不同的資料來源，例如某業務要用A資料來源，另一個業務要用B資料來源。我上網收集了一些資料整合了一下，雖然最後這個需求不了了之了，但是多資料來源動態切換還是蠻好用的，所以記錄一下，或許以後有用呢？或者自己感興趣又想玩呢！ 1.加個依賴 <de...

前幾天有個需求，需要使用不同的資料來源，例如某業務要用A資料來源，另一個業務要用B資料來源。我上網收集了一些資料整合了一下，雖然最後這個需求不了了之了，但是多資料來源動態切換還是蠻好用的，所以記錄一下，或許以後有用呢？或者自己感興趣又想玩呢！

1.加個依賴

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

2.application.properties配置檔案

#主從資料庫
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預設載入資料來源配置

@SpringBootApplication(exclude = {DataSourceAutoConfiguration.class})
public class Application {

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

4.資料來源配置類

/**
 * 主資料來源
 */
@Configuration
@ConfigurationProperties(prefix = "master.db")
public class MasterDataSourceConfig {
private String url;
private String username;
private String password;
private String driverClassName;
}

/**
 * 從資料來源配置
 */
@Configuration
@ConfigurationProperties(prefix = "slave.db")
public class SlaveDataSourceConfig {
private String url;
private String username;
private String password;
private String driverClassName;
}

/**
 * 資料來源配置類
 */
@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.資料來源上下文

/**
 * 資料來源上下文
 */
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路由類

/*
*重寫的函式決定了最後選擇的DataSource
*/
public class MultiRouteDataSource extends AbstractRoutingDataSource {

@Override
protected Object determineCurrentLookupKey() {
//通過繫結執行緒的資料來源上下文實現多資料來源的動態切換,有興趣的可以去查閱資料或原始碼
return DataSourceContext.getDataSource();
}

}

7.使用，修改上下文中的資料來源就可以切換自己想要使用的資料來源了。

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

這種是在業務中使用程式碼設定資料來源的方式，也可以使用AOP+註解的方式實現控制，方法多多！