Spring Boot + Mybatis多數(shù)據(jù)源和動(dòng)態(tài)數(shù)據(jù)源配置方法
網(wǎng)上的文章基本上都是只有多數(shù)據(jù)源或只有動(dòng)態(tài)數(shù)據(jù)源，而最近的項(xiàng)目需要同時(shí)使用兩種方式，記錄一下配置方法供大家參考。
應(yīng)用場景
項(xiàng)目需要同時(shí)連接兩個(gè)不同的數(shù)據(jù)庫A, B，并且它們都為主從架構(gòu)，一臺(tái)寫庫，多臺(tái)讀庫。
多數(shù)據(jù)源
首先要將spring boot自帶的DataSourceAutoConfiguration禁掉，因?yàn)樗鼤?huì)讀取application.properties文件的spring.datasource.*屬性并自動(dòng)配置單數(shù)據(jù)源。在@SpringBootApplication注解中添加exclude屬性即可：
@SpringBootApplication(exclude = {
DataSourceAutoConfiguration.class
})
public class TitanWebApplication {
public static void main(String[] args) {
SpringApplication.run(TitanWebApplication.class, args);
}
}
然后在application.properties中配置多數(shù)據(jù)源連接信息：
# titan庫 spring.datasource.titan-master.url=jdbc:mysql://X.X.X.X:port/titan?characterEncoding=UTF-8 spring.datasource.titan-master.username= spring.datasource.titan-master.password= spring.datasource.titan-master.driver-class-name=com.mysql.jdbc.Driver # 連接池配置 # 省略 # 其它庫 spring.datasource.db2.url=jdbc:mysql://X.X.X.X:port/titan2?characterEncoding=UTF-8 spring.datasource.db2.username= spring.datasource.db2.password= spring.datasource.db2.driver-class-name=com.mysql.jdbc.Driver
由于我們禁掉了自動(dòng)數(shù)據(jù)源配置，因些下一步就需要手動(dòng)將這些數(shù)據(jù)源創(chuàng)建出來：
@Configuration
public class DataSourceConfig {
@Bean(name = "titanMasterDS")
@ConfigurationProperties(prefix = "spring.datasource.titan-master") // application.properteis中對(duì)應(yīng)屬性的前綴
public DataSource dataSource1() {
return DataSourceBuilder.create().build();
}
@Bean(name = "ds2")
@ConfigurationProperties(prefix = "spring.datasource.db2") // application.properteis中對(duì)應(yīng)屬性的前綴
public DataSource dataSource2() {
return DataSourceBuilder.create().build();
}
}
接下來需要配置兩個(gè)mybatis的SqlSessionFactory分別使用不同的數(shù)據(jù)源：
@Configuration
@MapperScan(basePackages = {"titan.mapper"}, sqlSessionFactoryRef = "sqlSessionFactory1")
public class MybatisDbAConfig {
@Autowired
@Qualifier("titanMasterDS")
private DataSource ds1;
@Bean
public SqlSessionFactory sqlSessionFactory1() throws Exception {
SqlSessionFactoryBean factoryBean = new SqlSessionFactoryBean();
factoryBean.setDataSource(ds1); // 使用titan數(shù)據(jù)源, 連接titan庫
return factoryBean.getObject();
}
@Bean
public SqlSessionTemplate sqlSessionTemplate1() throws Exception {
SqlSessionTemplate template = new SqlSessionTemplate(sqlSessionFactory1()); // 使用上面配置的Factory
return template;
}
}
經(jīng)過上面的配置后，titan.mapper下的Mapper接口，都會(huì)使用titan數(shù)據(jù)源。同理可配第二個(gè)SqlSessionFactory:
@Configuration
@MapperScan(basePackages = {"other.mapper"}, sqlSessionFactoryRef = "sqlSessionFactory2")
public class MybatisDbBConfig {
@Autowired
@Qualifier("ds2")
private DataSource ds2;
@Bean
public SqlSessionFactory sqlSessionFactory2() throws Exception {
SqlSessionFactoryBean factoryBean = new SqlSessionFactoryBean();
factoryBean.setDataSource(ds2);
return factoryBean.getObject();
}
@Bean
public SqlSessionTemplate sqlSessionTemplate2() throws Exception {
SqlSessionTemplate template = new SqlSessionTemplate(sqlSessionFactory2());
return template;
}
}
完成這些配置后，假設(shè)有2個(gè)Mapper titan.mapper.UserMapper和other.mapper.RoleMapper，使用前者時(shí)會(huì)自動(dòng)連接titan庫，后者連接ds2庫。
動(dòng)態(tài)數(shù)據(jù)源
使用動(dòng)態(tài)數(shù)據(jù)源的初衷，是能在應(yīng)用層做到讀寫分離，即在程序代碼中控制不同的查詢方法去連接不同的庫。除了這種方法以外，數(shù)據(jù)庫中間件也是個(gè)不錯(cuò)的選擇，它的優(yōu)點(diǎn)是數(shù)據(jù)庫集群對(duì)應(yīng)用來說只暴露為單庫，不需要切換數(shù)據(jù)源的代碼邏輯。
我們通過自定義注解 + AOP的方式實(shí)現(xiàn)數(shù)據(jù)源動(dòng)態(tài)切換。
首先定義一個(gè)ContextHolder, 用于保存當(dāng)前線程使用的數(shù)據(jù)源名：
public class DataSourceContextHolder {
public static final Logger log = LoggerFactory.getLogger(DataSourceContextHolder.class);
/**
* 默認(rèn)數(shù)據(jù)源
*/
public static final String DEFAULT_DS = "titan-master";
private static final ThreadLocal<String> contextHolder = new ThreadLocal<>();
// 設(shè)置數(shù)據(jù)源名
public static void setDB(String dbType) {
log.debug("切換到{}數(shù)據(jù)源", dbType);
contextHolder.set(dbType);
}
// 獲取數(shù)據(jù)源名
public static String getDB() {
return (contextHolder.get());
}
// 清除數(shù)據(jù)源名
public static void clearDB() {
contextHolder.remove();
}
}
然后自定義一個(gè)javax.sql.DataSource接口的實(shí)現(xiàn)，這里只需要繼承Spring為我們預(yù)先實(shí)現(xiàn)好的父類AbstractRoutingDataSource即可：
public class DynamicDataSource extends AbstractRoutingDataSource {
private static final Logger log = LoggerFactory.getLogger(DynamicDataSource.class);
@Override
protected Object determineCurrentLookupKey() {
log.debug("數(shù)據(jù)源為{}", DataSourceContextHolder.getDB());
return DataSourceContextHolder.getDB();
}
}
創(chuàng)建動(dòng)態(tài)數(shù)據(jù)源：
/**
* 動(dòng)態(tài)數(shù)據(jù)源: 通過AOP在不同數(shù)據(jù)源之間動(dòng)態(tài)切換
* @return
*/
@Bean(name = "dynamicDS1")
public DataSource dataSource() {
DynamicDataSource dynamicDataSource = new DynamicDataSource();
// 默認(rèn)數(shù)據(jù)源
dynamicDataSource.setDefaultTargetDataSource(dataSource1());
// 配置多數(shù)據(jù)源
Map<Object, Object> dsMap = new HashMap(5);
dsMap.put("titan-master", dataSource1());
dsMap.put("ds2", dataSource2());
dynamicDataSource.setTargetDataSources(dsMap);
return dynamicDataSource;
}
自定義注釋@DS用于在編碼時(shí)指定方法使用哪個(gè)數(shù)據(jù)源：
@Retention(RetentionPolicy.RUNTIME)
@Target({
ElementType.METHOD
})
public @interface DS {
String value() default "titan-master";
}
編寫AOP切面，實(shí)現(xiàn)切換邏輯：
@Aspect
@Component
public class DynamicDataSourceAspect {
@Before("@annotation(DS)")
public void beforeSwitchDS(JoinPoint point){
//獲得當(dāng)前訪問的class
Class<?> className = point.getTarget().getClass();
//獲得訪問的方法名
String methodName = point.getSignature().getName();
//得到方法的參數(shù)的類型
Class[] argClass = ((MethodSignature)point.getSignature()).getParameterTypes();
String dataSource = DataSourceContextHolder.DEFAULT_DS;
try {
// 得到訪問的方法對(duì)象
Method method = className.getMethod(methodName, argClass);
// 判斷是否存在@DS注解
if (method.isAnnotationPresent(DS.class)) {
DS annotation = method.getAnnotation(DS.class);
// 取出注解中的數(shù)據(jù)源名
dataSource = annotation.value();
}
} catch (Exception e) {
e.printStackTrace();
}
// 切換數(shù)據(jù)源
DataSourceContextHolder.setDB(dataSource);
}
@After("@annotation(DS)")
public void afterSwitchDS(JoinPoint point){
DataSourceContextHolder.clearDB();
}
}
完成上述配置后，在先前SqlSessionFactory配置中指定使用DynamicDataSource就可以在Service中愉快的切換數(shù)據(jù)源了：
@Autowired
private UserAModelMapper userAMapper;
@DS("titan-master")
public String ds1() {
return userAMapper.selectByPrimaryKey(1).getName();
}
@DS("ds2")
public String ds2() {
return userAMapper.selectByPrimaryKey(1).getName();
}
總結(jié)
以上所述是小編給大家介紹的Spring Boot + Mybatis多數(shù)據(jù)源和動(dòng)態(tài)數(shù)據(jù)源配置方法，希望對(duì)大家有所幫助，如果大家有任何疑問請(qǐng)給我留言，小編會(huì)及時(shí)回復(fù)大家的。在此也非常感謝大家對(duì)腳本之家網(wǎng)站的支持！
相關(guān)文章
Java實(shí)現(xiàn)支付對(duì)接常用加密方式的示例代碼
這篇文章主要為大家詳細(xì)介紹了Java如何實(shí)現(xiàn)支付對(duì)接時(shí)常用加密方式，文中的示例代碼講解詳細(xì)，對(duì)我們學(xué)習(xí)Java有一點(diǎn)幫助，需要的可以參考一下2023-02-02
深度剖析java動(dòng)態(tài)靜態(tài)代理原理源碼
這篇文章主要介紹了深度剖析java動(dòng)態(tài)靜態(tài)代理原理源碼，關(guān)于Java中的動(dòng)態(tài)代理，我們首先需要了解的是一種常用的設(shè)計(jì)模式--代理模式，而對(duì)于代理，根據(jù)創(chuàng)建代理類的時(shí)間點(diǎn)，又可以分為靜態(tài)代理和動(dòng)態(tài)代理。,需要的朋友可以參考下2019-06-06
Springboot?Vue實(shí)現(xiàn)單點(diǎn)登陸功能示例詳解
這篇文章主要為大家介紹了Springboot?Vue實(shí)現(xiàn)單點(diǎn)登陸功能示例詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪2023-01-01
使用fileupload組件實(shí)現(xiàn)文件上傳功能
這篇文章主要為大家詳細(xì)介紹了使用fileupload實(shí)現(xiàn)文件上傳功能，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2019-10-10
java原生序列化和Kryo序列化性能實(shí)例對(duì)比分析
這篇文章主要介紹了java原生序列化和Kryo序列化性能實(shí)例對(duì)比分析，涉及Java和kryo序列化和反序列化相關(guān)實(shí)例，小編覺得很不錯(cuò)，這里分享給大家，希望給大家一個(gè)參考。2017-10-10
Windows下使用Graalvm將Springboot應(yīng)用編譯成exe大大提高啟動(dòng)和運(yùn)行效率(推薦)
這篇文章主要介紹了Windows下使用Graalvm將Springboot應(yīng)用編譯成exe大大提高啟動(dòng)和運(yùn)行效率,本文通過圖文并茂的形式給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下2023-02-02
大家好，本篇文章主要講的是Java實(shí)現(xiàn)猜數(shù)字小游戲，感興趣的同學(xué)趕快來看一看吧，對(duì)你有幫助的話記得收藏一下2022-01-01
SpringBoot中使用spring-retry 解決失敗重試調(diào)用
本文主要介紹了SpringBoot中使用spring-retry 解決失敗重試調(diào)用，文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2023-07-07
Java 并發(fā)編程ArrayBlockingQueue的實(shí)現(xiàn)
這篇文章主要介紹了Java 并發(fā)編程ArrayBlockingQueue的實(shí)現(xiàn),文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2021-02-02
本文主要介紹了Spring Boot集成Druid查看配置是否生效的方法，文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2023-06-06