SpringBoot利用dynamic-datasource-spring-boot-starter解决多数据源问题

 更新时间:2025年03月17日 15:47:14   作者:一个有女朋友的程序员  
dynamic-datasource-spring-boot-starter 是一个用于在 Spring Boot 项目中实现动态数据源切换的工具,下面我们看看如何使用dynamic-datasource-spring-boot-starter解决多数据源问题吧

概要

自己闲暇时间想实现一个多租户平台,需要实现数据分离,动态配置生成数据源,凭着自己的感觉搭建了一套简单的方案

dynamic-datasource-spring-boot-starter 是一个用于在 Spring Boot 项目中实现动态数据源切换的工具,这里主要通过这个工具实现在系统运行中创建数据库,生成数据库连接。

整体架构构想

【创建数据源】使用DefaultDataSourceCreator类下的createDataSource方法创建数据源。

【存储数据源】使用DynamicRoutingDataSource方法的addDataSource方法以键值对存储数据源。

【切换数据源】使用**@DS**注解动态切换数据源。

操作步骤

创建数据源

创建数据源方法大致如下,主要就是创建数据库的参数配置,让生成器生成数据源。

        //默认使用Mysql8.0,并且连接mysql数据库表,通过这个连接创建新的数据库数据
        DynamicRoutingDataSource ds = (DynamicRoutingDataSource) dataSource;
        // 数据源参数配置
        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进行区分
        dataSourceProperty.setPoolName(dbIp);
        dataSourceProperty.setUsername(userName);
        dataSourceProperty.setPassword(passWord);
        dataSourceProperty.setUrl(dbUrl);
        dataSourceProperty.setDriverClassName(dbClassType);
        // 数据源全部懒加载,避免一次性声明过多连接
        dataSourceProperty.setLazy(true);
        // 生成数据源
        DataSource dataSource = dataSourceCreator.createDataSource(dataSourceProperty);
		//最后存储到ds中,后面可以用个不同的key去连接不同的数据源
		ds.addDataSource(dbIp,dataSource);

后面在调用不同数据库的时候只需要使用**@DS**注解,代码如下:

    /**
     * 调用不同的数据连接生成新的数据库
     * @param onlySignIp 
     * @param dbName
     * @return
     */
    @DS(value = "#onlySignIp")
    public int createDataSource(String onlySignIp,@Param("dbName")String dbName);

切换数据源

一开始我觉得这样就可以直接在系统中自定义创建数据库,dynamic-datasource底层在初始化加载的时候会生成一条责任链,一共为三个节点分别为DsHeaderProcessor、DsSessionProcessor、DsSpelExpressionProcessor分别对应#header、#session和spel表达式,源码如下:

    @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;
    }

在这里 根据底层代码理论上来说是不会应该有问题的,好像是用它自定义的责任链不行,这里我就自己重写了这条责任链(也就自己复制源代码将他注入到spring容器中),然后在重写DsSpelExpressionProcessor,通过spel表达式来选择数据源。

public class DsSpelExpressionProcessor extends DsProcessor {

    /**
     * 参数发现器
     */
    private static final ParameterNameDiscoverer NAME_DISCOVERER = new DefaultParameterNameDiscoverer();
    /**
     * Express语法解析器
     */
    private static final ExpressionParser PARSER = new SpelExpressionParser();
    /**
     * 解析上下文的模板
     * 对于默认不设置的情况下,从参数中取值的方式 #param1
     * 设置指定模板 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);
        //默认使用第一个参数
        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注解就可以实现数据源的动态切换了。

后续问题

在需要切换数据源的情况下,将对应数据源Key信息作为Mapper第一个参数传递

例如:

    /**
     * 调用不同的数据连接生成新的数据库
     * @param onlySignIp 
     * @param dbName
     * @return
     */
    @DS(value = "#onlySignIp")
    public int createDataSource(String onlySignIp,@Param("dbName")String dbName);

在切换数据源的情况下必须要加上**@DS**注解,而且还必须将对应数据源的Key作为第一个参数进行传入

小结

通过对 dynamic-datasource-spring-boot-starter 的剖析,我们简单的实现了自定义创建数据源,切换数据源的操作

到此这篇关于SpringBoot利用dynamic-datasource-spring-boot-starter解决多数据源问题的文章就介绍到这了,更多相关SpringBoot多数据源内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

相关文章

  • RestTemplate发送请求时Cookie的影响及注意事项说明

    RestTemplate发送请求时Cookie的影响及注意事项说明

    这篇文章主要介绍了RestTemplate发送请求时Cookie的影响及注意事项说明,具有很好的参考价值,希望对大家有所帮助。
    2023-07-07
  • 解决eclipse中console控制选项不见了的方法

    解决eclipse中console控制选项不见了的方法

    eclipse是一款用于编译java语言的程序,利用这款软件我们可以制作很多有趣的小程序,也可以制作一些大型的软件项目,有的用户在使用eclipse的时候会遇到console消失的情况,所以本文给大家介绍了解决eclipse中console控制选项不见了的方法,需要的朋友可以参考下
    2024-03-03
  • Java 设计模式以虹猫蓝兔的故事讲解原型模式

    Java 设计模式以虹猫蓝兔的故事讲解原型模式

    原型模式是用于创建重复的对象,同时又能保证性能。这种类型的设计模式属于创建型模式,它提供了一种创建对象的最佳方式,今天通过本文给大家介绍下Java 原型设计模式,感兴趣的朋友一起看看吧
    2022-04-04
  • Java查看本机端口是否被占用源码

    Java查看本机端口是否被占用源码

    这篇文章主要介绍了Java查看本机端口是否被占用的主要原理,并结合具体实例给出了操作方法,需要的朋友可以参考下
    2017-09-09
  • 判断以逗号分隔的字符串中是否包含某个数的实例

    判断以逗号分隔的字符串中是否包含某个数的实例

    下面小编就为大家带来一篇判断以逗号分隔的字符串中是否包含某个数的实例。小编觉得挺不错的,现在就分享给大家,也给大家做个参考。一起跟随小编过来看看吧
    2016-11-11
  • 解决spring-boot使用logback的大坑

    解决spring-boot使用logback的大坑

    这篇文章主要介绍了解决spring-boot使用logback的大坑,具有很好的参考价值,希望对大家有所帮助。如有错误或未考虑完全的地方,望不吝赐教
    2021-07-07
  • JavaWeb Servlet实现文件上传与下载功能实例

    JavaWeb Servlet实现文件上传与下载功能实例

    因自己负责的项目中需要实现文件上传,所以下面下面这篇文章主要给大家介绍了关于JavaWeb Servlet实现文件上传与下载功能的相关资料,需要的朋友可以参考下
    2022-04-04
  • JAVA异常处理捕获与抛出原理解析

    JAVA异常处理捕获与抛出原理解析

    这篇文章主要介绍了JAVA异常处理捕获与抛出原理解析,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2020-09-09
  • Maven+Tomcat8 实现自动化部署的方法

    Maven+Tomcat8 实现自动化部署的方法

    本篇文章主要介绍了Maven+Tomcat8 实现自动化部署的方法,小编觉得挺不错的,现在分享给大家,也给大家做个参考。一起跟随小编过来看看吧
    2017-10-10
  • Java中解压缩文件的方法详解(通用)

    Java中解压缩文件的方法详解(通用)

    在软件开发和数据处理领域,文件的解压缩和压缩是常见的任务,下面这篇文章主要给大家介绍了关于Java中解压缩文件的相关资料,文中通过代码介绍的非常详细,需要的朋友可以参考下
    2024-06-06

最新评论