MyBatis在Spring Boot中的多數據源

在Spring Boot中配置MyBatis以支持多數據源，通常涉及以下幾個關鍵步驟：

數據源配置

• 在application.yml或application.properties文件中定義多個數據源。
• 每個數據源需要配置url、username、password和driver-class-name。

數據源配置類

• 創建數據源配置類，實現InitializingBean接口，并使用@ConfigurationProperties注解綁定數據源配置。
• 在配置類中，使用DataSourceBuilder創建數據源實例，并通過@Bean注解返回這些實例。

動態切換數據源

• 使用AbstractRoutingDataSource實現動態切換數據源的邏輯。
• 重寫determineCurrentLookupKey方法，根據業務需求決定使用哪個數據源。
• 在需要切換數據源的地方，通過DataSourceUtils.setDataSource方法設置當前線程的數據源。

Mapper掃描和事務管理

• 使用@MapperScan注解指定每個數據源對應的Mapper掃描路徑。
• 配置事務管理器，確保在多數據源環境下事務的正確處理。

示例代碼

@Configuration
@MapperScan(basePackages = "com.example.demo.mapper1", sqlSessionFactoryRef = "sqlSessionFactory1")
public class DataSource1Config {

    @Bean
    @ConfigurationProperties(prefix = "spring.datasource1")
    public DataSource dataSource1() {
        return DataSourceBuilder.create().build();
    }
}

@Configuration
@MapperScan(basePackages = "com.example.demo.mapper2", sqlSessionFactoryRef = "sqlSessionFactory2")
public class DataSource2Config {

    @Bean
    @ConfigurationProperties(prefix = "spring.datasource2")
    public DataSource dataSource2() {
        return DataSourceBuilder.create().build();
    }
}

@Configuration
public class DynamicDataSourceConfig {

    @Bean
    @Primary
    public DynamicDataSource dynamicDataSource(DataSource dataSource1, DataSource dataSource2) {
        Map<Object, Object> targetDataSources = new HashMap<>();
        targetDataSources.put("master", dataSource1);
        targetDataSources.put("slave", dataSource2);

        DynamicDataSource dynamicDataSource = new DynamicDataSource();
        dynamicDataSource.setTargetDataSources(targetDataSources);
        dynamicDataSource.setDefaultTargetDataSource(dataSource1);
        return dynamicDataSource;
    }
}

通過上述步驟，可以在Spring Boot中成功配置MyBatis以支持多數據源，并根據業務需求動態切換數據源。