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的动态数据库切换是怎么实现的了。至于其他的地方，大家有不明白的就看看源码。