SpringBoot多数据源切换方案分享

52 阅读3分钟

操作数据一般都是在DAO层进行处理,可以选择直接使用JDBC进行编程。

或者是使用多个DataSource 然后创建多个SessionFactory,在使用Dao层的时候通过不同的SessionFactory进行处理,不过这样的入侵性比较明显,一般的情况下我们都是使用继承HibernateSupportDao进行封装了的处理,如果多个SessionFactory这样处理就是比较的麻烦了,修改的地方估计也是蛮多的。

最后一个,也就是使用AbstractRoutingDataSource的实现类通过AOP或者手动处理实现动态的使用我们的数据源,这样的入侵性较低,非常好的满足使用的需求。比如我们希望对于读写分离或者其他的数据同步的业务场景。

  1. 单数据源的场景(一般的Web项目工程这样配置进行处理,就已经比较能够满足我们的业务需求)

2.多数据源多SessionFactory这样的场景,估计作为刚刚开始想象想处理在使用框架的情况下处理业务,配置多个SessionFactory,然后在Dao层中对于特定的请求,通过特定的SessionFactory即可处理实现这样的业务需求,不过这样的处理带来了很多的不便之处,所有很多情况下我们宁愿直接使用封装的JDBC编程,或者使用Mybatis处理这样的业务场景。

 

使用AbstractRoutingDataSource 的实现类,进行灵活的切换,可以通过AOP或者手动编程设置当前的DataSource,不用修改我们编写的对于继承HibernateSupportDao的实现类的修改,这样的编写方式比较好,至于其中的实现原理,让我细细到来。我们想看看如何去应用,实现原理慢慢的说!

编写AbstractRoutingDataSource的实现类,HandlerDataSource就是提供给我们动态选择数据源的数据的信息,我们这里编写一个根据当前线程来选择数据源,然后通过AOP拦截特定的注解,设置当前的数据源信息,也可以手动的设置当前的数据源,在编程的类中。

/*
** @author *guoqi7
** **@Description: *java 类作用描述
**** **@CreateDate: ***2021-08-31 20:40
** **@UpdateUser: ***guoqi7
** **@UpdateDate: ***2021-08-31 20:40
** **@UpdateRemark: *修改内容
**** **@Version: ***1.0
*/
*public class DynamicDataSource extends AbstractRoutingDataSource {
private Map<Object, Object> allResolvedDataSources = new HashMap<>();

    @Override
protected Object determineCurrentLookupKey() {
return DataSourceContextUtil.getDataSourceType();
}

    /*
** 添加数据源*
**** @param ***name
****** @param ***dataSource
****/
public void addDataSource(Object name, DataSource dataSource){
if(allResolvedDataSources.containsKey(name))
allResolvedDataSources.remove(name);
allResolvedDataSources.put(name,dataSource);
setTargetDataSources(allResolvedDataSources);
}
/

** 删除数据源

**** @param ***name
****** ***@return
*****/
*public Object deleteDataSource(Object name){
Object rst = null;
if(allResolvedDataSources.containsKey(name))
{
rst = allResolvedDataSources.get(name);
allResolvedDataSources.remove(name);
setTargetDataSources(allResolvedDataSources);
}
return rst;
}

    /*
** 检查数据源是否存在*
**** @param ***name
****** ***@return
*****/
*public Boolean isExistDataSource(Object name){
return allResolvedDataSources.containsKey(name);
}
}

设置动态选择的Datasource,这里的Set方法可以留给AOP调用,或者留给我们的具体的Dao层或者Service层中手动调用,在执行SQL语句之前。

public class DynamicDataSource extends AbstractRoutingDataSource {

    private Map<Object, Object> allResolvedDataSources = new HashMap<>();



    @Override

    protected Object determineCurrentLookupKey() {

        return DataSourceContextUtil.getDataSourceType();

    }



     /**

     * 添加数据源 

 *  @param name

 *  @param dataSource

 */

 public void addDataSource(Object name,DataSource dataSource){

        if(allResolvedDataSources.containsKey(name))

            allResolvedDataSources.remove(name);

        allResolvedDataSources.put(name,dataSource);

        setTargetDataSources(allResolvedDataSources);

    }

     /**

     * 删除数据源 

 *  @param name

 *  @return

 */

 public Object deleteDataSource(Object name){

        Object rst = null;

        if(allResolvedDataSources.containsKey(name))

        {

            rst = allResolvedDataSources.get(name);

            allResolvedDataSources.remove(name);

            setTargetDataSources(allResolvedDataSources);

        }

        return rst;

    }



     /**

     * 检查数据源是否存在 

 *  @param name

 *  @return

 */

 public Boolean isExistDataSource(Object name){

        return allResolvedDataSources.containsKey(name);

    }

}

在项目启动的时候初始化所有的数据源信息,通过DynamicDataSource类以键值对的形式存储

DruidDataSource dataSource = dataSource();

DruidDataSource masterDataSource = masterDataSource();

DruidDataSource clusterDataSource = clusterDataSource();

 //DruidDataSource bizDataSource = bizDataSource();


 try {

dataSource.init();

clusterDataSource.init();

masterDataSource.init();



 //   bizDataSource.init();
 }catch (SQLException sql){

sql.printStackTrace();

}



DynamicDataSource dynamicDataSource = new DynamicDataSource();

dynamicDataSource.addDataSource(DSEnum.DATA_SOURCE_NAME,dataSource);                     dynamicDataSource.addDataSource(DSEnum.MASTER_DATA_SOURCE_NAME,masterDataSource);

dynamicDataSource.addDataSource(DSEnum.CLUSTER_DATA_SOURCE_NAME,clusterDataSource);

DataSourceContextUtil.setDataSourceType(DSEnum.DATA_SOURCE_NAME); // 设置默认的数据源的 Key
 dynamicDataSource.setDefaultTargetDataSource(dataSource); // 设置默认的数据源

代码使用方法,注意的是这里使用过后需要调用清空的方法,让其返回默认的数据源,否则将会带着这个数据源满世界的报错找不到表,找不到列,映射不到类。。。。。。

DataSourceContextUtil.setDataSourceType(DSEnum.CLUSTER_DATA_SOURCE_NAME);

        String sql2="select * from sys_config";

        List<Map<String,Object>>sqlListPage2=commonDao.getSqlListNotPage(sql2.toString(),null);

        DataSourceContextUtil.clearDataSourceType();