springboot使用DynamicDataSource動(dòng)態(tài)切換數(shù)據(jù)源的實(shí)現(xiàn)過(guò)程
DynamicDataSource是一個(gè)數(shù)據(jù)源路由器,可以根據(jù)上下文動(dòng)態(tài)選擇數(shù)據(jù)源。可以在每個(gè)請(qǐng)求或線程中將數(shù)據(jù)源設(shè)置為當(dāng)前需要使用的數(shù)據(jù).
1. 創(chuàng)建一個(gè)DynamicDataSource類來(lái)實(shí)現(xiàn)數(shù)據(jù)源路由邏輯
創(chuàng)建一個(gè)DynamicDataSource類,它繼承自AbstractRoutingDataSource。在該類中重寫**determineCurrentLookupKey()**方法,該方法返回一個(gè)字符串,用于指示當(dāng)前要使用哪個(gè)數(shù)據(jù)源
public class DynamicDataSource extends AbstractRoutingDataSource {
@Override
protected Object determineCurrentLookupKey() {
return DynamicDataSourceContextHolder.getDataSourceName();
}
public DynamicDataSource(DataSource defaultTargetDataSource, Map<Object, Object> targetDataSources) {
super.setDefaultTargetDataSource(defaultTargetDataSource);
super.setTargetDataSources(targetDataSources);
super.afterPropertiesSet();
}
}2. 創(chuàng)建DynamicDataSourceContextHolder線程安全類
在多線程環(huán)境下,如果多個(gè)線程同時(shí)訪問(wèn)同一個(gè)方法,并且每個(gè)線程要使用不同的數(shù)據(jù)源,那么就需要對(duì)數(shù)據(jù)源進(jìn)行動(dòng)態(tài)切換。如果在方法中使用一個(gè)共享的變量來(lái)存儲(chǔ)當(dāng)前要使用的數(shù)據(jù)源名稱,那么就會(huì)存在線程安全問(wèn)題,可能會(huì)導(dǎo)致不同線程之間的數(shù)據(jù)源切換混亂,或者數(shù)據(jù)源切換不成功的情況發(fā)生。
為了解決這個(gè)問(wèn)題,可以使用ThreadLocal來(lái)存儲(chǔ)當(dāng)前要使用的數(shù)據(jù)源名稱。ThreadLocal是一種線程本地存儲(chǔ)機(jī)制,它可以為每個(gè)線程提供一個(gè)獨(dú)立的變量副本,使得每個(gè)線程都可以獨(dú)立地操作自己的變量副本,而不會(huì)影響其他線程的變量副本。
在使用DynamicDataSource進(jìn)行數(shù)據(jù)源切換時(shí),每個(gè)線程都可以通過(guò)ThreadLocal來(lái)獨(dú)立地設(shè)置和獲取當(dāng)前要使用的數(shù)據(jù)源名稱,避免了多個(gè)線程之間數(shù)據(jù)源切換的混亂和不成功的情況。同時(shí),在方法執(zhí)行完畢后,使用ThreadLocal也可以避免內(nèi)存泄漏問(wèn)題。使用ThreadLocal來(lái)存儲(chǔ)當(dāng)前要使用的數(shù)據(jù)源名稱。**setDataSource()**方法用于設(shè)置當(dāng)前要使用的數(shù)據(jù)源名稱,**getDataSource()**方法用于獲取當(dāng)前要使用的數(shù)據(jù)源名稱,**clearDataSource()**方法用于清除當(dāng)前要使用的數(shù)據(jù)源名稱
public class DynamicDataSourceContextHolder {
private static final ThreadLocal<String> contextHolder = new ThreadLocal<>();
/**
* 設(shè)置當(dāng)前數(shù)據(jù)源的名稱
*/
public static void setDataSourceName(String dataSourceName) {
contextHolder.set(dataSourceName);
}
/**
* 獲取當(dāng)前數(shù)據(jù)源的名稱
*/
public static String getDataSourceName() {
return contextHolder.get();
}
/**
* 清除當(dāng)前數(shù)據(jù)源的名稱
*/
public static void clearDataSourceName() {
contextHolder.remove();
}
}3.創(chuàng)建多數(shù)據(jù)源配置
通過(guò)yml配置,將多數(shù)據(jù)源獲取到MutilDataSourceProperties中
spring:
datasource:
driver-class-name: com.mysql.cj.jdbc.Driver
url: jdbc:mysql://****:3306/emp_ts?nullNamePatternMatchesAll=true&tinyInt1isBit=false&useSSL=false&nullCatalogMeansCurrent=true&useUnicode=true&characterEncoding=UTF-8&serverTimezone=GMT%2B8
username: root
password: root
mutil-datasource:
connection:
- dbName: dataSource1
dbDriver: com.mysql.cj.jdbc.Driver
dbUrl: jdbc:mysql://****:3306/emp_ts_1?nullNamePatternMatchesAll=true&tinyInt1isBit=false&useSSL=false&nullCatalogMeansCurrent=true&useUnicode=true&characterEncoding=UTF-8&serverTimezone=GMT%2B8
dbUsername: root
dbPassword: rootpublic class DbConnection {
private String dbName;
private String dbDialect;
private String dbDriver;
private String dbUrl;
private String dbUsername;
private String dbPassword;
public DbConnection() {
}
public String getDbName() {
return this.dbName;
}
public String getDbDialect() {
return this.dbDialect;
}
public String getDbDriver() {
return this.dbDriver;
}
public String getDbUrl() {
return this.dbUrl;
}
public String getDbUsername() {
return this.dbUsername;
}
public String getDbPassword() {
return this.dbPassword;
}
public void setDbName(String dbName) {
this.dbName = dbName;
}
public void setDbDialect(String dbDialect) {
this.dbDialect = dbDialect;
}
public void setDbDriver(String dbDriver) {
this.dbDriver = dbDriver;
}
public void setDbUrl(String dbUrl) {
this.dbUrl = dbUrl;
}
public void setDbUsername(String dbUsername) {
this.dbUsername = dbUsername;
}
public void setDbPassword(String dbPassword) {
this.dbPassword = dbPassword;
}
}@Configuration
@ConfigurationProperties(
prefix = "mutil-datasource"
)
public class MutilDataSourceProperties {
private List<DbConnection> connection = new ArrayList();
public MutilDataSourceProperties() {
}
public List<DbConnection> getConnection() {
return this.connection;
}
public void setConnection(List<DbConnection> connection) {
this.connection = connection;
}
}4.創(chuàng)建DataSourceConfig
在配置類中創(chuàng)建默認(rèn)數(shù)據(jù)源及獲取其他多數(shù)據(jù)源進(jìn)行創(chuàng)建,默認(rèn)數(shù)據(jù)源為spring.datasource下配置的數(shù)據(jù)源
@Configuration
public class DataSourceConfig {
@Resource
private MutilDataSourceProperties mutilDataSourceProperties;
public DataSourceConfig() {
}
@Bean
@ConfigurationProperties("spring.datasource")
public DataSource hikariDataSource(DataSourceProperties properties) {
return DataSourceBuilder.create(properties.getClassLoader()).driverClassName(properties.determineDriverClassName()).url(properties.determineUrl()).username(properties.determineUsername()).password(properties.determinePassword()).build();
}
@Bean
@Primary
public DynamicDataSource dynamicDataSource(DataSource hikariDataSource) {
Map<Object, Object> targetDataSources = this.createTargetDataSource();
return new DynamicDataSource(hikariDataSource, targetDataSources);
}
private Map<Object, Object> createTargetDataSource() {
Map<Object, Object> targetDataSources = new HashMap();
List<DbConnection> connections = this.mutilDataSourceProperties.getConnection();
connections.forEach(e -> {
DataSource dataSource = this.createDataSource(e);
targetDataSources.put(e.getDbName(), dataSource);
});
return targetDataSources;
}
public DataSource createDataSource(DbConnection connection) {
return DataSourceBuilder.create().driverClassName(connection.getDbDriver()).url(connection.getDbUrl())
.username(connection.getDbName()).password(connection.getDbPassword()).build();
}
}5. 使用
@Service
public class UserServiceImpl implements UserService {
@Autowired
private UserMapper userMapper;
@Override
public void addUser(User user) {
DynamicDataSource.setDataSource("dataSource1");
userMapper.insert(user);
}
@Override
public void updateUser(User user) {
DynamicDataSource.setDataSource("dataSource1");
userMapper.updateById(user);
}
}6.優(yōu)缺點(diǎn)
優(yōu)點(diǎn):
- 簡(jiǎn)單易用:使用 DynamicDataSource 和 ThreadLocal 進(jìn)行動(dòng)態(tài)數(shù)據(jù)源切換,配置相對(duì)簡(jiǎn)單,易于上手。
- 可擴(kuò)展性強(qiáng):通過(guò)繼承 AbstractRoutingDataSource 類,可以實(shí)現(xiàn)自定義的數(shù)據(jù)源路由策略。并且,由于采用了抽象類,擴(kuò)展性也比較好。
- 線程安全:使用 ThreadLocal 來(lái)存儲(chǔ)當(dāng)前數(shù)據(jù)源的名稱,可以避免多線程之間數(shù)據(jù)源切換的混亂和不成功的情況。
缺點(diǎn):
- 不能同時(shí)訪問(wèn)多個(gè)數(shù)據(jù)源:在使用 DynamicDataSource 進(jìn)行動(dòng)態(tài)數(shù)據(jù)源切換時(shí),同一時(shí)間只能訪問(wèn)一個(gè)數(shù)據(jù)源,不能同時(shí)訪問(wèn)多個(gè)數(shù)據(jù)源。
- 執(zhí)行效率稍低:在使用 DynamicDataSource 進(jìn)行動(dòng)態(tài)數(shù)據(jù)源切換時(shí),每次數(shù)據(jù)源切換都需要進(jìn)行一次路由選擇,會(huì)稍微影響執(zhí)行效率。
- 不支持事務(wù)嵌套:在使用 DynamicDataSource 進(jìn)行動(dòng)態(tài)數(shù)據(jù)源切換時(shí),如果在一個(gè)事務(wù)中需要訪問(wèn)多個(gè)數(shù)據(jù)源,那么就需要進(jìn)行事務(wù)管理,而 DynamicDataSource 并不支持事務(wù)嵌套。
綜上所述,使用 DynamicDataSource 和 ThreadLocal 進(jìn)行動(dòng)態(tài)數(shù)據(jù)源切換,優(yōu)點(diǎn)是簡(jiǎn)單易用、可擴(kuò)展性強(qiáng)、線程安全,缺點(diǎn)是不能同時(shí)訪問(wèn)多個(gè)數(shù)據(jù)源、執(zhí)行效率稍低、不支持事務(wù)嵌套。在具體使用時(shí)需要根據(jù)業(yè)務(wù)場(chǎng)景進(jìn)行選擇。
總結(jié)
到此這篇關(guān)于springboot使用DynamicDataSource動(dòng)態(tài)切換數(shù)據(jù)源實(shí)現(xiàn)的文章就介紹到這了,更多相關(guān)springboot動(dòng)態(tài)切換數(shù)據(jù)源內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
相關(guān)文章
springboot jpa實(shí)現(xiàn)優(yōu)雅處理isDelete的默認(rèn)值
如果多個(gè)實(shí)體類都有 isDelete 字段,并且你希望在插入時(shí)為它們統(tǒng)一設(shè)置默認(rèn)值時(shí)改怎么做呢,本文為大家整理了一些方法,希望對(duì)大家有所幫助2024-11-11
Mybatis事務(wù)如何跟Spring結(jié)合(數(shù)據(jù)庫(kù)事務(wù)特性和Spring事務(wù)管理源碼)
MyBatis與Spring的事務(wù)結(jié)合主要是通過(guò)Spring的事務(wù)管理和MyBatis的數(shù)據(jù)庫(kù)操作來(lái)實(shí)現(xiàn)的,在本文中,我們將從數(shù)據(jù)庫(kù)事務(wù)特性和Spring事務(wù)管理源碼兩個(gè)角度來(lái)分析MyBatis事務(wù)如何與Spring結(jié)合到一起的原理,感興趣的朋友一起看看吧2024-01-01
SpringBoot里使用Servlet進(jìn)行請(qǐng)求的實(shí)現(xiàn)示例
這篇文章主要介紹了SpringBoot里使用Servlet進(jìn)行請(qǐng)求的實(shí)現(xiàn)示例,文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2021-01-01
Spring Boot中@RequestParam參數(shù)的5種情況說(shuō)明
這篇文章主要介紹了Spring Boot中@RequestParam參數(shù)的5種情況說(shuō)明,具有很好的參考價(jià)值,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方,望不吝賜教2021-08-08
Spring中的@EnableWebSecurity注解詳解
這篇文章主要介紹了Spring中的@EnableWebSecurity注解詳解,EnableWebSecurity注解是個(gè)組合注解,它的注解中,又使用了@EnableGlobalAuthentication注解,需要的朋友可以參考下2023-12-12

