前言
前幾天有個需求，需要使用不同的數據源，例如某業務要用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.Drivermaster.db.url=jdbc:mysql://localhost:3306/cbd?characterEncoding=UTF-8&useUnicode=true&useSSL=falsemaster.db.username=rootmaster.db.password=adminslave.db.driverClassName=com.mysql.jdbc.Driverslave.db.url=jdbc:mysql://localhost:3306/cbd_test?characterEncoding=UTF-8&useUnicode=true&useSSL=falseslave.db.username=rootslave.db.password=adminmybatis.config-location= classpath:config/mybatis-config.xmlmybatis.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;}/** * 數據源配置類 */@Configurationpublic 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+注解的方式實現控制，方法多多！
總結
以上就是這篇文章的全部內容了，希望本文的內容對大家的學習或者工作具有一定的參考學習價值，如果有疑問大家可以留言交流，謝謝大家對VeVb武林網的支持。
新聞熱點
疑難解答
圖片精選
