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

概要

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

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作为第一个参数进行传入