SpringBoot多数据源切换实现代码(Mybaitis)

下面我详细讲解一下如何实现Spring Boot多数据源切换，以及如何在Mybatis框架下使用多数据源。

1. 准备工作

在开始之前，我们需要引入必要的依赖：

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-jdbc</artifactId>
</dependency>

<dependency>
    <groupId>org.mybatis.spring.boot</groupId>
    <artifactId>mybatis-spring-boot-starter</artifactId>
    <version>2.1.3</version>
</dependency>

<dependency>
    <groupId>com.zaxxer</groupId>
    <artifactId>HikariCP</artifactId>
    <version>3.4.5</version>
</dependency>

这里我们使用了Spring Boot的JDBC、Mybatis和HikariCP三个依赖，其中HikariCP是一个轻量级的连接池工具，可以大幅提升数据库操作性能。

2. 实现多数据源切换

2.1 配置文件

首先，我们需要在配置文件中配置数据源信息。假设我们有两个数据源，分别为db_1和db_2，则我们的配置如下：

# 数据源1
spring.datasource.db1.url=jdbc:mysql://localhost:3306/db1?useUnicode=true&characterEncoding=utf-8
spring.datasource.db1.username=root
spring.datasource.db1.password=root
spring.datasource.db1.driver-class-name=com.mysql.jdbc.Driver

# 数据源2
spring.datasource.db2.url=jdbc:mysql://localhost:3306/db2?useUnicode=true&characterEncoding=utf-8
spring.datasource.db2.username=root
spring.datasource.db2.password=root
spring.datasource.db2.driver-class-name=com.mysql.jdbc.Driver

2.2 配置多数据源

接下来，我们需要在代码中配置多数据源。这里我们使用@Bean注解来创建数据源 bean：

@Configuration
@EnableTransactionManagement
public class DataSourceConfig {

    @Bean(name = "db1")
    @ConfigurationProperties(prefix = "spring.datasource.db1")
    public DataSource db1() {
        return DataSourceBuilder.create().type(HikariDataSource.class).build();
    }

    @Bean(name = "db2")
    @ConfigurationProperties(prefix = "spring.datasource.db2")
    public DataSource db2() {
        return DataSourceBuilder.create().type(HikariDataSource.class).build();
    }
}

这里使用了@Configuration注解，表示这是一个配置类。@EnableTransactionManagement注解表示开启事务管理。然后，使用@Bean注解定义了db1和db2两个数据源，并配置了数据源的属性。

2.3 配置Mybatis

接下来，我们需要配置Mybatis。这里我们使用@Configuration注解来配置Mybatis：

@Configuration
@MapperScan(basePackages = "com.example.demo.dao")
public class MybatisConfig {

    @Autowired
    @Qualifier("db1")
    private DataSource db1;

    @Autowired
    @Qualifier("db2")
    private DataSource db2;

    @Bean
    public SqlSessionFactory sqlSessionFactory() throws Exception {
        SqlSessionFactoryBean sessionFactoryBean = new SqlSessionFactoryBean();
        sessionFactoryBean.setDataSource(dynamicDataSource());
        return sessionFactoryBean.getObject();
    }

    @Bean
    public DataSource dynamicDataSource() {
        DynamicDataSource dynamicDataSource = new DynamicDataSource();
        Map<Object, Object> targetDataSources = new HashMap<>();
        targetDataSources.put(DataSourceName.DB1, db1);
        targetDataSources.put(DataSourceName.DB2, db2);
        dynamicDataSource.setTargetDataSources(targetDataSources);
        dynamicDataSource.setDefaultTargetDataSource(db1);
        return dynamicDataSource;
    }

}

这里使用了@Configuration注解表示这是一个配置类，并且使用了@MapperScan注解来扫描Mapper接口所在的包。

其中，我们注入了两个数据源db1和db2，并且定义了sqlSessionFactory()方法，将动态数据源注入到SessionFactory中。同时，我们还定义了dynamicDataSource()方法，创建了一个DynamicDataSource对象，并将db1和db2两个数据源加入到targetDataSources中，设置了默认数据源为db1。

2.4 配置事务

接下来，我们需要配置事务。这里我们使用了@Transactional注解来标识事务：

@Service
@Transactional(transactionManager = "transactionManager")
public class UserServiceImpl implements UserService {

    @Autowired
    private UserMapper userMapper;

    @Override
    public List<User> listUsersFromDb1() {
        DataSourceContextHolder.setDataSource(DataSourceName.DB1);
        return userMapper.listUsers();
    }

    @Override
    public List<User> listUsersFromDb2() {
        DataSourceContextHolder.setDataSource(DataSourceName.DB2);
        return userMapper.listUsers();
    }
}

这里使用了@Service注解表示这是一个Service类，并且使用了@Transactional注解，用于开启事务。transactionManager属性表示使用哪个事务管理器，默认使用DataSourceTransactionManager，也可以自定义事务管理器。

3. 使用示例

下面，我们来看下具体的使用示例。

3.1 示例1

@RestController
@RequestMapping("/users")
public class UserController {

    @Autowired
    private UserService userService;

    @GetMapping("/db1")
    public List<User> listUsersFromDb1() {
        return userService.listUsersFromDb1();
    }

    @GetMapping("/db2")
    public List<User> listUsersFromDb2() {
        return userService.listUsersFromDb2();
    }
}

这里我们定义了一个UserController类，提供两个接口/users/db1和/users/db2，用于获取db1和db2中的用户列表。其中，使用了@Autowired注解来自动注入UserService。

3.2 示例2

@Service
@Transactional(transactionManager = "transactionManager")
public class OrderServiceImpl implements OrderService {

    @Autowired
    private OrderMapper orderMapper;

    @Override
    public List<Order> listOrderFromDb1() {
        DataSourceContextHolder.setDataSource(DataSourceName.DB1);
        return orderMapper.listOrders();
    }

    @Override
    public List<Order> listOrderFromDb2() {
        DataSourceContextHolder.setDataSource(DataSourceName.DB2);
        return orderMapper.listOrders();
    }
}

这里我们定义了一个OrderServiceImpl类，提供两个方法listOrderFromDb1()和listOrderFromDb2()，用于获取db1和db2中的订单列表。其中，使用了@Autowired注解来自动注入OrderMapper。

4. 总结

通过以上的步骤，我们已经成功地实现了Spring Boot多数据源切换。具体实现过程中，我们需要配置多个数据源，并将这些数据源注入到动态数据源中；还需要配置Mybatis和事务，确保在切换数据源时可以正确地执行Mapper接口中的操作。

在实际开发中，我们可以根据具体的情况来配置多个数据源，比如多个数据库、多个schema等。对于不同的数据源，我们可以通过调用DataSourceContextHolder.setDataSource()方法来切换数据源，从而实现对不同数据源的操作。