spring + hibernate动态数据源配置

 在Spring 2.0.1中引入了AbstractRoutingDataSource, 该类充当了DataSource的路由中介, 能有在运行时, 根据某种key值来动态切换到真正的DataSource上, 同时对于不支持事务隔离级别的JTA事务来说, Spring还提供了另外一个类IsolationLevelDataSourceRouter来处理这个问题。

定义数据源类别：

/**
 * com.ctc.common.DataSourceMap.java
 * 2011-11-15
 */
package com.ctc.common;

/**
 * 数据源类别
 */
public class DataSourceMap {

	/**
	 * 主数据源
	 */
	public static final String MASTER = "master";
	
	/**
	 * 从数据源
	 */
	public static final String SLAVE = "slave";
}

获取key值，从而动态切换数据源：

/**
 * com.ctc.common.ContextHolder.java
 * 2011-11-15
 */
package com.ctc.common;

/**
 * 用户当前上下文支持类<br />
 * 主要负责设置上下文环境和获得上下文环境
 * 
 */
public class ContextHolder {

 private static final ThreadLocal<Object> contextHolder = new ThreadLocal<Object>();

 public static void setCustomerType(String customerType) {
  contextHolder.set(customerType);
 }

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

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

动态数据源的切换：

/**
 * com.ctc.common.DynamicDataSource.java
 * 2011-11-15
 */
package com.ctc.common;

import org.apache.log4j.Logger;
import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;

/**
 *
 */
public class DynamicDataSource extends AbstractRoutingDataSource {
 
 private static final Logger logger = Logger.getLogger(DynamicDataSource.class);

 /* (non-Javadoc)
  * @see org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource#determineCurrentLookupKey()
  */
 @Override
 protected Object determineCurrentLookupKey() {
  Object key = ContextHolder.getCustomerType();
  if(key != null){
   logger.debug("当前线程使用的数据源标识为 [ " +key.toString()+ " ].");
  }
  return key;
 }

}

spring配置文件中的配置：

<!-- 基础数据源 -->
	<bean id="baseDataSource" class="com.mchange.v2.c3p0.ComboPooledDataSource" destroy-method="close">
       <property name="driverClass" value="${jdbc.driver}" />
       <property name="initialPoolSize" value="${c3p0.initialPoolSize}" />
       <property name="minPoolSize" value="${c3p0.minPoolSize}" /><!--连接池中保留的最小连接数。-->
       <property name="maxPoolSize" value="${c3p0.maxPoolSize}" /><!--连接池中保留的最大连接数。Default: 15 -->
       <property name="maxIdleTime" value="${c3p0.maxIdleTime}" /><!--最大空闲时间,60秒内未使用则连接被丢弃。若为0则永不丢弃。Default: 0 -->
       <property name="acquireIncrement" value="${c3p0.acquireIncrement}" /><!--当连接池中的连接耗尽的时候c3p0一次同时获取的连接数。Default: 3 -->
       <property name="idleConnectionTestPeriod" value="${c3p0.idleConnectionTestPeriod}" /><!--每60秒检查所有连接池中的空闲连接。Default: 0 -->
       <property name="automaticTestTable" value="${c3p0.automaticTestTable}" /><!-- 连接检测的表名 -->
    </bean>

	<!-- 主数据源配置，一个数据库对应一个数据源 -->
	<bean id="masterDataSource" parent="baseDataSource">
       <property name="jdbcUrl" value="${jdbc.master.url}" />
       <property name="user" value="${jdbc.master.username}" />
       <property name="password" value="${jdbc.master.password}" />
       
    </bean>
    
    <!-- 从数据源配置，一个数据库对应一个数据源 -->
	<bean id="slaveDataSource"  parent="baseDataSource">
       <property name="jdbcUrl" value="${jdbc.slave.url}" />
       <property name="user" value="${jdbc.slave.username}" />
       <property name="password" value="${jdbc.slave.password}" />
    </bean>
    
    <bean id="dataSource" class="com.ctc.common.DynamicDataSource">  
       <property name="targetDataSources">  
          <map key-type="java.lang.String">
          	<entry key="master" value-ref="masterDataSource"/>
			<entry key="slave" value-ref="slaveDataSource"/>
          </map>  
       </property>  
       <property name="defaultTargetDataSource" ref="masterDataSource"/>  
    </bean> 

把dataSource注入到SessionFactory中：

 <!-- Hibernate sessionFactory注入配置 -->
    <bean id="sessionFactory" class="org.springframework.orm.hibernate3.annotation.AnnotationSessionFactoryBean">
       <property name="dataSource" ref="dataSource" />
       <property name="hibernateProperties">
           <props>
           	  <prop key="hibernate.jdbc.fetch_size">${hibernate.jdbc.fetch_size}</prop>
              <prop key="hibernate.jdbc.batch_size">${hibernate.jdbc.batch_size}</prop>
              <prop key="hibernate.dialect">${hibernate.dialect}</prop>
              <prop key="hibernate.show_sql">false</prop>
              <prop key="hibernate.format_sql">true</prop>
              <prop key="hibernate.hbm2ddl.auto">${hibernate.hbm2ddl.auto}</prop>
              <prop key="hibernate.cache.use_second_level_cache">true</prop>
              <prop key="hibernate.cache.provider_class">org.hibernate.cache.EhCacheProvider</prop>
              <prop key="hibernate.cache.use_query_cache">true</prop>
              <prop key="hibernate.connection.provider_class">org.hibernate.connection.C3P0ConnectionProvider</prop><!-- 启用C3P0 -->
           </props>
       </property>
		<property name="annotatedClasses">
			<list>
				<value>com.ctc.rbca.user.entity.User</value>
				<value>com.ctc.rbca.accessobject.entity.AccessObject</value>			
			</list>
		</property>
	</bean>

在调用方法的前面加上ContextHolder.setCustomerType(DataSourceMap.SLAVE)就可以就可以动态切换了。