在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";
- }
/**
* 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.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;
- }
- }
/**
* 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>
<!-- 基础数据源 -->
<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>
<!-- 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)就可以就可以动态切换了。