【Java】SpringBoot+Mybatis实现多数据源连接

骑猪看日落 2023-09-30 23:35 1阅读 0赞

#### 文章目录 ####

*  前言
 *  一、导入所需Jar包（pom.xml）
 *  二、定义两个数据库的常量
 *  三、创建dao以及mapper文件夹
 *  四、创建datasource配置类
 *  五、定义application文件
 *  总结

--------------------

## 前言 ##

在项目开发过程中，经常会碰到需要连接多个数据库的情况。或者是多个不同数据库的情况。本篇文章在于实现利用SpringBoot+Mybatis框架进行多数据源连接，数据库使用的是SQL Server。希望能够帮到你！

--------------------

## 一、导入所需Jar包（pom.xml） ##

<dependency>
                <groupId>org.mybatis.spring.boot</groupId>
                <artifactId>mybatis-spring-boot-starter</artifactId>
                <version>2.1.4</version>
            </dependency>
            
            <dependency>
                <groupId>com.alibaba</groupId>
                <artifactId>druid-spring-boot-starter</artifactId>
                <version>1.2.11</version>
            </dependency>
            
            <dependency>
                <groupId>com.microsoft.sqlserver</groupId>
                <artifactId>mssql-jdbc</artifactId>
                <scope>runtime</scope>
            </dependency>
            
            <dependency>
                <groupId>org.springframework.boot</groupId>
                <artifactId>spring-boot-starter-jdbc</artifactId>
            </dependency>

## 二、定义两个数据库的常量 ##

代码如下（示例）：

/**
     * @author 孤巷.
     * @description
     * @date 2022/8/9 15:58
     */
    public class DataBaseNameConstant {
        
    
        /**
         * HIS数据库
         */
        public static final String HIS = "his";
    
        /**
         * 预约挂号数据库
         */
        public static final String APPOINTMENT = "appointment";
    }

## 三、创建dao以及mapper文件夹 ##

（1）dao文件夹目录下区分两个数据源  
![在这里插入图片描述][3c9f4c2bebe04192a55349eae9f8e819.png]  
（2）mapper文件夹下也要区分两个数据源

![在这里插入图片描述][38259fabd24a49ab95259752cc71397a.png]

## 四、创建datasource配置类 ##

建一个文件夹datasource里面新建五个配置类  
![在这里插入图片描述][9324b2fdeefb428485b3472ed94e153a.png]  
具体代码如下：  
（1）DataSource

import java.lang.annotation.*;
    
    /**
     * mybatis自定义注解,用来声明数据源
     *
     * @author wcs
     */
    @Target({
        ElementType.TYPE})
    @Retention(RetentionPolicy.RUNTIME)
    @Documented
    public @interface DataSource {
        
        String value();
    }

（2）DataSourceAspect

import lombok.extern.slf4j.Slf4j;
    import org.aspectj.lang.JoinPoint;
    import org.aspectj.lang.annotation.Aspect;
    import org.aspectj.lang.annotation.Before;
    import org.aspectj.lang.annotation.Pointcut;
    import org.springframework.stereotype.Component;
    
    /**
     * mybatis切面，用来切换数据源
     *
     * @author wcs
     */
    @Aspect
    @Component
    @Slf4j
    public class DataSourceAspect {
        
    
        @Pointcut("execution(* com.example.web.dao.*.*.*(..))")
        public void dataSourcePointCut() {
        
    
        }
    
        @Before("dataSourcePointCut()")
        public void before(JoinPoint point) {
        
            Object target = point.getTarget();
            DataSource dataSource = target.getClass().getInterfaces()[0].getAnnotation(DataSource.class);
            if (dataSource != null) {
        
                String key = dataSource.value();
                log.info("当前运行的数据源为 == {}",key);
                DynamicDataSourceHolder.setDataSource(key);
            }
        }
    }

（3）DatasourceConfig

import com.alibaba.druid.pool.DruidDataSource;
    import com.example.web.constant.DataBaseNameConstant;
    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 javax.sql.DataSource;
    import java.util.HashMap;
    import java.util.Map;
    
    /**
     * @author wcs
     * @description 多数据源配置
     * @date 2021/3/13 11:44
     */
    @Configuration
    public class DatasourceConfig {
        
    
        @Bean(name = DataBaseNameConstant.HIS)
        @ConfigurationProperties(prefix = "spring.datasource.druid.his")
        public DataSource hisDruid() {
        
            return new DruidDataSource();
        }
    
        @Bean(name = DataBaseNameConstant.APPOINTMENT)
        @ConfigurationProperties(prefix = "spring.datasource.druid.appointemnt")
        public DataSource appointemntDruid() {
        
            return new DruidDataSource();
        }
    
    
    
        @Bean
        @Primary
        public DynamicDataSource dataSource() {
        
            Map<String, DataSource> targetDataSources = new HashMap<>(4);
            targetDataSources.put(DataBaseNameConstant.HIS, hisDruid());
            targetDataSources.put(DataBaseNameConstant.APPOINTMENT, appointemntDruid());
            return new DynamicDataSource(targetDataSources.get(DataBaseNameConstant.HIS), targetDataSources);
        }
    }

（4）DynamicDataSource

import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
    
    import javax.sql.DataSource;
    import java.util.HashMap;
    import java.util.Map;
    
    /**
     * mybatis自定义数据源
     *
     * @author wcs
     */
    public class DynamicDataSource extends AbstractRoutingDataSource {
        
    
        public DynamicDataSource(DataSource defaultTargetDataSource, Map<String, DataSource> targetDataSources) {
        
            super.setDefaultTargetDataSource(defaultTargetDataSource);
            super.setTargetDataSources(new HashMap<>(targetDataSources));
            super.afterPropertiesSet();
        }
    
        @Override
        protected Object determineCurrentLookupKey() {
        
            return DynamicDataSourceHolder.getDataSouce();
        }
    
    }

（5）DynamicDataSourceHolder

/**
     * mybatis自定义数据源管理器
     *
     * @author wcs
     */
    public class DynamicDataSourceHolder {
        
        public static final ThreadLocal<String> holder = new ThreadLocal<String>();
    
        public static void setDataSource(String name) {
        
            holder.set(name);
        }
    
        public static String getDataSouce() {
        
            return holder.get();
        }
    }

## 五、定义application文件 ##

`提示：本文中还会用到lombok以及Druid，这两个jar包在网上都能找到`  
（1）application.yml文件内容  
代码如下：

#开发环境配置
    server:
      port: 9803
    spring:
      servlet:
        multipart:
          max-file-size: 10MB
          max-request-size: 10MB
      datasource:
        druid:
          his:
            url:  jdbc:sqlserver://127.0.0.1:1433;databaseName=test;
            driver-class-name: com.microsoft.sqlserver.jdbc.SQLServerDriver
            username: test
            password: test
            initial-size: 5 #初始化时建立物理连接的个数。
            max-active: 20  #最大连接池数量
            min-idle: 5     #最小连接池数量
            max-wait: 3000  #获取连接时最大等待时间，单位毫秒
            pool-prepared-statements: false #是否缓存preparedStatement，也就是PSCache,PSCache对支持游标的数据库性能提升巨大，比如说oracle,在mysql下建议关闭。
            #要启用PSCache，必须配置大于0，当大于0时，poolPreparedStatements自动触发修改为true。在Druid中，不会存在Oracle下PSCache占用内存过多的问题，可以把这个数值配置大一些，比如说100
            max-open-prepared-statements: 100
            time-between-eviction-runs-millis: 60000 #配置检测可以关闭的空闲连接间隔时间
            min-evictable-idle-time-millis: 300000 # 配置连接在池中的最小生存时间
            max-evictable-idle-time-millis: 400000
            filters: stat,slf4j #监控统计的stat,以及防sql注入的wall
            web-stat-filter:
              enabled: true
            validation-query: SELECT 1
            # StatViewServlet配置，说明请参考Druid Wiki，配置_StatViewServlet配置
            #是否启用StatViewServlet（监控页面）默认值为false（考虑到安全问题默认并未启动，如需启用建议设置密码或白名单以保障安全）
            stat-view-servlet:
              enabled: true       #是否启用StatViewServlet默认值true
              url-pattern: /druid/* #访问路径为/druid时，跳转到StatViewServlet
              reset-enable: true #是否能够重置数据
              login-username: whxx
              login-password: whxx666.
              allow: localhost
            test-on-borrow: false
            test-on-return: false
            test-while-idle: true
            max-pool-prepared-statement-per-connection-size: 20
            filter:
              stat:
                log-slow-sql: true
                slow-sql-millis: 2000
              slf4j:
                enabled: true
                result-set-log-enabled: false
                statement-create-after-log-enabled: false
                statement-close-after-log-enabled: false
                result-set-open-after-log-enabled: false
                result-set-close-after-log-enabled: false
          appointemnt:
            url: jdbc:sqlserver://127.0.0.1:1433;databaseName=test1;
            driver-class-name: com.microsoft.sqlserver.jdbc.SQLServerDriver
            username: test1
            password: test1
            initial-size: 5 #初始化时建立物理连接的个数。
            max-active: 20  #最大连接池数量
            min-idle: 5     #最小连接池数量
            max-wait: 3000  #获取连接时最大等待时间，单位毫秒
            pool-prepared-statements: false #是否缓存preparedStatement，也就是PSCache,PSCache对支持游标的数据库性能提升巨大，比如说oracle,在mysql下建议关闭。
            #要启用PSCache，必须配置大于0，当大于0时，poolPreparedStatements自动触发修改为true。在Druid中，不会存在Oracle下PSCache占用内存过多的问题，可以把这个数值配置大一些，比如说100
            max-open-prepared-statements: 100
            time-between-eviction-runs-millis: 60000 #配置检测可以关闭的空闲连接间隔时间
            min-evictable-idle-time-millis: 300000 # 配置连接在池中的最小生存时间
            max-evictable-idle-time-millis: 400000
            filters: stat,slf4j #监控统计的stat,以及防sql注入的wall
            web-stat-filter:
              enabled: true
            validation-query: SELECT 1
            # StatViewServlet配置，说明请参考Druid Wiki，配置_StatViewServlet配置
            #是否启用StatViewServlet（监控页面）默认值为false（考虑到安全问题默认并未启动，如需启用建议设置密码或白名单以保障安全）
            stat-view-servlet:
              enabled: true       #是否启用StatViewServlet默认值true
              url-pattern: /druid/* #访问路径为/druid时，跳转到StatViewServlet
              reset-enable: true #是否能够重置数据
              login-username: whxx
              login-password: whxx666.
              allow: localhost
            test-on-borrow: false
            test-on-return: false
            test-while-idle: true
            max-pool-prepared-statement-per-connection-size: 20
            filter:
              stat:
                log-slow-sql: true
                slow-sql-millis: 2000
              slf4j:
                enabled: true
                result-set-log-enabled: false
                statement-create-after-log-enabled: false
                statement-close-after-log-enabled: false
                result-set-open-after-log-enabled: false
                result-set-close-after-log-enabled: false
    mybatis:
      mapper-locations: classpath:mapper/*/*.xml #xml文件位置
      type-aliases-package: com.example.web.bean
      configuration:
        map-underscore-to-camel-case: true
        call-setters-on-nulls: true #处理数据为null时键值对不存在

--------------------

## 总结 ##

该贴的代码都贴上去了，其他的可在下方评论区提问，博主会及时提供回复！谢谢关注！

[3c9f4c2bebe04192a55349eae9f8e819.png]: https://img-blog.csdnimg.cn/3c9f4c2bebe04192a55349eae9f8e819.png
[38259fabd24a49ab95259752cc71397a.png]: https://img-blog.csdnimg.cn/38259fabd24a49ab95259752cc71397a.png
[9324b2fdeefb428485b3472ed94e153a.png]: https://img-blog.csdnimg.cn/9324b2fdeefb428485b3472ed94e153a.png