SpringBoot如何在运行时动态添加数据源

让我们来详细讲解一下Spring Boot如何在运行时动态添加数据源。

1. 引入依赖

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

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

2. 创建数据源

我们需要在应用程序代码中动态创建数据源。我们可以实现Spring的DataSource接口来创建我们自己的数据源。下面是一个示例：

import javax.sql.DataSource;

public interface MyDataSource extends DataSource {}

在这个示例中，我们创建了一个自定义的数据源MyDataSource，它继承了Spring中的DataSource接口。

3. 动态添加数据源

接下来，我们需要创建一个DataSourceConfig类，它负责创建和管理我们的数据源。我们可以使用Spring提供的@Configuration注解来标识DataSourceConfig类：

import org.springframework.context.annotation.Configuration;

@Configuration
public class DataSourceConfig {}

现在，我们需要实现在运行时动态添加数据源的功能。我们可以使用Spring提供的AbstractRoutingDataSource类，它可以根据一些上下文信息选择数据源。下面是一个示例：

import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;

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

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

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

    @Override
    protected Object determineCurrentLookupKey() {
        return contextHolder.get();
    }
}

在这个示例中，我们创建了一个DynamicDataSource类，它继承了Spring中的AbstractRoutingDataSource类。DynamicDataSource类维护了一个ThreadLocal变量contextHolder，用于存储当前的数据源。setDataSource和clearDataSource方法用于设置和清除当前的数据源。

下面是如何在DataSourceConfig类中配置和管理动态数据源：

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.autoconfigure.jdbc.DataSourceProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Primary;
import javax.sql.DataSource;
import java.util.HashMap;
import java.util.Map;

@Configuration
public class DataSourceConfig {
    private final DataSourceProperties dataSourceProperties;
    private DataSource dynamicDataSource;

    @Autowired
    public DataSourceConfig(DataSourceProperties dataSourceProperties) {
        this.dataSourceProperties = dataSourceProperties;
    }

    @Bean
    public DataSource dataSource() {
        Map<Object, Object> targetDataSources = new HashMap<>();
        targetDataSources.put("default", defaultDataSource());
        targetDataSources.put("dataSource1", dataSource1());
        targetDataSources.put("dataSource2", dataSource2());

        DynamicDataSource dynamicDataSource = new DynamicDataSource();
        dynamicDataSource.setTargetDataSources(targetDataSources);
        dynamicDataSource.setDefaultTargetDataSource(defaultDataSource());

        this.dynamicDataSource = dynamicDataSource;

        return dynamicDataSource;
    }

    private DataSource defaultDataSource() {
        // 创建默认数据源
        return dataSourceProperties.initializeDataSourceBuilder().build();
    }

    private DataSource dataSource1() {
        // 动态创建 dataSource1 数据源
        return createDataSource("dataSource1", "jdbc:mysql://localhost:3306/db1", "user1", "password1");
    }

    private DataSource dataSource2() {
        // 动态创建 dataSource2 数据源
        return createDataSource("dataSource2", "jdbc:mysql://localhost:3306/db2", "user2", "password2");
    }

    private DataSource createDataSource(String dataSourceName, String url, String username, String password) {
        // 创建数据源
        return DataSourceBuilder.create()
                .driverClassName(dataSourceProperties.determineDriverClassName())
                .url(url)
                .username(username)
                .password(password)
                .build();
    }

    public void addDataSource(String dataSourceName, String url, String username, String password) {
        // 动态添加数据源
        Map<Object, Object> targetDataSources = new HashMap<>(dynamicDataSource.getTargetDataSources());
        targetDataSources.put(dataSourceName, createDataSource(dataSourceName, url, username, password));
        dynamicDataSource.setTargetDataSources(targetDataSources);
    }

    public void removeDataSource(String dataSourceName) {
        // 动态移除数据源
        Map<Object, Object> targetDataSources = new HashMap<>(dynamicDataSource.getTargetDataSources());
        targetDataSources.remove(dataSourceName);
        dynamicDataSource.setTargetDataSources(targetDataSources);
    }

    @Bean
    @Primary
    public DataSourceContextHolder dataSourceContextHolder() {
        return new DataSourceContextHolder();
    }
}

在这个示例中，我们首先创建了一个DataSource，这个DataSource包含了我们动态添加的数据源。我们通过调用DynamicDataSource类的setTargetDataSources和setDefaultTargetDataSource方法来设置我们的数据源。

动态添加数据源的功能通过addDataSource和removeDataSource方法实现。这两个方法分别用于添加和移除数据源。

4. 使用动态数据源

现在，我们已经可以动态添加和移除数据源了，接下来我们需要使用它。我们可以通过使用@Qualifier注解来标识我们需要使用的数据源。下面是一个示例：

@Service
public class UserService {
    private final JdbcTemplate jdbcTemplate;

    @Autowired
    public UserService(@Qualifier("dataSource") DataSource dataSource) {
        jdbcTemplate = new JdbcTemplate(dataSource);
    }

    public void insertUser(User user) {
        jdbcTemplate.update("INSERT INTO users(name) VALUES (?)", user.getName());
    }
}

在这个示例中，我们创建了一个UserService类，它使用@Qualifier注解来标识我们需要使用dataSource数据源。我们可以使用jdbcTemplate对象来执行数据库操作。

5. 示例

下面是一个示例，它演示了如何动态添加和移除数据源：

@RestController
public class DataSourceController {
    private final DataSourceConfig dataSourceConfig;

    @Autowired
    public DataSourceController(DataSourceConfig dataSourceConfig) {
        this.dataSourceConfig = dataSourceConfig;
    }

    @PostMapping("/addDataSource")
    public void addDataSource(@RequestParam String name, @RequestParam String url,
                              @RequestParam String username, @RequestParam String password) {
        dataSourceConfig.addDataSource(name, url, username, password);
    }

    @PostMapping("/removeDataSource")
    public void removeDataSource(@RequestParam String name) {
        dataSourceConfig.removeDataSource(name);
    }
}

在这个示例中，我们创建了一个DataSourceController类，它包含了两个方法，分别用于添加和移除数据源。我们可以通过调用这两个方法来动态添加和移除数据源。