手撸一个动态多数据源注入框架

创建一个动态多数据源注入插件涉及到设计一个能够在运行时根据配置或参数动态选择和使用不同数据源的机制。在Java Spring框架中，这通常通过自定义数据源路由、配置管理和依赖注入来实现。以下是一个简化的示例，展示了如何创建一个基本的动态多数据源注入插件。
步骤 1: 定义数据源配置
首先，您需要定义一个数据源配置类，用于存储每个数据源的信息：

public class DataSourceConfig {
    private String name;
    private String url;
    private String username;
    private String password;
    // 省略Getter和Setter方法
}

步骤 2: 创建数据源路由
接下来，创建一个数据源路由类，用于根据某种条件（如注解、参数等）动态选择数据源：

import javax.sql.DataSource;
import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
public class DynamicDataSource extends AbstractRoutingDataSource {
    private static final ThreadLocal<String> CONTEXT_HOLDER = ThreadLocal.withInitial(() -> "default");
    @Override
    protected Object determineCurrentLookupKey() {
        return CONTEXT_HOLDER.get();
    }
    public static void setDataSourceName(String dataSourceName) {
        CONTEXT_HOLDER.set(dataSourceName);
    }
    public static String getDataSourceName() {
        return CONTEXT_HOLDER.get();
    }
    public static void clearDataSourceName() {
        CONTEXT_HOLDER.remove();
    }
}

步骤 3: 配置数据源和路由
在Spring配置中，您需要定义所有可能的数据源，并配置数据源路由：

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.jdbc.datasource.DataSourceTransactionManager;
import org.springframework.transaction.PlatformTransactionManager;
import javax.sql.DataSource;
import com.zaxxer.hikari.HikariDataSource;
import java.util.HashMap;
import java.util.Map;
@Configuration
public class DynamicDataSourceConfig {
@Bean
public DynamicDataSource dynamicDataSource() {
DynamicDataSource dynamicDataSource = new DynamicDataSource();
Map<Object, Object> targetDataSources = new HashMap<>();
// 根据配置动态添加数据源
DataSourceConfig defaultDataSourceConfig = ... // 获取默认数据源配置
DataSource defaultDataSource = createDataSource(defaultDataSourceConfig);
targetDataSources.put("default", defaultDataSource);
// 添加其他数据源
dynamicDataSource.setTargetDataSources(targetDataSources);
dynamicDataSource.setDefaultTargetDataSource(defaultDataSource);
return dynamicDataSource;
}
private DataSource createDataSource(DataSourceConfig config) {
HikariDataSource dataSource = new HikariDataSource();
dataSource.setJdbcUrl(config.getUrl());
dataSource.setUsername(config.getUsername());
dataSource.setPassword(config.getPassword());
// 设置其他数据源属性
return dataSource;
}
@Bean
public PlatformTransactionManager transactionManager(DynamicDataSource dynamicDataSource) {
return new DataSourceTransactionManager(dynamicDataSource);
}
}

步骤 4: 使用注解或拦截器切换数据源
您可以创建一个注解和相应的拦截器，用于在执行数据库操作前切换数据源：

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface DataSourceSwitch {
String value();
}

拦截器的实现可能会使用AOP来拦截标记了@DataSourceSwitch注解的方法，并在方法执行前设置数据源，执行后清除数据源。
步骤 5: 测试和集成
最后，您需要测试您的动态数据源插件，确保它能够在不同的场景下正确地切换数据源。您还需要考虑如何将这个插件集成到您的应用程序中，可能涉及到配置文件的管理、Spring Boot自动配置类的编写等。
请注意，上述代码是一个非常简化的示例，实际的实现可能会更加复杂，需要考虑事务管理、数据源的懒加载、配置的热更新、性能优化等多个方面。此外，多数据源管理可能会引入一些复杂性，如数据源的一致性、分布式事务处理等，这些都需要在设计和实现时充分考虑。