java 多数据源配置

在Java中，实现多数据源配置可以通过使用Spring框架中的AbstractRoutingDataSource类实现。这个类是一个抽象类，继承自javax.sql.DataSource，并且需要重写其中的determineCurrentLookupKey()方法。该方法用于确定当前使用的数据源。

具体实现步骤如下：
1. 创建一个继承自AbstractRoutingDataSource的类，例如MultipleDataSource或DynamicDataSource。
2. 在这个类中，定义一个ThreadLocal变量，用于存储当前使用的数据源的标识。
3. 重写determineCurrentLookupKey()方法，从ThreadLocal变量中获取当前数据源的标识，并返回。
4. 可以通过设置默认数据源和目标数据源来配置多个数据源，并在需要切换数据源时，调用setDataSourceKey()或setDataSource()方法设置当前数据源的标识。

下面是一个简单的示例代码：

public class MultipleDataSource extends AbstractRoutingDataSource {
    private static final ThreadLocal<String> dataSourceKey = new InheritableThreadLocal<String>();

    public static void setDataSourceKey(String dataSource) {
        dataSourceKey.set(dataSource);
    }

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

public class DataSourceUtil {
    public static void selectTestDataSource() {
        MultipleDataSource.setDataSourceKey("testDataSource");
    }

    public static void selectDataSource() {
        MultipleDataSource.setDataSourceKey("dataSource");
    }
}

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

    public DynamicDataSource(DataSource defaultTargetDataSource, Map<Object, Object> targetDataSources) {
        super.setDefaultTargetDataSource(defaultTargetDataSource);
        super.setTargetDataSources(targetDataSources);
        super.afterPropertiesSet();
    }

    @Override
    protected Object determineCurrentLookupKey() {
        return getDataSource();
    }

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

    public static String getDataSource() {
        return CONTEXT_HOLDER.get();
    }

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