MyBatis在Spring Boot中的多數(shù)據(jù)源

在Spring Boot中配置MyBatis以支持多數(shù)據(jù)源，通常涉及以下幾個關(guān)鍵步驟：

數(shù)據(jù)源配置

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

數(shù)據(jù)源配置類

• 創(chuàng)建數(shù)據(jù)源配置類，實現(xiàn)InitializingBean接口，并使用@ConfigurationProperties注解綁定數(shù)據(jù)源配置。
• 在配置類中，使用DataSourceBuilder創(chuàng)建數(shù)據(jù)源實例，并通過@Bean注解返回這些實例。

動態(tài)切換數(shù)據(jù)源

• 使用AbstractRoutingDataSource實現(xiàn)動態(tài)切換數(shù)據(jù)源的邏輯。
• 重寫determineCurrentLookupKey方法，根據(jù)業(yè)務(wù)需求決定使用哪個數(shù)據(jù)源。
• 在需要切換數(shù)據(jù)源的地方，通過DataSourceUtils.setDataSource方法設(shè)置當(dāng)前線程的數(shù)據(jù)源。

Mapper掃描和事務(wù)管理

• 使用@MapperScan注解指定每個數(shù)據(jù)源對應(yīng)的Mapper掃描路徑。
• 配置事務(wù)管理器，確保在多數(shù)據(jù)源環(huán)境下事務(wù)的正確處理。

示例代碼

@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以支持多數(shù)據(jù)源，并根據(jù)業(yè)務(wù)需求動態(tài)切換數(shù)據(jù)源。