SpringBoot利用dynamic-datasource-spring-boot-starter解決多數(shù)據(jù)源問題
概要
自己閑暇時間想實現(xiàn)一個多租戶平臺,需要實現(xiàn)數(shù)據(jù)分離,動態(tài)配置生成數(shù)據(jù)源,憑著自己的感覺搭建了一套簡單的方案
dynamic-datasource-spring-boot-starter 是一個用于在 Spring Boot 項目中實現(xiàn)動態(tài)數(shù)據(jù)源切換的工具,這里主要通過這個工具實現(xiàn)在系統(tǒng)運(yùn)行中創(chuàng)建數(shù)據(jù)庫,生成數(shù)據(jù)庫連接。
整體架構(gòu)構(gòu)想
【創(chuàng)建數(shù)據(jù)源】使用DefaultDataSourceCreator類下的createDataSource方法創(chuàng)建數(shù)據(jù)源。
【存儲數(shù)據(jù)源】使用DynamicRoutingDataSource方法的addDataSource方法以鍵值對存儲數(shù)據(jù)源。
【切換數(shù)據(jù)源】使用**@DS**注解動態(tài)切換數(shù)據(jù)源。
操作步驟
創(chuàng)建數(shù)據(jù)源
創(chuàng)建數(shù)據(jù)源方法大致如下,主要就是創(chuàng)建數(shù)據(jù)庫的參數(shù)配置,讓生成器生成數(shù)據(jù)源。
//默認(rèn)使用Mysql8.0,并且連接mysql數(shù)據(jù)庫表,通過這個連接創(chuàng)建新的數(shù)據(jù)庫數(shù)據(jù) DynamicRoutingDataSource ds = (DynamicRoutingDataSource) dataSource; // 數(shù)據(jù)源參數(shù)配置 DataSourceProperty dataSourceProperty = new DataSourceProperty(); //固定使用mysql8 String dbClassType = "com.mysql.cj.jdbc.Driver"; String dbUrl = String.format("jdbc:mysql://%s:%s/%s?characterEncoding=utf8&useSSL=false&serverTimezone=GMT%2B8", dbIp, dbPort, dbName); //這里使用IP進(jìn)行區(qū)分 dataSourceProperty.setPoolName(dbIp); dataSourceProperty.setUsername(userName); dataSourceProperty.setPassword(passWord); dataSourceProperty.setUrl(dbUrl); dataSourceProperty.setDriverClassName(dbClassType); // 數(shù)據(jù)源全部懶加載,避免一次性聲明過多連接 dataSourceProperty.setLazy(true); // 生成數(shù)據(jù)源 DataSource dataSource = dataSourceCreator.createDataSource(dataSourceProperty); //最后存儲到ds中,后面可以用個不同的key去連接不同的數(shù)據(jù)源 ds.addDataSource(dbIp,dataSource);
后面在調(diào)用不同數(shù)據(jù)庫的時候只需要使用**@DS**注解,代碼如下:
/** * 調(diào)用不同的數(shù)據(jù)連接生成新的數(shù)據(jù)庫 * @param onlySignIp * @param dbName * @return */ @DS(value = "#onlySignIp") public int createDataSource(String onlySignIp,@Param("dbName")String dbName);
切換數(shù)據(jù)源
一開始我覺得這樣就可以直接在系統(tǒng)中自定義創(chuàng)建數(shù)據(jù)庫,dynamic-datasource底層在初始化加載的時候會生成一條責(zé)任鏈,一共為三個節(jié)點分別為DsHeaderProcessor、DsSessionProcessor、DsSpelExpressionProcessor分別對應(yīng)#header、#session和spel表達(dá)式,源碼如下:
@Bean @ConditionalOnMissingBean public DsProcessor dsProcessor(BeanFactory beanFactory) { DsHeaderProcessor headerProcessor = new DsHeaderProcessor(); DsSessionProcessor sessionProcessor = new DsSessionProcessor(); DsSpelExpressionProcessor spelExpressionProcessor = new DsSpelExpressionProcessor(); spelExpressionProcessor.setBeanResolver(new BeanFactoryResolver(beanFactory)); headerProcessor.setNextProcessor(sessionProcessor); sessionProcessor.setNextProcessor(spelExpressionProcessor); return headerProcessor; }
在這里 根據(jù)底層代碼理論上來說是不會應(yīng)該有問題的,好像是用它自定義的責(zé)任鏈不行,這里我就自己重寫了這條責(zé)任鏈(也就自己復(fù)制源代碼將他注入到spring容器中),然后在重寫DsSpelExpressionProcessor,通過spel表達(dá)式來選擇數(shù)據(jù)源。
public class DsSpelExpressionProcessor extends DsProcessor { /** * 參數(shù)發(fā)現(xiàn)器 */ private static final ParameterNameDiscoverer NAME_DISCOVERER = new DefaultParameterNameDiscoverer(); /** * Express語法解析器 */ private static final ExpressionParser PARSER = new SpelExpressionParser(); /** * 解析上下文的模板 * 對于默認(rèn)不設(shè)置的情況下,從參數(shù)中取值的方式 #param1 * 設(shè)置指定模板 ParserContext.TEMPLATE_EXPRESSION 后的取值方式: #{#param1} * issues: https://github.com/baomidou/dynamic-datasource-spring-boot-starter/issues/199 */ private ParserContext parserContext = new ParserContext() { @Override public boolean isTemplate() { return false; } @Override public String getExpressionPrefix() { return null; } @Override public String getExpressionSuffix() { return null; } }; private BeanResolver beanResolver; @Override public boolean matches(String key) { return true; } @Override public String doDetermineDatasource(MethodInvocation invocation, String key) { Object[] arguments = invocation.getArguments(); StandardEvaluationContext context = new StandardEvaluationContext(arguments); //默認(rèn)使用第一個參數(shù) String replace = key.replace("#", ""); context.setVariable(replace,arguments[0]); final Object value = PARSER.parseExpression(key, parserContext).getValue(context); return value == null ? null : value.toString(); } public void setParserContext(ParserContext parserContext) { this.parserContext = parserContext; } public void setBeanResolver(BeanResolver beanResolver) { this.beanResolver = beanResolver; } }
這樣配合@DS注解就可以實現(xiàn)數(shù)據(jù)源的動態(tài)切換了。
后續(xù)問題
在需要切換數(shù)據(jù)源的情況下,將對應(yīng)數(shù)據(jù)源Key信息作為Mapper第一個參數(shù)傳遞
例如:
/** * 調(diào)用不同的數(shù)據(jù)連接生成新的數(shù)據(jù)庫 * @param onlySignIp * @param dbName * @return */ @DS(value = "#onlySignIp") public int createDataSource(String onlySignIp,@Param("dbName")String dbName);
在切換數(shù)據(jù)源的情況下必須要加上**@DS**注解,而且還必須將對應(yīng)數(shù)據(jù)源的Key作為第一個參數(shù)進(jìn)行傳入
小結(jié)
通過對 dynamic-datasource-spring-boot-starter 的剖析,我們簡單的實現(xiàn)了自定義創(chuàng)建數(shù)據(jù)源,切換數(shù)據(jù)源的操作
到此這篇關(guān)于SpringBoot利用dynamic-datasource-spring-boot-starter解決多數(shù)據(jù)源問題的文章就介紹到這了,更多相關(guān)SpringBoot多數(shù)據(jù)源內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
- springboot項目如何配置多數(shù)據(jù)源
- SpringBoot實現(xiàn)JPA多數(shù)據(jù)源配置小結(jié)
- SpringBoot實現(xiàn)多數(shù)據(jù)源配置的示例詳解
- SpringBoot中實現(xiàn)多數(shù)據(jù)源連接和切換的方案
- 淺析SpringBoot多數(shù)據(jù)源實現(xiàn)方案
- springboot項目實現(xiàn)多數(shù)據(jù)源配置使用dynamic-datasource-spring-boot-starter的操作步驟
- springboot配置多數(shù)據(jù)源的一款框架(dynamic-datasource-spring-boot-starter)
相關(guān)文章
RestTemplate發(fā)送請求時Cookie的影響及注意事項說明
這篇文章主要介紹了RestTemplate發(fā)送請求時Cookie的影響及注意事項說明,具有很好的參考價值,希望對大家有所幫助。2023-07-07Java?設(shè)計模式以虹貓藍(lán)兔的故事講解原型模式
原型模式是用于創(chuàng)建重復(fù)的對象,同時又能保證性能。這種類型的設(shè)計模式屬于創(chuàng)建型模式,它提供了一種創(chuàng)建對象的最佳方式,今天通過本文給大家介紹下Java 原型設(shè)計模式,感興趣的朋友一起看看吧2022-04-04JavaWeb?Servlet實現(xiàn)文件上傳與下載功能實例
因自己負(fù)責(zé)的項目中需要實現(xiàn)文件上傳,所以下面下面這篇文章主要給大家介紹了關(guān)于JavaWeb?Servlet實現(xiàn)文件上傳與下載功能的相關(guān)資料,需要的朋友可以參考下2022-04-04Maven+Tomcat8 實現(xiàn)自動化部署的方法
本篇文章主要介紹了Maven+Tomcat8 實現(xiàn)自動化部署的方法,小編覺得挺不錯的,現(xiàn)在分享給大家,也給大家做個參考。一起跟隨小編過來看看吧2017-10-10