spring多数据源的配置-以及原理

spring多数据源的配置

创建一个类 继承 org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource
重写方法 determineCurrentLookupKey

package com.sky.lp.util;
import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;

public class ExtendsAbstractRoutingDataSource extends AbstractRoutingDataSource{
    //使用ThreadLocal 保证线程安全
    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();
    }
}

数据源配置

<!-- 定义数据源 使用 druid 包 -->
     <bean id ="dataSource1" class= "com.alibaba.druid.pool.DruidDataSource" init-method= "init" destroy-method ="close">
            <property name ="name" value="druidOne" />
            <property name ="url" value= "jdbc:mysql://localhost:3306/littledemo" />
            <property name ="username" value="root" />
            <property name ="password" value= "laixu785^@#"></property >
            <property name ="driverClassName" value= "com.mysql.jdbc.Driver"></property >
            <property name ="initialSize" value="2" />
            <property name ="maxActive" value="10" />
            <property name ="minIdle" value="5" />
            <property name ="validationQuery" value= "SELECT COUNT(*) FROM DUAL" />
            <property name ="testWhileIdle" value="true" />
            <property name ="timeBetweenEvictionRunsMillis" value= "5000" />
     </bean >

     <!-- 定义数据源 使用 druid 包 -->
     <bean id ="dataSource2" class= "com.alibaba.druid.pool.DruidDataSource" init-method= "init" destroy-method ="close">
            <property name ="name" value="druidTwo" />
            <property name ="url" value= "jdbc:mysql://localhost:3306/doubleo" />
            <property name ="username" value="root" />
            <property name ="password" value= "laixu785^@#"></property >
            <property name ="driverClassName" value= "com.mysql.jdbc.Driver"></property >
            <property name ="initialSize" value="2" />
            <property name ="maxActive" value="10" />
            <property name ="minIdle" value="5" />
            <property name ="validationQuery" value= "SELECT COUNT(*) FROM DUAL" />
            <property name ="testWhileIdle" value="true" />
            <property name ="timeBetweenEvictionRunsMillis" value= "5000" />
     </bean >

<!-- 多数据源 -->
     <bean id ="extendsAbstractRoutingDataSource" class= "com.sky.lp.util.ExtendsAbstractRoutingDataSource" >
<!-- 默认数据源 -->
        <property name ="defaultTargetDataSource" ref= "dataSource2"/>
<!--    不用的数据源对应不用的key值 -->
        <property name ="targetDataSources">
            <map >
                <entry key ="dataSource1" value-ref= "dataSource1"/>
                <entry key ="dataSource2" value-ref= "dataSource2"/>
            </map >
        </property >
     </bean >

spring jdbcTemplate 数据源的使用，
<bean id= "jdbcTemplate" class= "org.springframework.jdbc.core.JdbcTemplate" >
    <property name ="dataSource" ref= "extendsAbstractRoutingDataSource" />
</bean >

注意：
spring集成 ibatis 或者是 hibernate 或者其他的 ORM 框架，使用的数据源都是，extendsAbstractRoutingDataSource。

测试代码

public static void main(String[] args) {
ClassPathXmlApplicationContext appCtx = new ClassPathXmlApplicationContext("spring-servlet.xml");

        ExtendsAbstractRoutingDataSource myDataSource = appCtx.getBean("extendsAbstractRoutingDataSource", ExtendsAbstractRoutingDataSource.class);
        myDataSource.setDataSourceKey("dataSource1");

        JdbcTemplate jdbc = appCtx.getBean("jdbcTemplate", JdbcTemplate.class);
        List<?> list = jdbc.queryForList("select * from user");
        System.out.println(list);
}

spring多数据源工作原理（即多个数据源到底应该选择哪一下）

原理分析（这里以 spring 的 JdbcTemplate 为例子）
JdbcTemplate 中执行sql语句，是从 dataSource 中获取 数据库 Connection，通过Connection 执行sql 语句。
那么Connection是怎么获取到的呢，是通过
org.springframework.util.Assert.DataSourceUtils.getConnection(getDataSource()) 获取Connection的。

方法 getDataSource() 源码

public DataSource getDataSource() {
    return dataSource;
}

返回的dataSource，即配置 jdbcTemplate bean 的属性dataSource<property name ="dataSource" ref= "extendsAbstractRoutingDataSource" />
看一下org.springframework.util.Assert.DataSourceUtils.getConnection(getDataSource()) 的源码

public static Connection getConnection(DataSource dataSource) throws CannotGetJdbcConnectionException {
            try {
                 return doGetConnection(dataSource);
           } catch (SQLException ex) {
                 throw new CannotGetJdbcConnectionException("Could not get JDBC Connection", ex);
           }
}

public static Connection doGetConnection(DataSource dataSource) throws SQLException {
            Assert.notNull(dataSource, "No DataSource specified");
            ConnectionHolder conHolder = (ConnectionHolder) TransactionSynchronizationManager.getResource(dataSource);
            if (conHolder != null && (conHolder.hasConnection() || conHolder.isSynchronizedWithTransaction()))

                conHolder.requested();
            if (!conHolder.hasConnection())

                 logger.debug("Fetching resumed JDBC Connection from DataSource");
           conHolder.setConnection(dataSource.getConnection());

            return conHolder.getConnection();

            logger.debug("Fetching JDBC Connection from DataSource");
           Connection con = dataSource.getConnection();
            if (TransactionSynchronizationManager.isSynchronizationActive())

                 logger.debug("Registering transaction synchronization for JDBC Connection");
            ConnectionHolder holderToUse = conHolder;
            if (holderToUse == null)
                holderToUse = new ConnectionHolder(con);
            else
                holderToUse.setConnection(con);
           holderToUse.requested();
           TransactionSynchronizationManager.registerSynchronization( new ConnectionSynchronization(holderToUse, dataSource));
           holderToUse.setSynchronizedWithTransaction( true);
            if (holderToUse != conHolder)
                 TransactionSynchronizationManager.bindResource(dataSource, holderToUse);

            return con;
}

解读上面的代码，最终的目的就是要从dataSource中获取数据库的Connection。而获取数据库的Connection必须调用dataSource的getConnection。
（即 javax.sql.DataSource的接口规定 getConnection() 方法 ）。
所以，可以在从dataSource获取Connection的过程中做手脚。这就是
org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource 做的工作。
AbstractRoutingDataSource 继承 org.springframework.jdbc.datasource.AbstractDataSource,AbstractDataSource 实现 javax.sql.DataSource 接口。

下面看一下 AbstractRoutingDataSource 的方法 getConnection 的源码

public Connection getConnection() throws SQLException {
    return determineTargetDataSource().getConnection();
}

protected DataSource determineTargetDataSource() {
    Assert.notNull(resolvedDataSources, "DataSource router not initialized");
    Object lookupKey = determineCurrentLookupKey();
    DataSource dataSource = (DataSource)resolvedDataSources.get(lookupKey);
    if(dataSource == null && (lenientFallback || lookupKey == null))
        dataSource = resolvedDefaultDataSource;

    if(dataSource == null)
        throw new IllegalStateException((new StringBuilder()).append("Cannot determine target DataSource for lookup key [").append(lookupKey).append("]").toString());

     return dataSource;
}

注意 源码中的方法 determineCurrentLookupKey()，我们自己定义的类 ExtendsAbstractRoutingDataSource 重写了 AbstractRoutingDataSource 中的 determineCurrentLookupKey 方法。

到这里大家应该明白了，spring的动态数据库切换是怎么实现的了。至于其他的地方，大家有不明白的就看看源码。