Spring配置多数据源

一、前言

Spring配置多数据源mysql+oracle

二 、操作步骤

1、在项目中创建2个类：

DynamicDataSource.java、

DynamicDataSourceHolder.java

package cn.common.datasource;import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;/** * @author  * @date  *  */public class DynamicDataSource extends AbstractRoutingDataSource {@Overrideprotected Object determineCurrentLookupKey() { // 从自定义的位置获取数据源标识 return DynamicDataSourceHolder.getDataSource();}}

package cn.common.datasource;/** * @author  * @date  *  */public class DynamicDataSourceHolder {    /**       * 注意：数据源标识保存在线程变量中，避免多线程操作数据源时互相干扰     */      private static final ThreadLocal<String> THREAD_DATA_SOURCE = new ThreadLocal<String>();        public static String getDataSource() {         return THREAD_DATA_SOURCE.get();      }       public static void setDataSource(String dataSource) {         THREAD_DATA_SOURCE.set(dataSource);      }       public static void clearDataSource() {        THREAD_DATA_SOURCE.remove();      }}

2、修改Spring配置文件

<!-- 配置数据源 --><bean name="dataSource"class="com.alibaba.druid.pool.DruidDataSource" init-method="init"destroy-method="close"><property name="url" value="${jdbc_url}" /><property name="username" value="${jdbc_username}" /><property name="password" value="${jdbc_password}" /><!-- 初始化连接大小 --><property name="initialSize" value="0" /><!-- 连接池最大使用连接数量 --><property name="maxActive" value="20" /><!-- 连接池最小空闲 --><property name="minIdle" value="0" /><!-- 获取连接最大等待时间 --><property name="maxWait" value="60000" /><!-- <property name="poolPreparedStatements" value="true" /> <property name="maxPoolPreparedStatementPerConnectionSize" value="33" /> --><property name="validationQuery" value="${validationQuery}" /><property name="testOnBorrow" value="false" /><property name="testOnReturn" value="false" /><property name="testWhileIdle" value="true" /><!-- 配置间隔多久才进行一次检测，检测需要关闭的空闲连接，单位是毫秒 --><property name="timeBetweenEvictionRunsMillis" value="60000" /><!-- 配置一个连接在池中最小生存的时间，单位是毫秒 --><property name="minEvictableIdleTimeMillis" value="25200000" /><!-- 打开removeAbandoned功能 --><property name="removeAbandoned" value="true" /><!-- 1800秒，也就是30分钟 --><property name="removeAbandonedTimeout" value="1800" /><!-- 关闭abanded连接时输出错误日志 --><property name="logAbandoned" value="true" /><!-- 监控数据库 --><!-- <property name="filters" value="mergeStat" /> --><property name="filters" value="stat" /></bean><bean name="dataSource1"class="com.alibaba.druid.pool.DruidDataSource" init-method="init"destroy-method="close"><property name="url" value="${jdbc1_url}" /><property name="username" value="${jdbc1_username}" /><property name="password" value="${jdbc1_password}" /><!-- 初始化连接大小 --><property name="initialSize" value="0" /><!-- 连接池最大使用连接数量 --><property name="maxActive" value="20" /><!-- 连接池最小空闲 --><property name="minIdle" value="0" /><!-- 获取连接最大等待时间 --><property name="maxWait" value="60000" /><!-- <property name="poolPreparedStatements" value="true" /> <property name="maxPoolPreparedStatementPerConnectionSize" value="33" /> --><property name="validationQuery" value="${validationQuery}" /><property name="testOnBorrow" value="false" /><property name="testOnReturn" value="false" /><property name="testWhileIdle" value="true" /><!-- 配置间隔多久才进行一次检测，检测需要关闭的空闲连接，单位是毫秒 --><property name="timeBetweenEvictionRunsMillis" value="60000" /><!-- 配置一个连接在池中最小生存的时间，单位是毫秒 --><property name="minEvictableIdleTimeMillis" value="25200000" /><!-- 打开removeAbandoned功能 --><property name="removeAbandoned" value="true" /><!-- 1800秒，也就是30分钟 --><property name="removeAbandonedTimeout" value="1800" /><!-- 关闭abanded连接时输出错误日志 --><property name="logAbandoned" value="true" /><!-- 监控数据库 --><!-- <property name="filters" value="mergeStat" /> --><property name="filters" value="stat" /></bean><bean id="dynamicDataSource" class="cn.chinaunicom.pmis.common.datasource.DynamicDataSource">      <property name="targetDataSources">          <map key-type="java.lang.String">            <!-- 指定lookupKey和与之对应的数据源 -->            <entry key="dataSource" value-ref="dataSource"></entry>              <entry key="dataSource1" value-ref="dataSource1"></entry>          </map>      </property>      <!-- 这里可以指定默认的数据源 -->    <property name="defaultTargetDataSource" ref="dataSource" />     </bean><bean id="jdbcTemplate"class="org.springframework.jdbc.core.JdbcTemplate"><property name="dataSource" ref="dynamicDataSource" /></bean>

3、使用 

@Overridepublic DataGrid unionDataGridQuery(PmCustRequMVO pm, PageFilter ph) {// TODO Auto-generated method stubDynamicDataSourceHolder.setDataSource("dataSource1");List<Map<String, Object>> aa=custDao.getQPA_PROV_PROJPB_PMIS_T();DynamicDataSourceHolder.setDataSource("dataSource");return custDao.unionDataGridQuery(pm, ph);}

二 、实现原理

 在spring2.0.1中引入了AbstractRoutingDataSource, 该类充当了DataSource的路由中介, 能在运行时, 根据某种key值来动态切换到真正的DataSource上。

public abstract class AbstractRoutingDataSource extends AbstractDataSource implements InitializingBean

既然是AbstractDataSource，当然就是javax.sql.DataSource的子类，于是我们自然地回去看它的getConnection方法：

public Connection getConnection() throws SQLException {          return determineTargetDataSource().getConnection();     }      public Connection getConnection(String username, String password) throws SQLException {          return determineTargetDataSource().getConnection(username, password);      }

再看determineTargetDataSource()：

protected DataSource determineTargetDataSource() {Assert.notNull(this.resolvedDataSources, "DataSource router not initialized");Object lookupKey = determineCurrentLookupKey();DataSource dataSource = (DataSource) this.resolvedDataSources.get(lookupKey);if (dataSource == null) {dataSource = this.resolvedDefaultDataSource;}if (dataSource == null) {throw new IllegalStateException("Cannot determine target DataSource for lookup key [" + lookupKey + "]");}return dataSource;

可以看到AbstractRoutingDataSource获取数据源之前会先调用determineCurrentLookupKey方法查找当前的lookupKey，这个lookupKey就是数据源标识。

因此通过重写这个查找数据源标识的方法就可以让spring切换到指定的数据源了。