【Java】SpringBoot+Mybatis实现多数据源连接
文章目录
- 前言
- 一、导入所需Jar包（pom.xml）
- 二、定义两个数据库的常量
- 三、创建dao以及mapper文件夹
- 四、创建datasource配置类
- 五、定义application文件
- 总结
前言
在项目开发过程中，经常会碰到需要连接多个数据库的情况。或者是多个不同数据库的情况。本篇文章在于实现利用SpringBoot+Mybatis框架进行多数据源连接，数据库使用的是SQL Server。希望能够帮到你！
一、导入所需Jar包（pom.xml）
<!-- Mybatis -->
<dependency>
<groupId>org.mybatis.spring.boot</groupId>
<artifactId>mybatis-spring-boot-starter</artifactId>
<version>2.1.4</version>
</dependency>
<!-- Druid 数据连接池依赖 -->
<dependency>
<groupId>com.alibaba</groupId>
<artifactId>druid-spring-boot-starter</artifactId>
<version>1.2.11</version>
</dependency>
<!--SQL Server-->
<dependency>
<groupId>com.microsoft.sqlserver</groupId>
<artifactId>mssql-jdbc</artifactId>
<scope>runtime</scope>
</dependency>
<!--SpringBoot 添加jdbc支持-->
<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文件夹目录下区分两个数据源
（2）mapper文件夹下也要区分两个数据源
四、创建datasource配置类
建一个文件夹datasource里面新建五个配置类
具体代码如下：
（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时键值对不存在
总结
该贴的代码都贴上去了，其他的可在下方评论区提问，博主会及时提供回复！谢谢关注！