spring mybaits多数据源动态切换

场景说明

日常开发中,连接多个数据库是一个很常见的需求,我们的系统是基于spring boot+mybatis进行数据库的操作,网上常见的思路是基于不同的数据库创建不同的bean,大概的实现方式如下:

package com.joylee.fd.crontab.config;import org.apache.ibatis.session.SqlSessionFactory;import org.mybatis.spring.SqlSessionFactoryBean;import org.mybatis.spring.SqlSessionTemplate;import org.mybatis.spring.annotation.MapperScan;import org.springframework.beans.factory.annotation.Qualifier;import org.springframework.boot.autoconfigure.jdbc.DataSourceBuilder;import org.springframework.boot.context.properties.ConfigurationProperties;import org.springframework.context.annotation.Bean;import org.springframework.context.annotation.Configuration;import org.springframework.context.annotation.Primary;import org.springframework.core.io.support.PathMatchingResourcePatternResolver;import org.springframework.jdbc.datasource.DataSourceTransactionManager;import javax.sql.DataSource;@Configuration@MapperScan(basePackages = "com.joylee.fd.crontab.mapper.distribution", sqlSessionTemplateRef  = "distributionSqlSessionTemplate")public class DistributionDBConfiguration {        @Bean(name = "distributionDataSource")        @ConfigurationProperties(prefix = "spring.datasource")        @Primary
        public DataSource distributionDataSource() {            return DataSourceBuilder.create().build();
        }        @Bean(name = "distributionSqlSessionFactory")        @Primary
        public SqlSessionFactory distributionSqlSessionFactory(@Qualifier("distributionDataSource") DataSource dataSource) throws Exception {
            SqlSessionFactoryBean bean = new SqlSessionFactoryBean();
            bean.setDataSource(dataSource);
            bean.setMapperLocations(new PathMatchingResourcePatternResolver().getResources("classpath*:com/joylee/distribution/mapper/*.xml"));            return bean.getObject();
        }        @Bean(name = "distributionDataSourceTransactionManager")        @Primary
        public DataSourceTransactionManager distributionTransactionManager(@Qualifier("distributionDataSource") DataSource dataSource) {            return new DataSourceTransactionManager(dataSource);
        }        @Bean(name = "distributionSqlSessionTemplate")        @Primary
        public SqlSessionTemplate distributionSqlSessionTemplate(@Qualifier("distributionSqlSessionFactory") SqlSessionFactory sqlSessionFactory) throws Exception {            return new SqlSessionTemplate(sqlSessionFactory);
        }

}

类似方式,创建多个Configuration以及多个Bean,因为设为@Configuration,所以在启动的时候Bean都会创建,然后使用的时候只需要将不同的数据库mapper放在设置好的classpath(第35行代码),默认走@Primary中的数据库连接,如果需要走其他的,只要在数据库DAO中加上@@qualifier("bean名")即可。

此方案网上很多实现,这里就是细说了。有兴趣大家可以去搜索一下spring boot mybatis多数据库 解决方案。

此方案适用于数据库数量固定这样的需求,如果数据库是动态实时修改的,那么该怎么处理呢,或者数据库数量很多而且一直在扩充(如分库场景),这样处理显然不行,那么我们应该怎么处理呢?这里先说说解决思路。

基于Mybatis实现思路

为了了解mybatis,这里先简单介绍下mybatis的几个和数据库连接相关的核心类。

说明

SqlSessionFactory

SqlSession的工厂,负责创建SqlSession

SqlSession

mybatis的核心api,负责和数据库交互的回话,该类的方法负责执行数据库的操作

Configuration

mybaits的配置类

Environment

数据库环境类,主要是配置事务和数据库连接

MappedStatement

这两个类负责管理具体需要执行的内和方法

*Handler

主要是基于执行的方法输入和输出参数类型转换处理

mybatis的实现代码结构还是比较容易理解的,我们这里重点管理数据库连接的切换,所有我们重点关注的主要是:SqlSessionFactory、SqlSession、Configuration、Environment,当然还有一个datasouce类,这个类是JDBC的类,用来管理数据库连接。

我最开始的思路就是就是绕过SqlSessionFactory Bean创建,因为Bean的生命周期是在程序启动的时候执行的,看起来是无法改变的。所以我考虑了如下的方式:

针对每一个数据库连接,在使用的时候去创建一个新的SqlSessionFactory,通过mybatis的api new SqlSessionFactory().build()来创建多个SqlSessionFactory,然后根据SqlSessionFactory来创建SqlSession,后面思考了一下,这是一个非常严重的错误,SqlSessionFactory本来就是用来进行SqlSession管理的,显然创建多个SqlSessionFactory是非常不合适的,SqlSessionFactory最好是单例,SqlSessionFactory可以根据不同的数据库创建不同的SqlSession。

基于java mybatis实现,大概的步骤就是:

//代码来自mybaits官网DataSource dataSource = BaseDataTest.createBlogDataSource();
TransactionFactory transactionFactory = new JdbcTransactionFactory();

Environment environment = new Environment("development", transactionFactory, dataSource);

Configuration configuration = new Configuration(environment);
configuration.setLazyLoadingEnabled(true);
configuration.setEnhancementEnabled(true);
configuration.getTypeAliasRegistry().registerAlias(Blog.class);
configuration.getTypeAliasRegistry().registerAlias(Post.class);
configuration.getTypeAliasRegistry().registerAlias(Author.class);
configuration.addMapper(BoundBlogMapper.class);
configuration.addMapper(BoundAuthorMapper.class);

SqlSessionFactoryBuilder builder = new SqlSessionFactoryBuilder();
SqlSessionFactory factory = builder.build(configuration);

基于spring mybatis 实现思路

现在的java项目几乎都是基于spring进行开发,对于这样的需求,肯定是可以以spring的方式进行解决的,我陷入了一个误区,一直收到bean在创建后很难修改这个思路的影响,放弃spring bean管理的方式,但是其实这也是一个错误的方向。

spring bean的确是程序启动的时候就完成了bean的创建,但是每个bean本身是有提供很多方法和属性的,其实bean的很多属性是可以修改的,SqlSessionFactorybean肯定会有这样的属性,果然,我们只要获取到程序启动时创建的bean,然后修改属性的值就可以了。

 /**
     * @param corpDatabaseBO 数据库实体
     * @return SqlSessionFactory
     * @throws PropertyVetoException
     */
    public  SqlSessionFactory changeSqlSessionFactory(CorpDatabaseBO corpDatabaseBO) throws Exception {        //获取当前SqlSessionFactory bean
        
        SqlSessionFactory bean = SpringUtils.getBean(SqlSessionFactory.class);        
        if(bean==null){            throw new NullPointerException("default SqlSessionFactory bean is not created");
            
        }
        
        logger.info(String.format("当前的sqlsessionfactory为:%s",bean.getConfiguration().getEnvironment().getDataSource().getConnection().getCatalog()));        		//因为业务需要,我用的是sqlserver
        SQLServerDataSource sqlServerDataSource = new SQLServerDataSource();
        sqlServerDataSource.setServerName(corpDatabaseBO.getUrl());
        sqlServerDataSource.setDatabaseName(corpDatabaseBO.getDatabasename());
        sqlServerDataSource.setUser(corpDatabaseBO.getUsername());
        sqlServerDataSource.setPassword(corpDatabaseBO.getPassword());
        sqlServerDataSource.setPortNumber(corpDatabaseBO.getPort());        //数据库连接池用的是Hikari
        HikariDataSource hikariDataSource = new HikariDataSource();
        hikariDataSource.setDataSource(sqlServerDataSource);

        TransactionFactory transactionFactory =  new JdbcTransactionFactory();
        Environment environment = new Environment(corpDatabaseBO.getDatabasename(), transactionFactory, hikariDataSource);//        修改environment,这样就可以修改数据库地址了。
        bean.getConfiguration().setEnvironment(environment);        return bean;
    }

在第30行:

bean.getConfiguration().setEnvironment(environment);

通过bean获取configuration,然后再设置Environment 即可修改数据库连接。每次需要执行数据库切换的时候,只要重新调用changeSqlSessionFactory方法即可。

总结

  • 开发中遇到问题,在自己没有很好的解决思路的情况下,可以去网上查找相关的资料,但是有时候,因为自己业务的特殊性,其实网上是没有很好的解决方案的,甚至网上的一些解决方案并非一个很好的解决方案,我们参考别人方案的时候,一定要弄清楚代码每一层的意思,是否是自己真正需要的,而不是一味的copy。
  • 当网上也没有解决方案的时候,建议多花点时间去阅读源码,因为这个功能,虽然不至于阅读了mybatis所有的源码,但是对mybatis整体的结构有了一个清晰的认识,大部分源码也有阅读,这算是此次最大的收获。

原文发布于微信公众号 - IT米粉(itmifen)

原文发表时间:2018-05-14

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏程序猿DD

Spring Cloud构建微服务架构:Hystrix监控面板【Dalston版】

前言 在上一篇《服务容错保护(hystrix断路器)》的介绍中,我们提到断路器是根据一段时间窗内的请求情况来判断并操作断路器的打开和关闭状态的。而这些请求情况的...

2197
来自专栏技术博文

Linux下ps命令详解

linux上进程有5种状态:  1. 运行(正在运行或在运行队列中等待)  2. 中断(休眠中, 受阻, 在等待某个条件的形成或接受到信号)  3. 不可中断(...

4024
来自专栏蓝天

Hive 1.2.1&Spark&Sqoop安装指南

本文的安装参照《Hive 0.12.0安装指南》,内容来源于官方的:GettingStarted,将Hive 1.2.1安装在Hadoop 2.7.1上。本...

2441
来自专栏JavaEE

springboot快速入门前言:一、springboot简介:二、springboot常用知识点:总结:

正所谓,天下武功,唯快不破,在当今生活节奏越来越快的时代,我们也要讲求效率,也要追求一个快字(不过有些方面还是不能快的,不要当快男哦)。springboot就是...

1071
来自专栏小二的折腾日记

服务器-Nginx-一个简单的例子

直接贴一个几经磨难的简单的例子吧,虽然简单,但是有些不理解的地方,就出了很多错,但是因为这些错误,在对nginx理解稍微深刻了那么一点点的同时,对Linux的操...

1241
来自专栏IT笔记

springMvc4+hibernate4的一个奇葩的网上众说纷纭的错误

springMvc4+hibernate4项目运行报错如下: org.hibernate.HibernateException: Could not obtai...

3117
来自专栏散尽浮华

Linux下部署Samba服务环境的操作记录

关于Linux和Windows系统之间的文件传输,很多人选择使用FTP,相对较安全,但是有时还是会出现一些问题,比如上传文件时,文件名莫名出现乱码,文件大小改变...

4578
来自专栏加米谷大数据

技术干货 | hive安装部署

解压hive压缩包 apache-hive-2.1.0-bin.tar.gz(官网下载) 配置HADOOP_HOME环境变量 配置HIVE_HOME环境变量 在...

3368
来自专栏owent

注册表常用键值意义

[HKEY_CURRENT_USER\Software\Policies\Microsoft\Internet Explorer\Control Panel]

1622
来自专栏mantou大数据

【大数据安全】Apache Kylin 安全配置(Kerberos)

本文首先会简单介绍Kylin的安装配置,然后介绍启用Kerberos的CDH集群中如何部署及使用Kylin。

3083

扫码关注云+社区

领取腾讯云代金券