springboot使用DynamicDataSource動態(tài)切換數(shù)據(jù)源的實現(xiàn)過程
DynamicDataSource是一個數(shù)據(jù)源路由器,可以根據(jù)上下文動態(tài)選擇數(shù)據(jù)源??梢栽诿總€請求或線程中將數(shù)據(jù)源設(shè)置為當前需要使用的數(shù)據(jù).
1. 創(chuàng)建一個DynamicDataSource類來實現(xiàn)數(shù)據(jù)源路由邏輯
創(chuàng)建一個DynamicDataSource類,它繼承自AbstractRoutingDataSource。在該類中重寫**determineCurrentLookupKey()**方法,該方法返回一個字符串,用于指示當前要使用哪個數(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)境下,如果多個線程同時訪問同一個方法,并且每個線程要使用不同的數(shù)據(jù)源,那么就需要對數(shù)據(jù)源進行動態(tài)切換。如果在方法中使用一個共享的變量來存儲當前要使用的數(shù)據(jù)源名稱,那么就會存在線程安全問題,可能會導(dǎo)致不同線程之間的數(shù)據(jù)源切換混亂,或者數(shù)據(jù)源切換不成功的情況發(fā)生。
為了解決這個問題,可以使用ThreadLocal來存儲當前要使用的數(shù)據(jù)源名稱。ThreadLocal是一種線程本地存儲機制,它可以為每個線程提供一個獨立的變量副本,使得每個線程都可以獨立地操作自己的變量副本,而不會影響其他線程的變量副本。
在使用DynamicDataSource進行數(shù)據(jù)源切換時,每個線程都可以通過ThreadLocal來獨立地設(shè)置和獲取當前要使用的數(shù)據(jù)源名稱,避免了多個線程之間數(shù)據(jù)源切換的混亂和不成功的情況。同時,在方法執(zhí)行完畢后,使用ThreadLocal也可以避免內(nèi)存泄漏問題。使用ThreadLocal來存儲當前要使用的數(shù)據(jù)源名稱。**setDataSource()**方法用于設(shè)置當前要使用的數(shù)據(jù)源名稱,**getDataSource()**方法用于獲取當前要使用的數(shù)據(jù)源名稱,**clearDataSource()**方法用于清除當前要使用的數(shù)據(jù)源名稱
public class DynamicDataSourceContextHolder {
private static final ThreadLocal<String> contextHolder = new ThreadLocal<>();
/**
* 設(shè)置當前數(shù)據(jù)源的名稱
*/
public static void setDataSourceName(String dataSourceName) {
contextHolder.set(dataSourceName);
}
/**
* 獲取當前數(shù)據(jù)源的名稱
*/
public static String getDataSourceName() {
return contextHolder.get();
}
/**
* 清除當前數(shù)據(jù)源的名稱
*/
public static void clearDataSourceName() {
contextHolder.remove();
}
}3.創(chuàng)建多數(shù)據(jù)源配置
通過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)建默認數(shù)據(jù)源及獲取其他多數(shù)據(jù)源進行創(chuàng)建,默認數(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)缺點
優(yōu)點:
- 簡單易用:使用 DynamicDataSource 和 ThreadLocal 進行動態(tài)數(shù)據(jù)源切換,配置相對簡單,易于上手。
- 可擴展性強:通過繼承 AbstractRoutingDataSource 類,可以實現(xiàn)自定義的數(shù)據(jù)源路由策略。并且,由于采用了抽象類,擴展性也比較好。
- 線程安全:使用 ThreadLocal 來存儲當前數(shù)據(jù)源的名稱,可以避免多線程之間數(shù)據(jù)源切換的混亂和不成功的情況。
缺點:
- 不能同時訪問多個數(shù)據(jù)源:在使用 DynamicDataSource 進行動態(tài)數(shù)據(jù)源切換時,同一時間只能訪問一個數(shù)據(jù)源,不能同時訪問多個數(shù)據(jù)源。
- 執(zhí)行效率稍低:在使用 DynamicDataSource 進行動態(tài)數(shù)據(jù)源切換時,每次數(shù)據(jù)源切換都需要進行一次路由選擇,會稍微影響執(zhí)行效率。
- 不支持事務(wù)嵌套:在使用 DynamicDataSource 進行動態(tài)數(shù)據(jù)源切換時,如果在一個事務(wù)中需要訪問多個數(shù)據(jù)源,那么就需要進行事務(wù)管理,而 DynamicDataSource 并不支持事務(wù)嵌套。
綜上所述,使用 DynamicDataSource 和 ThreadLocal 進行動態(tài)數(shù)據(jù)源切換,優(yōu)點是簡單易用、可擴展性強、線程安全,缺點是不能同時訪問多個數(shù)據(jù)源、執(zhí)行效率稍低、不支持事務(wù)嵌套。在具體使用時需要根據(jù)業(yè)務(wù)場景進行選擇。
總結(jié)
到此這篇關(guān)于springboot使用DynamicDataSource動態(tài)切換數(shù)據(jù)源實現(xiàn)的文章就介紹到這了,更多相關(guān)springboot動態(tài)切換數(shù)據(jù)源內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
相關(guān)文章
springboot jpa實現(xiàn)優(yōu)雅處理isDelete的默認值
如果多個實體類都有 isDelete 字段,并且你希望在插入時為它們統(tǒng)一設(shè)置默認值時改怎么做呢,本文為大家整理了一些方法,希望對大家有所幫助2024-11-11
Mybatis事務(wù)如何跟Spring結(jié)合(數(shù)據(jù)庫事務(wù)特性和Spring事務(wù)管理源碼)
MyBatis與Spring的事務(wù)結(jié)合主要是通過Spring的事務(wù)管理和MyBatis的數(shù)據(jù)庫操作來實現(xiàn)的,在本文中,我們將從數(shù)據(jù)庫事務(wù)特性和Spring事務(wù)管理源碼兩個角度來分析MyBatis事務(wù)如何與Spring結(jié)合到一起的原理,感興趣的朋友一起看看吧2024-01-01
SpringBoot里使用Servlet進行請求的實現(xiàn)示例
這篇文章主要介紹了SpringBoot里使用Servlet進行請求的實現(xiàn)示例,文中通過示例代碼介紹的非常詳細,對大家的學習或者工作具有一定的參考學習價值,需要的朋友們下面隨著小編來一起學習學習吧2021-01-01
Spring Boot中@RequestParam參數(shù)的5種情況說明
這篇文章主要介紹了Spring Boot中@RequestParam參數(shù)的5種情況說明,具有很好的參考價值,具有很好的參考價值,希望對大家有所幫助。如有錯誤或未考慮完全的地方,望不吝賜教2021-08-08
Spring中的@EnableWebSecurity注解詳解
這篇文章主要介紹了Spring中的@EnableWebSecurity注解詳解,EnableWebSecurity注解是個組合注解,它的注解中,又使用了@EnableGlobalAuthentication注解,需要的朋友可以參考下2023-12-12

