Spring整合多数据源实现动态切换的实例讲解

在系统中，经常需要连接多个数据库，例如MySQL、Oracle等。Spring提供了很好的支持来整合多数据源，下面就来具体讲解如何实现。

基本配置

首先，需要在pom文件中添加Springjdbc依赖。在applicationContext.xml文件中配置数据源和JdbcTemplate。具体配置如下：

<!-- 数据源配置1 -->
<bean id="dataSource1" class="org.springframework.jdbc.datasource.DriverManagerDataSource">  
    <property name="driverClassName" value="${spring.datasource.driver-class-name1}" />  
    <property name="url" value="${spring.datasource.url1}" />  
    <property name="username" value="${spring.datasource.username1}" />  
    <property name="password" value="${spring.datasource.password1}" />  
</bean>  

<!-- 数据源配置2 -->
<bean id="dataSource2" class="org.springframework.jdbc.datasource.DriverManagerDataSource">  
    <property name="driverClassName" value="${spring.datasource.driver-class-name2}" />  
    <property name="url" value="${spring.datasource.url2}" />  
    <property name="username" value="${spring.datasource.username2}" />  
    <property name="password" value="${spring.datasource.password2}" />  
</bean>

<!-- JdbcTemplate配置1 -->
<bean id="jdbcTemplate1" class="org.springframework.jdbc.core.JdbcTemplate">  
    <property name="dataSource" ref="dataSource1" />  
</bean> 

<!-- JdbcTemplate配置2 -->
<bean id="jdbcTemplate2" class="org.springframework.jdbc.core.JdbcTemplate">  
    <property name="dataSource" ref="dataSource2" />  
</bean>   

动态切换数据源

在Spring中，使用AbstractRoutingDataSource实现数据源的动态切换。需要自定义一个类，继承AbstractRoutingDataSource，实现determineCurrentLookupKey()方法，该方法用于确定当前使用的数据源。

public class DynamicDataSource extends AbstractRoutingDataSource {  
    @Override  
    protected Object determineCurrentLookupKey() {  
        return DataSourceContextHolder.getDataSource(); // 获取当前数据源
    }  
}  

上述代码中的DataSourceContextHolder，是一个用于存放当前数据源key的容器类。使用ThreadLocal保存当前线程使用的数据源key。

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

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

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

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

配置动态数据源

将上述定义的数据源和JdbcTemplate的对象实例与动态数据源绑定。

<bean id="dataSource" class="com.xxx.DynamicDataSource">  
    <property name="defaultTargetDataSource" ref="dataSource1" />  
    <property name="targetDataSources">  
        <map>  
            <entry key="dataSource1" value-ref="dataSource1" />  
            <entry key="dataSource2" value-ref="dataSource2" />  
        </map>  
    </property>  
</bean>  

演示

下面演示一个根据不同的请求，动态切换数据源的例子。

示例1

来看一个例子，假设用户请求时，根据URL参数判断要使用哪个数据源。如下所示：

@RequestMapping("/user/find")
@ResponseBody
public List<User> findUsers(String dbName) {
    DataSourceContextHolder.setDataSource(dbName);  // 设置数据源
    List<User> userList = jdbcTemplate.query("select * from user;", new UserMapper());
    DataSourceContextHolder.clearDataSource();  // 清空数据源
    return userList;
}

这段代码中，请求中的dbName参数将被保存到DataSourceContextHolder中的ThreadLocal对象中。在UserService中，调用jdbcTemplate来获取用户数据，由于动态数据源的配置，jdbcTemplate将自动从正确的数据库中读取数据。

示例2

在实际场景中，动态切换数据源的触发不仅仅是在请求中获取URL参数，还可以在代码中通过开关来切换数据源。如下所示：

public class UserService {
    private JdbcTemplate jdbcTemplate1;
    private JdbcTemplate jdbcTemplate2;
    private boolean useJdbcTemplate1 = true;  // 是否使用jdbcTemplate1

    public List<User> findUsers() {
        JdbcTemplate jdbcTemplate = useJdbcTemplate1 ? jdbcTemplate1 : jdbcTemplate2;
        List<User> userList = jdbcTemplate.query("select * from user;", new UserMapper());
        return userList;
    }

    public void setUseJdbcTemplate1(boolean flag) {
        useJdbcTemplate1 = flag;
    }

    public void setJdbcTemplate1(JdbcTemplate jdbcTemplate1) {
        this.jdbcTemplate1 = jdbcTemplate1;
    }

    public void setJdbcTemplate2(JdbcTemplate jdbcTemplate2) {
        this.jdbcTemplate2 = jdbcTemplate2;
    }
}

上述代码中，使用JdbcTemplate1或JdbcTemplate2数据源，由操作setUseJdbcTemplate1来控制。

总结

本文介绍了Spring如何整合多数据源，并实现动态切换。通过实现自定义的AbstractRoutingDataSource，可以很方便地对多个数据源进行管理和切换。使用ThreadLocal来保存每个线程对应的数据源key，可以保证线程安全。该技术在实际开发中应用广泛，如分库分表、读写分离等。