当前位置：首页 > news >正文

springboot提升-多数据源配置

news 来源：原创 2024/9/20 4:42:45

文章目录

- - 1. 添加依赖
  - 2. 配置数据源
  - - 示例配置：
  - 3. 创建数据源 Bean
  - 4. 创建动态数据源
  - 5. 配置 MyBatis SqlSessionFactory
  - 6. 在业务代码中使用
  - 注意事项

在 Spring Boot 中配置 MyBatis 以支持多数据源涉及几个关键步骤，包括配置数据源、集成 MyBatis 以及动态数据源切换机制。以下是详细的配置步骤：

1. 添加依赖

确保你的项目中包含必要的依赖，例如 MySQL 驱动、Spring Boot Starter JDBC 或者 MyBatis 的相关依赖。

<!-- MySQL 驱动 -->
<dependency><groupId>mysql</groupId><artifactId>mysql-connector-java</artifactId>
</dependency><!-- Spring Boot JDBC Starter -->
<dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter-jdbc</artifactId>
</dependency><!-- MyBatis Starter -->
<dependency><groupId>org.mybatis.spring.boot</groupId><artifactId>mybatis-spring-boot-starter</artifactId><version>2.1.1</version>
</dependency>

2. 配置数据源

在 application.properties 或 application.yml 文件中添加每个数据源的配置信息。

示例配置：

# 数据源1配置
spring.datasource.db1.url=jdbc:mysql://host1:3306/db1
spring.datasource.db1.username=user1
spring.datasource.db1.password=password1
spring.datasource.db1.driver-class-name=com.mysql.cj.jdbc.Driver# 数据源2配置
spring.datasource.db2.url=jdbc:mysql://host2:3306/db2
spring.datasource.db2.username=user2
spring.datasource.db2.password=password2
spring.datasource.db2.driver-class-name=com.mysql.cj.jdbc.Driver

3. 创建数据源 Bean

在 Spring Boot 的配置类中，为每个数据源创建对应的 DataSource Bean。

import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.boot.jdbc.DataSourceBuilder;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import javax.sql.DataSource;@Configuration
public class DataSourceConfig {@Bean(name = "db1")@ConfigurationProperties(prefix = "spring.datasource.db1")public DataSource db1DataSource() {return DataSourceBuilder.create().build();}@Bean(name = "db2")@ConfigurationProperties(prefix = "spring.datasource.db2")public DataSource db2DataSource() {return DataSourceBuilder.create().build();}
}

4. 创建动态数据源

创建一个 DynamicDataSource 类来动态地改变当前线程绑定的数据源。

线程池切换范围在一个线程内，如果涉及多个线程，那么需要在线程范围内进行线程池切换

import javax.sql.DataSource;
import java.util.HashMap;
import java.util.Map;public class DynamicDataSource extends AbstractRoutingDataSource {private static final ThreadLocal<String> CONTEXT_HOLDER = new ThreadLocal<>();public static void setDataSource(String dataSourceName) {CONTEXT_HOLDER.set(dataSourceName);}public static void clearDataSource() {CONTEXT_HOLDER.remove();}@Overrideprotected Object determineCurrentLookupKey() {return CONTEXT_HOLDER.get();}public void afterPropertiesSet() {super.afterPropertiesSet();Map<Object, Object> targetDataSources = new HashMap<>();targetDataSources.put("db1", db1DataSource);targetDataSources.put("db2", db2DataSource);setTargetDataSources(targetDataSources);}
}

5. 配置 MyBatis SqlSessionFactory

配置 SqlSessionFactory 来使用动态数据源。

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.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.core.io.support.PathMatchingResourcePatternResolver;import javax.sql.DataSource;@Configuration
@MapperScan(basePackages = "com.example.mapper")
public class MyBatisConfig {@Autowiredprivate DataSource dynamicDataSource;@Beanpublic SqlSessionFactory sqlSessionFactory() throws Exception {SqlSessionFactoryBean factory = new SqlSessionFactoryBean();factory.setDataSource(dynamicDataSource);factory.setMapperLocations(new PathMatchingResourcePatternResolver().getResources("classpath:mapper/*.xml"));return factory.getObject();}
}

6. 在业务代码中使用

在业务代码中使用之前定义的动态数据源，可以通过如下的方式直接切换数据源。也可以通过在方法上添加注解来指定使用哪个数据源，这里会涉及到切面的开发。

import com.example.datasource.DynamicDataSource;public class Service {@Transactional(readOnly = true)public void serviceMethodForDb1() {DynamicDataSource.setDataSource("db1");// 操作 db1 数据源DynamicDataSource.clearDataSource();}@Transactional(readOnly = true)public void serviceMethodForDb2() {DynamicDataSource.setDataSource("db2");// 操作 db2 数据源DynamicDataSource.clearDataSource();}
}