Springboot + Mybatis 多数据源配置

清疚 2023-06-02 05:26 136阅读 0赞

多数据源，很多项目都用得到，比较实用。在 [DEMO: springboot 与 freemarker 集成][DEMO_ springboot _ freemarker] 基础上进行修改。

修改后的项目结构大致这样：

![1078922-20170614153111290-1121907118.png][]

### 1、src/main/resources/application.properties 中配置好多个数据源 ###

![ContractedBlock.gif][] ![ExpandedBlockStart.gif][]

datasource.jdbc.driverClassName = com.mysql.jdbc.Driver
    datasource.jdbc.url = jdbc:mysql://127.0.0.1:3306/test
    datasource.jdbc.username = root
    datasource.jdbc.password = test
    # function
    datasource.jdbc.url_stg = jdbc:mysql://10.199.xx.35:3306/test
    datasource.jdbc.username_stg = root
    datasource.jdbc.password_stg = test
    
    
    # mybatis #
    mybatis.typeAliasesPackage=demo.springboot.model 
    mybatis.mapperLocations=classpath:/mybatis/mapper/*.xml

### 2、定义一个枚举类型 DatabaseType.java，表示不同的环境 ###

![ContractedBlock.gif][] ![ExpandedBlockStart.gif][]

package demo.springboot.enums;
    
    /**
     * @author meng.geng
     * function  功能环境
     * regression 回归环境
     * 
     * */
    public enum DatabaseType {
        function("function", "1"), 
        regression("regression", "2");
    
        DatabaseType(String name, String value){
            this.name = name;
            this.value = value;
        }
        
        private String name;
        private String value;
        
        public String getName(){
            return name;
        }
        
        public String getValue(){
            return value;
        }    
    }

### 3、定义一个 DatabaseContextHolder.java， 保存一个线程安全的DatabaseType容器 ###

![ContractedBlock.gif][] ![ExpandedBlockStart.gif][]

package demo.springboot.conf;
    
    import demo.springboot.enums.DatabaseType;
    
    /**
     * @author danny.yao
     * 作用：保存一个线程安全的DatabaseType容器
     * 
     **/
    public class DatabaseContextHolder {
        private static final ThreadLocal<DatabaseType> contextHolder = new ThreadLocal<>();
        
        public static void setDatabaseType(DatabaseType type){
            contextHolder.set(type);
        }
        
        public static DatabaseType getDatabaseType(){
            return contextHolder.get();
        }    
    }

### 4、定义动态数据源获取的方法 DynamicDataSource.java，集成 AbstractRoutingDataSource ###

![ContractedBlock.gif][] ![ExpandedBlockStart.gif][]

package demo.springboot.conf;
    
    import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
    
    /**
     * @author danny.yao
     * 动态数据源（需要继承AbstractRoutingDataSource）
     * 作用：使用DatabaseContextHolder获取当前线程的DatabaseType
     * 
     * */
    public class DynamicDataSource extends AbstractRoutingDataSource {
        
         protected Object determineCurrentLookupKey() {
                return DatabaseContextHolder.getDatabaseType();
            }    
    }

### 5、mybatis多数据源的配置 MybatisConfig.java ###

![ContractedBlock.gif][] ![ExpandedBlockStart.gif][]

package demo.springboot.conf;
    
    import java.util.HashMap;
    import java.util.Map;
    
    import javax.sql.DataSource;
    
    import org.apache.ibatis.session.SqlSessionFactory;
    import org.mybatis.spring.SqlSessionFactoryBean;
    import org.mybatis.spring.annotation.MapperScan;
    import org.springframework.beans.factory.annotation.Autowired;
    import org.springframework.beans.factory.annotation.Qualifier;
    import org.springframework.beans.factory.annotation.Value;
    import org.springframework.context.annotation.Bean;
    import org.springframework.context.annotation.Configuration;
    import org.springframework.context.annotation.Primary;
    import org.springframework.core.env.Environment;
    import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
    
    import com.alibaba.druid.pool.DruidDataSource;
    
    import demo.springboot.enums.DatabaseType;
    
    /**
     * 
     * @author danny.yao
     * springboot集成mybatis基本入口
     * 1、创建数据源
     * 2、创建SqlSessionFactory
     */
    @Configuration
    @MapperScan(basePackages="demo.springboot.mapper", sqlSessionFactoryRef="sessionFactory")
    public class MybatisConfig {
        
        @Autowired
        Environment environment;
        
        @Value("${datasource.jdbc.driverClassName}")
        private String dbDriver;
        
        @Value("${datasource.jdbc.url}")
        private String dbUrl;
        
        @Value("${datasource.jdbc.username}")
        private String dbUsername;
        
        @Value("${datasource.jdbc.password}")
        private String dbPassword;
        
        @Value("${datasource.jdbc.url_stg}")
        private String dbUrl_stg;
        
        @Value("${datasource.jdbc.username_stg}")
        private String dbUsername_stg;
        
        @Value("${datasource.jdbc.password_stg}")
        private String dbPassword_stg;
        
        
        /**
         * 创建 local环境 dataSource
         * @throws Exception 
         */
        @Bean(name="dataSourceFunctional")
        public DataSource dataSourceLocal() throws Exception{            
            DruidDataSource dataSource = new DruidDataSource();
            dataSource.setDriverClassName(dbDriver);
            dataSource.setUrl(dbUrl);
            dataSource.setUsername(dbUsername);
            dataSource.setPassword(dbPassword);
            
            return dataSource;
        }
        
        /**
         * 创建 回归环境 dataSource
         * @throws Exception 
         */
        @Bean(name="dataSourceRegression")
        public DataSource dataSourceStaging() throws Exception{
            DruidDataSource dataSource = new DruidDataSource();
            dataSource.setDriverClassName(dbDriver);
            dataSource.setUrl(dbUrl_stg);
            dataSource.setUsername(dbUsername_stg);
            dataSource.setPassword(dbPassword_stg);
            
            return dataSource;
        }
        
        /**
         * 1、创建动态数据源
         * @throws Exception 
         * @Primary该注解表示在同一个接口有多个类可以注入的时候，默认选择哪个，而不是让@Autowired报错
         */
        @Bean(name="dynamicDataSource")
        @Primary
        public DynamicDataSource DataSource(@Qualifier("dataSourceFunctional") DataSource dataSourceFunctional, 
                                            @Qualifier("dataSourceRegression") DataSource dataSourceRegression){
            Map<Object, Object> targetDataSource = new HashMap<>();
            targetDataSource.put(DatabaseType.function, dataSourceFunctional);
            targetDataSource.put(DatabaseType.regression, dataSourceRegression);
            DynamicDataSource dataSource = new DynamicDataSource();
            dataSource.setTargetDataSources(targetDataSource);
            dataSource.setDefaultTargetDataSource(dataSourceFunctional);
            
            return dataSource;
        }
        
        /**
         * 2、根据数据源创建SqlSessionFactory
         * @throws Exception 
         */
        @Bean(name="sessionFactory")
        public SqlSessionFactory sessionFactory(@Qualifier("dynamicDataSource")DynamicDataSource dataSource) throws Exception{
            SqlSessionFactoryBean sessionFactoryBean = new SqlSessionFactoryBean();
            sessionFactoryBean.setDataSource(dataSource);
            PathMatchingResourcePatternResolver resolver = new PathMatchingResourcePatternResolver();
            sessionFactoryBean.setMapperLocations(resolver.getResources(environment.getProperty("mybatis.mapperLocations")));    //*Mapper.xml位置
            return sessionFactoryBean.getObject();
        }
    }

### 6、修改服务类 StudentService.java：添加环境切换方法setDataSourceByEnvironment，同时在获取数据的方法中切换数据源 ###

![ContractedBlock.gif][] ![ExpandedBlockStart.gif][]

package demo.springboot.service;
    
    import java.util.ArrayList;
    import java.util.List;
    
    import org.springframework.beans.factory.annotation.Autowired;
    import org.springframework.stereotype.Service;
    
    import demo.springboot.conf.DatabaseContextHolder;
    import demo.springboot.enums.DatabaseType;
    import demo.springboot.mapper.StudentMapper;
    import demo.springboot.model.Student;
    
    @Service
    public class StudentService {
        
        @Autowired
        private StudentMapper studentMapper;
        
        public void setDataSourceByEnvironment(String environment){
            if (environment.equals(DatabaseType.function.getValue())){
                DatabaseContextHolder.setDatabaseType(DatabaseType.function);
            }
            if (environment.equals(DatabaseType.regression.getValue())){
                DatabaseContextHolder.setDatabaseType(DatabaseType.regression);
            }
        }
        
        public List<Student> listStudents(String environment){
            setDataSourceByEnvironment(environment);
            List<Student> students = new ArrayList<>();
            students = studentMapper.listStudents();
            return students;
        }
    
    }

### 7、修改controller类 StudentController.java ： 请求需要带上环境参数 @RequestParam(value="environment", required=true)String environment ###

![ContractedBlock.gif][] ![ExpandedBlockStart.gif][]

package demo.springboot.controller;
    
    import java.util.ArrayList;
    import java.util.List;
    import java.util.Map;
    
    import org.slf4j.Logger;
    import org.slf4j.LoggerFactory;
    import org.springframework.beans.factory.annotation.Autowired;
    import org.springframework.stereotype.Controller;
    import org.springframework.web.bind.annotation.RequestMapping;
    import org.springframework.web.bind.annotation.RequestParam;
    
    import demo.springboot.model.Student;
    import demo.springboot.service.StudentService;
    
    @Controller
    @RequestMapping("/student")
    public class StudentController {
    
        private static final Logger LOGGER =  LoggerFactory.getLogger(StudentController.class);
        
        @Autowired
        private StudentService studentService;
        
        @RequestMapping("/list")
        public String listStudent(@RequestParam(value="environment", required=true)String environment, Map<String, Object> map){
            List<Student> students = new ArrayList<>();
            students = studentService.listStudents(environment);
            
            for (Student student : students){
                LOGGER.info(student.getId() + " - " + student.getName() + " - " + student.getNo() + " - " + student.getSex());
            }
            
            map.put("students", students);
            
            return "student";
        }    
    }

### 8、测试 ###

1）请求环境1的数据：http://localhost:8080/student/list?environment=1

![1078922-20170614154654853-595199117.png][]

2）请求环境2的数据：http://localhost:8080/student/list?environment=2

![1078922-20170614154731275-576572993.png][]

转载于:https://www.cnblogs.com/dannyyao/p/7009264.html

[DEMO_ springboot _ freemarker]: http://www.cnblogs.com/dannyyao/p/6993290.html
[1078922-20170614153111290-1121907118.png]: /images/20230601/35c82e0761b1403695288fa185aec709.png
[ContractedBlock.gif]: https://images.cnblogs.com/OutliningIndicators/ContractedBlock.gif
[ExpandedBlockStart.gif]: /images/20230601/c3fbe362298a4391b8dc36ebde621566.png
[1078922-20170614154654853-595199117.png]: /images/20230601/fc25ce88f87d4b09a65b0478a1951069.png
[1078922-20170614154731275-576572993.png]: /images/20230601/72b3faaf16f4427289a6453dba4329e5.png

发表评论取消回复

表情：

评论列表（有 0 条评论，136人围观）

还没有评论，来说两句吧...

相关阅读

相关 SpringBoot+Druid+Mybatis配置多数据源

[github源码demo][github_demo] 1.application.yml配置，区分个数据源的相关配置 server: po...

末蓝、/ 2024年04月17日 15:07/ 0 赞/ 202 阅读

相关 springboot+mybatis配置多数据源

1、首先你自己实现了单数据源。 2、然后在application.properties中配置第二数据源第一数据源 spring.datasource.db

你的名字/ 2023年10月17日 09:47/ 0 赞/ 153 阅读

相关多数据源配置springboot+Mybatis

多数据源用到的场景很多，所以就写了本篇文章进行演示如何配置多数据源。[项目源码地址点击此处查看][Link 1] 项目依赖springboot2+mybatis+mysql

忘是亡心i/ 2023年07月06日 09:51/ 0 赞/ 136 阅读

相关 Springboot + Mybatis 多数据源配置

多数据源，很多项目都用得到，比较实用。在 [DEMO: springboot 与 freemarker 集成][DEMO_ springboot _ freemarker] 基

清疚/ 2023年06月02日 05:26/ 0 赞/ 137 阅读

相关 Springboot + Mybatis配置多数据源

0、pom依赖 <dependency> <groupId>org.springframework.boot</groupId> <a

今天药忘吃喽~/ 2023年03月13日 15:27/ 0 赞/ 281 阅读

相关 SpringBoot、MyBatis配置多数据源XML方法

转载请注明来源-作者@loongshawn：[http://blog.csdn.net/loongshawn/article/details/52164730][http_bl

短命女/ 2022年07月13日 00:46/ 0 赞/ 796 阅读

相关 springboot配置多数据源 mybatis

application.yml配置 spring.datasource.primary.url=jdbc:mysql://192.168.10.109:3306/de

红太狼/ 2022年06月02日 03:43/ 0 赞/ 425 阅读

相关 SpringBoot配置Mybatis多数据源

1. SpringBoot版本不同的版本其配置方式可能有所区别，我采用的是SpringBoot2.0.4.RELEASE。 2. 项目结构 ![在这里插入图片描

旧城等待，/ 2022年01月29日 16:29/ 0 赞/ 541 阅读

相关 SpringBoot整合mybatis配置多数据源

一、添加数据库配置 spring.datasource.url=jdbc:mysql://localhost:3306/study spring.data

素颜马尾好姑娘i/ 2021年11月08日 18:40/ 0 赞/ 531 阅读

相关 SpringBoot+Druid+MyBatis 多数据源配置

1.工程目录结构 ![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubm

川长思鸟来/ 2021年10月14日 00:26/ 0 赞/ 765 阅读