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