解決mybatis-plus動態(tài)數(shù)據(jù)源切換不生效的問題
一、問題描述
在我們項(xiàng)目中，既要連接mysql，又要連接TDEngine（taos），正確配置后也無法動態(tài)切換數(shù)據(jù)源執(zhí)行sql
二、環(huán)境
1.依賴
<!--連接另外一種數(shù)據(jù)庫的驅(qū)動-->
<dependency>
<groupId>com.taosdata.jdbc</groupId>
<artifactId>taos-jdbcdriver</artifactId>
<!-- <version>2.0.32</version>-->
<version>3.0.0</version>
</dependency>
<!--mybatis plus 動態(tài)切換數(shù)據(jù)源的依賴-->
<dependency>
<groupId>com.baomidou</groupId>
<artifactId>dynamic-datasource-spring-boot-starter</artifactId>
<version>3.3.2</version>
</dependency>
2.配置
spring:
datasource:
dynamic:
strict: true
primary: mysql
datasource:
mysql:
driver-class-name: com.mysql.cj.jdbc.Driver
url: jdbc:mysql://192.xxx.xxx.xxx:3306/db_iot?useUnicode=true&characterEncoding=UTF-8&useSSL=false&serverTimezone=Asia/Shanghai
username: root
password: 123456
tdengine:
driver-class-name: com.taosdata.jdbc.TSDBDriver
url: jdbc:TAOS://192.xxx.xxx.xxx:6030/iot_data?timezone=UTC-8&charset=UTF-8&locale=en_US.UTF-8
username: root
password: taosdata
mysql1:
driver-class-name: com.mysql.cj.jdbc.Driver
url: jdbc:mysql://192.xxx.xxx.xxx:3306/db_portal?useUnicode=true&characterEncoding=UTF-8&useSSL=false&serverTimezone=Asia/Shanghai
username: root
password: 123456
三、解決方法
直接上代碼，最后會有問題分析，因?yàn)樯婕暗皆创a，這里暫時(shí)不講
這里的dynamicRoutingDataSource()不能有DynamicDataSourceAutoConfiguration里的那樣命名和返回值，那樣在注入時(shí)就會因?yàn)槠渌麛?shù)據(jù)源的注入導(dǎo)致無法注入動態(tài)數(shù)據(jù)源，需要細(xì)化到具體的類型
package com.xxx.project.iotconf.configs;
import com.baomidou.dynamic.datasource.DynamicRoutingDataSource;
import com.baomidou.dynamic.datasource.provider.DynamicDataSourceProvider;
import com.baomidou.dynamic.datasource.provider.YmlDynamicDataSourceProvider;
import com.baomidou.dynamic.datasource.spring.boot.autoconfigure.DataSourceProperty;
import com.baomidou.dynamic.datasource.spring.boot.autoconfigure.DynamicDataSourceProperties;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;
import javax.sql.DataSource;
import java.util.Map;
@Configuration
public class TDEngineConfig {
@Autowired
private DynamicDataSourceProperties properties;
@Bean
@ConditionalOnMissingBean
@Primary
public DynamicDataSourceProvider dynamicDataSourceProvider() {
Map<String, DataSourceProperty> datasourceMap = this.properties.getDatasource();
return new YmlDynamicDataSourceProvider(datasourceMap);
}
@Bean
@ConditionalOnMissingBean
public DynamicRoutingDataSource dynamicRoutingDataSource(DynamicDataSourceProvider dynamicDataSourceProvider) {
DynamicRoutingDataSource dynamicRoutingDataSource = new DynamicRoutingDataSource();
dynamicRoutingDataSource.setPrimary(this.properties.getPrimary());
dynamicRoutingDataSource.setStrict(this.properties.getStrict());
dynamicRoutingDataSource.setStrategy(this.properties.getStrategy());
dynamicRoutingDataSource.setProvider(dynamicDataSourceProvider);
dynamicRoutingDataSource.setP6spy(this.properties.getP6spy());
dynamicRoutingDataSource.setSeata(this.properties.getSeata());
Map<String, DataSource> dataSourceMap = dynamicDataSourceProvider.loadDataSources();
for (String key : dataSourceMap.keySet()) {
dynamicRoutingDataSource.addDataSource(key, dataSourceMap.get(key));
}
return dynamicRoutingDataSource;
}
}
四、測試
放入TAOS創(chuàng)建超表的SQL，由JdbcTemplate去執(zhí)行，執(zhí)行成功
package com.xxx.project.iot.pulsar.handler;
import com.baomidou.dynamic.datasource.annotation.DS;
import com.shandy.project.api.basic.dto.SDevice;
import com.shandy.project.api.basic.dto.Tuple2;
import com.shandy.project.iot.pulsar.utils.ReflectUtils;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.stereotype.Service;
import javax.annotation.PostConstruct;
import java.util.List;
import java.util.Map;
@Service
@Slf4j
public class DataHandler implements DsHandler{
@Autowired
private JdbcTemplate jdbcTemplate;
@PostConstruct
public void init() {
}
/**
* 執(zhí)行單條DDL sql
*/
@DS(value = "tdengine")
public void execute(String sql) {
jdbcTemplate.execute(sql);
log.info(sql);
}
/**
* 執(zhí)行單條DML sql
*/
@DS(value = "tdengine")
public List<Map<String, Object>> query(String sql, Object[] args) {
log.info(String.format("%s，params is {%s}", sql, args));
return jdbcTemplate.queryForList(sql, args);
}
/**
* 保存設(shè)備數(shù)據(jù)(批量，多表多條)
*/
@DS(value = "tdengine")
public void batchInsertDevice(List<SDevice> devices, String table) {
String[] sqls = null;
jdbcTemplate.batchUpdate(sqls);
log.info(sqls.toString());
}
/**
* 單條插入
*
* @param device 消息結(jié)構(gòu)體
* @param mqttObj 模型對象，IMqttR或IMqttS對象
*/
@DS(value = "tdengine")
public void insertDevice(SDevice device, Object mqttObj) {
Tuple2<String, List<Object>> t2 = ReflectUtils.getInsSql(mqttObj.getClass(), device);
List<Object> list = t2.getField(1);
String sql = t2.getField(0);
jdbcTemplate.update(sql, list.toArray());
log.info(sql);
}
}
五、問題分析
1.一開始執(zhí)行時(shí)，總是報(bào)錯(cuò)誤的SQL語句，要我檢查Mysql的版本，所以從這個(gè)提示來看，是沒有動態(tài)切換到我們的taos數(shù)據(jù)庫的。
什么原因呢？我們看com.baomidou.dynamic.datasource.spring.boot.autoconfigure.DynamicDataSourceAutoConfiguration類的方法
@Bean
@ConditionalOnMissingBean
public DataSource dataSource(DynamicDataSourceProvider dynamicDataSourceProvider) {
DynamicRoutingDataSource dataSource = new DynamicRoutingDataSource();
dataSource.setPrimary(this.properties.getPrimary());
dataSource.setStrict(this.properties.getStrict());
dataSource.setStrategy(this.properties.getStrategy());
dataSource.setProvider(dynamicDataSourceProvider);
dataSource.setP6spy(this.properties.getP6spy());
dataSource.setSeata(this.properties.getSeata());
return dataSource;
}
這里有個(gè)@ConditionalOnMissingBean注解，意思是當(dāng)dataSource對象不存在時(shí)才會進(jìn)行注入。
我發(fā)現(xiàn)我除了配置了動態(tài)數(shù)據(jù)源，也配置了druid數(shù)據(jù)源，在項(xiàng)目啟動是肯定是會注入druid的DataSource對象的，那就導(dǎo)致我們的動態(tài)數(shù)據(jù)源的DataSource對象無法注入，可能這就是切換不了的原因。這里或許可以嘗試把druid數(shù)據(jù)源去掉，但我沒有往這個(gè)方向去深究。
spring.datasource.druid.db-type=mysql spring.datasource.druid.driver-class-name=com.mysql.cj.jdbc.Driver spring.datasource.druid.url=jdbc:mysql://192.xxx.xxx.xxx:3306/db_iot?useUnicode=true&characterEncoding=UTF-8&useSSL=false&serverTimezone=Asia/Shanghai spring.datasource.druid.username=root spring.datasource.druid.password=123456
既然沒有注入DynamicRoutingDataSource對象，我自己就寫了上文中的TDEngineConfig去注入，其中方法和DynamicDataSourceAutoConfiguration類似，只是稍微做改造。
在最開始我原封不動的把DynamicDataSourceAutoConfiguration方法抄下來，在執(zhí)行SQL時(shí)報(bào)：dynamic-datasource could not find a datasource named tdengine
異常來自于在DynamicRoutingDataSource的getDataSource（）方法
public DataSource getDataSource(String ds) {
if (StringUtils.isEmpty(ds)) {
return this.determinePrimaryDataSource();
} else if (!this.groupDataSources.isEmpty() && this.groupDataSources.containsKey(ds)) {
log.debug("dynamic-datasource switch to the datasource named [{}]", ds);
return ((GroupDataSource)this.groupDataSources.get(ds)).determineDataSource();
} else if (this.dataSourceMap.containsKey(ds)) {
log.debug("dynamic-datasource switch to the datasource named [{}]", ds);
return (DataSource)this.dataSourceMap.get(ds);
} else if (this.strict) {
throw new CannotFindDataSourceException("dynamic-datasource could not find a datasource named" + ds);
} else {
return this.determinePrimaryDataSource();
}
}
從這里發(fā)現(xiàn)，DynamicRoutingDataSource的dataSourceMap是空的，那自然報(bào)錯(cuò)，但是DynamicDataSourceProperties的dataSourceMap并不是空的，所以配置并沒有出錯(cuò)，只是項(xiàng)目在啟動時(shí)沒有填充DynamicRoutingDataSource的dataSourceMap，那不妨我們自己來完成這件事情。
在TDEngineConfig的dynamicDataSourceProvider()方法中將properties的datasourceMap封裝到Y(jié)mlDynamicDataSourceProvider中，再看它的loadDataSources()，不就可以獲取到Map<String, DataSource>類型的一個(gè)對象嗎，我們把這個(gè)想辦法賦值給DynamicRoutingDataSource的dataSourceMap
public Map<String, DataSource> loadDataSources() {
return this.createDataSourceMap(this.dataSourcePropertiesMap);
}
所以在TDEngineConfig的dynamicRoutingDataSource方法中通過如下代碼獲取到dataSourceMap
Map<String, DataSource> dataSourceMap = dynamicDataSourceProvider.loadDataSources();
再通過下面代碼就將DynamicRoutingDataSource的dataSourceMap填充好了
for (String key : dataSourceMap.keySet()) {
dynamicRoutingDataSource.addDataSource(key, dataSourceMap.get(key));
}
此時(shí)完成DynamicRoutingDataSource的注入。
這樣再調(diào)用SQL時(shí)，就可以獲取@DS注解的value值，充當(dāng)key去dataSourceMap里找到對應(yīng)的數(shù)據(jù)源進(jìn)行切換
到此這篇關(guān)于解決mybatis-plus動態(tài)數(shù)據(jù)源切換不生效的問題的文章就介紹到這了,更多相關(guān)mybatis-plus動態(tài)數(shù)據(jù)源切換內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
這篇文章主要介紹了淺談SpringMVC請求映射handler源碼解讀，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2021-03-03
本文是關(guān)于Java IO文件流和垃圾回收問題，一個(gè)小的測試程序搞清楚Java IO的問題，希望能幫助有需要的小伙伴2016-07-07
這篇文章主要介紹了java回調(diào)機(jī)制實(shí)例詳解的相關(guān)資料,需要的朋友可以參考下2017-05-05
java 刪除文件夾中的所有內(nèi)容而不刪除文件夾本身的實(shí)例
下面小編就為大家?guī)硪黄猨ava 刪除文件夾中的所有內(nèi)容而不刪除文件夾本身的實(shí)例。小編覺得挺不錯(cuò)的，現(xiàn)在就分享給大家，也給大家做個(gè)參考。一起跟隨小編過來看看吧2016-12-12
東方通TongWeb結(jié)合Spring-Boot使用的實(shí)現(xiàn)
本文主要介紹了東方通TongWeb結(jié)合Spring-Boot使用的實(shí)現(xiàn),文中通過示例代碼介紹的非常詳細(xì),對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2024-07-07
詳解springboot-mysql-pagehelper分頁插件集成
這篇文章主要介紹了springboot-mysql-pagehelper分頁插件集成,本文通過實(shí)例代碼給大家介紹的非常詳細(xì)，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下2023-07-07
本文將全面介紹 Spring 5。我將介紹 Java SE 8 和 Java EE 7 API 的基準(zhǔn)升級、Spring 5 的新反應(yīng)式編程模型、HTTP/2 支持，以及 Spring 通過 Kotlin 對函數(shù)式編程的全面支持。2020-10-10
兩兩比較待排序記錄的關(guān)鍵字，發(fā)現(xiàn)兩個(gè)記錄的次序相反時(shí)即進(jìn)行交換，直到?jīng)]有反序的記錄為止2013-01-01
這篇文章主要介紹了Maven 插件開發(fā)的詳細(xì)整理的相關(guān)資料,需要的朋友可以看下，希望能夠給你帶來幫助2021-09-09
本篇文章是對SQLite多線程的使用進(jìn)行了詳細(xì)的分析介紹，需要的朋友參考下2013-05-05