如何实现mybatis多数据源切换

一、概述

随着项目的发展和数据量的增长，数据库表的数量也会不断增加，如果沿用单一数据源的设计，不但会对项目带来风险，而且还会对开发的效率造成影响。多数据源的设计正是为了解决这个问题，可以有效地降低项目的风险，并且提高开发的效率。

二、多数据源配置

在使用mybatis实现多数据源的切换前，我们需要在项目的application.yml中添加多个数据源的配置，如下所示:

spring:
  datasource:
    data-source-1:
      url: jdbc:mysql://localhost:3306/db1?useUnicode=true&characterEncoding=utf-8&useSSL=false&serverTimezone=GMT
      driver-class-name: com.mysql.cj.jdbc.Driver
      username: root
      password: root
    data-source-2:
      url: jdbc:mysql://localhost:3306/db2?useUnicode=true&characterEncoding=utf-8&useSSL=false&serverTimezone=GMT
      driver-class-name: com.mysql.cj.jdbc.Driver
      username: root
      password: root

以上配置中，我们使用了springboot的多数据源配置方式，即在application.yml中，添加多个数据源的配置。这里我们配置了两个数据源，data-source-1和data-source-2,其中两个数据源的url分别指向两个不同的数据库。在实际开发中，我们可以根据具体的情况配置多个数据源。

三、实现多数据源切换

在多数据源配置完成后，我们需要通过代码实现数据源的动态切换，在mybatis中，可以通过使用不同的SqlSessionFactory来实现多数据源切换。以下是使用mybatis实现多数据源切换的代码示例:

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

  @Bean
  public SqlSessionFactoryBean sqlSessionFactoryBean(@Qualifier("dataSource1")DataSource dataSource1,
                                                       @Qualifier("dataSource2")DataSource dataSource2) throws Exception {
    DynamicDataSource dynamicDataSource = new DynamicDataSource();// 实例化动态数据源
    Map targetDataSources = new HashMap<>();
    targetDataSources.put("data-source-1", dataSource1);
    targetDataSources.put("data-source-2", dataSource2);
    dynamicDataSource.setTargetDataSources(targetDataSources);//把所有数据源放入动态数据源里
    dynamicDataSource.setDefaultTargetDataSource(dataSource1);//设置默认数据源
    SqlSessionFactoryBean factory = new SqlSessionFactoryBean();
    factory.setDataSource(dynamicDataSource);//设置动态数据源
    factory.setMapperLocations(new PathMatchingResourcePatternResolver().getResources("classpath:mapper/*.xml"));
    //设置mapper.xml位置
    factory.setConfigLocation(new ClassPathResource("mybatis-config.xml"));//设置mybatis-config.xml位置
    return factory;
  }

  @Bean
  public SqlSessionTemplate sqlSessionTemplate(SqlSessionFactory sqlSessionFactory) {
    return new SqlSessionTemplate(sqlSessionFactory);
  }

  @Bean(name = "dataSource1")
  @ConfigurationProperties(prefix = "spring.datasource.data-source-1")
  public DataSource dataSource1() {
    return DataSourceBuilder.create().build();
  }


  @Bean(name = "dataSource2")
  @ConfigurationProperties(prefix = "spring.datasource.data-source-2")
  public DataSource dataSource2() {
    return DataSourceBuilder.create().build();
  }
}

  

在上述代码中，我们创建了一个DynamicDataSource类，该类继承自AbstractRoutingDataSource，并重写了determineCurrentLookupKey()方法，用于动态切换数据源。

public class DynamicDataSource extends AbstractRoutingDataSource {
  private static final ThreadLocal contextHolder = new ThreadLocal<>();

  @Override
  protected Object determineCurrentLookupKey() {
    return getDataSource();
  }

  public static void setDataSource(String dataSource) {
    contextHolder.set(dataSource);
  }

  public static String getDataSource() {
    return contextHolder.get();
  }

  public static void clearDataSource() {
    contextHolder.remove();
  }
}

  

在具体的使用中，我们需要在代码中指定数据源，在需要切换数据源的地方，调用DynamicDataSource.setDataSource()方法来指定访问哪个数据源。如下所示：

@Service
public class UserService {

  @Autowired
  private UserMapper userMapper;

  public List findUser(int dataSourceType) {
    String dataSource = "";
    if (dataSourceType == 1) {
      dataSource = "data-source-1";
    } else if (dataSourceType == 2) {
      dataSource = "data-source-2";
    }
    DynamicDataSource.setDataSource(dataSource);//设置数据源
    return userMapper.findUser();
  }
}

  

在以上示例代码中，我们定义了一个方法findUser()来获取用户信息，其中dataSourceType用于指示需要访问哪个数据源。通过调用DynamicDataSource.setDataSource()方法，我们可以在代码中指定数据源，从而实现数据源的动态切换。

四、小结

以上就是使用mybatis实现多数据源切换的整个过程，首先需要在项目的配置文件中配置多个数据源，在代码中使用DynamicDataSource来实现动态切换数据源。通过这种方式，我们可以很方便地实现多数据源的切换，提高项目的开发效率，减轻项目风险。