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