Spring Boot 和 Spring Data 应用程序在运行时创建多个数据源

Spring Boot和Spring Data应用程序在运行时创建多个数据源

java
@Configuration
public class DataSourceConfig {
    @Primary
    @Bean(name = "dataSource1")
    @ConfigurationProperties(prefix = "datasource1")
    public DataSource dataSource1() {
        return DataSourceBuilder.create().build();
    }
    @Bean(name = "dataSource2")
    @ConfigurationProperties(prefix = "datasource2")
    public DataSource dataSource2() {
        return DataSourceBuilder.create().build();
    }
}
@Service
public class DataService {
    @Autowired
    @Qualifier("dataSource1")
    private DataSource dataSource1;
    @Autowired
    @Qualifier("dataSource2")
    private DataSource dataSource2;
    public List> getDataFromDataSource1() {
        JdbcTemplate jdbcTemplate = new JdbcTemplate(dataSource1);
        return jdbcTemplate.queryForList("SELECT * FROM table1");
    }
    public List> getDataFromDataSource2() {
        JdbcTemplate jdbcTemplate = new JdbcTemplate(dataSource2);
        return jdbcTemplate.queryForList("SELECT * FROM table2");
    }
}

java
@Configuration
public class DataSourceConfig {
    @Bean
    public AbstractRoutingDataSource routingDataSource(
            @Qualifier("dataSource1") DataSource dataSource1,
            @Qualifier("dataSource2") DataSource dataSource2) {
        Map targetDataSources = new HashMap<>();
        targetDataSources.put("dataSource1", dataSource1);
        targetDataSources.put("dataSource2", dataSource2);
        AbstractRoutingDataSource routingDataSource = new AbstractRoutingDataSource() {
            @Override
            protected Object determineCurrentLookupKey() {
                return DataSourceContextHolder.getDataSourceKey();
            }
        };
        routingDataSource.setDefaultTargetDataSource(dataSource1);
        routingDataSource.setTargetDataSources(targetDataSources);
        return routingDataSource;
    }
}
@Service
public class DataService {
    @Autowired
    private AbstractRoutingDataSource routingDataSource;
    public List> getData(String dataSourceKey) {
        DataSourceContextHolder.setDataSourceKey(dataSourceKey);
        JdbcTemplate jdbcTemplate = new JdbcTemplate(routingDataSource);
        return jdbcTemplate.queryForList("SELECT * FROM table");
    }
}
public class DataSourceContextHolder {
    private static final ThreadLocal contextHolder = new ThreadLocal<>();
    public static void setDataSourceKey(String dataSourceKey) {
        contextHolder.set(dataSourceKey);
    }
    public static String getDataSourceKey() {
        return contextHolder.get();
    }
}