Spring 的动态数据源实现

Xml代码  

1. <bean id="dataSourceFrom" class="com.mchange.v2.c3p0.ComboPooledDataSource">  
2.     <property name="driverClass" value="${jdbc.driver}" />  
3.     <property name="jdbcUrl" value="${jdbc.from.url}" />  
4.     <property name="user" value="${jdbc.from.username}" />  
5.     <property name="password" value="${jdbc.from.password}" />  
6.     <property name="autoCommitOnClose" value="true" />  
7.     <property name="checkoutTimeout" value="${cpool.checkoutTimeout}" />  
8.     <property name="initialPoolSize" value="${cpool.minPoolSize}" />  
9.     <property name="minPoolSize" value="${cpool.minPoolSize}" />  
10.     <property name="maxPoolSize" value="${cpool.maxPoolSize}" />  
11.     <property name="maxIdleTime" value="${cpool.maxIdleTime}" />  
12.     <property name="acquireIncrement" value="${cpool.acquireIncrement}" />  
13.     <property name="maxIdleTimeExcessConnections" value="${cpool.maxIdleTimeExcessConnections}" />  
14. </bean>  

 数据插入库的连接池数据源配置：

Xml代码  

1. <bean id="dataSourceTo" class="com.mchange.v2.c3p0.ComboPooledDataSource">  
2.     <property name="driverClass" value="${jdbc.driver}" />  
3.     <property name="jdbcUrl" value="${jdbc.to.url}" />  
4.     <property name="user" value="${jdbc.to.username}" />  
5.     <property name="password" value="${jdbc.to.password}" />  
6.     <property name="autoCommitOnClose" value="true" />  
7.     <property name="checkoutTimeout" value="${cpool.checkoutTimeout}" />  
8.     <property name="initialPoolSize" value="${cpool.minPoolSize}" />  
9.     <property name="minPoolSize" value="${cpool.minPoolSize}" />  
10.     <property name="maxPoolSize" value="${cpool.maxPoolSize}" />  
11.     <property name="maxIdleTime" value="${cpool.maxIdleTime}" />  
12.     <property name="acquireIncrement" value="${cpool.acquireIncrement}" />  
13.     <property name="maxIdleTimeExcessConnections" value="${cpool.maxIdleTimeExcessConnections}" />  
14. </bean>  

   注意：上面url，user，password等值是从classpath下的jdbc.properties中取得的。

   通过Spring获取属性文件中的值，以供配置文件使用：

Xml代码  

1. <bean class="org.springframework.beans.factory.config.PropertyPlaceholderConfigurer">  
2.     <property name="locations" value="classpath:jdbc.properties" />  
3. </bean>  

     2. 扩展Spring的AbstractRoutingDataSource抽象类，实现动态数据源。

    AbstractRoutingDataSource中的抽象方法determineCurrentLookupKey是实现数据源的route的核心.这里对该方法进行Override。

Java代码  

1. public class DynamicDataSource extends AbstractRoutingDataSource{  
2.   
3.     @Override  
4.     protected Object determineCurrentLookupKey() {  
5.         return DBContextHolder.getDBType();  
6.     }  
7. }  

   上下文DbContextHolder为一线程安全的ThreadLocal，具体代码如下：

Java代码  

1. public class DBContextHolder{  
2.     public static final String DATA_SOURCE_FROM = "dataSourceFrom";  
3.     public static final String DATA_SOURCE_TO = "dataSourceTo";  
4.       
5.     private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>();  
6.       
7.     public static void setDBType(String dbType) {  
8.         contextHolder.set(dbType);  
9.     }  
10.       
11.     public static String getDBType() {  
12.         return contextHolder.get();  
13.     }  
14.       
15.     public static void clearDBType() {  
16.         contextHolder.remove();  
17.     }  
18. }  

    3.配置动态数据源

 将DynamicDataSource Bean加入到Spring的上下文xml配置文件中去，同时配置DynamicDataSource的targetDataSources(多数据源目标)属性的Map映射。

Xml代码  

1. <bean id="dynamicDataSource" class="datasource.DynamicDataSource" >  
2.     <!-- 通过key-value的形式来关联数据源 -->  
3.     <property name="targetDataSources">  
4.         <map>  
5.             <entry value-ref="dataSourceFrom" key="dataSourceFrom"></entry>  
6.             <entry value-ref="dataSourceTo" key="dataSourceTo"></entry>  
7.         </map>  
8.     </property>  
9.     <property name="defaultTargetDataSource" ref="dataSourceFrom" />  
10. </bean>   

    4.使用动态数据源

    例子中DynamicDataSource是继承与AbstractRoutingDataSource，而AbstractRoutingDataSource又是继承于org.springframework.jdbc.datasource.AbstractDataSource，AbstractDataSource实现了统一的DataSource接口，所以DynamicDataSource同样可以当一个DataSource使用。

 在Spring的JdbcTemplate使用动态数据源的配置示例：

Xml代码  

1. <!-- JdbcTemplate使用动态数据源的配置 -->  
2. <bean id="jdbcTemplate" class="org.springframework.jdbc.core.JdbcTemplate">  
3.     <property name="dataSource">  
4.         <ref bean="dynamicDataSource" />  
5.     </property>  
6. </bean>  
7.   
8. <!-- 对JdbcTemplate的应用封装类 -->  
9. <bean id="sqlBaseDAO" class="com.whty.dao.BaseDAOImpl">  
10.     <property name="jdbcTemplate">  
11.         <ref bean="jdbcTemplate" />  
12.     </property>  
13. </bean>  

在ORM框架Hibernate中的使用配置示例：

Xml代码  

1. <bean id="sessionFactory" class="org.springframework.orm.hibernate3.annotation.AnnotationSessionFactoryBean">  
2.     <!-- 和普通的dataSource用法一样 -->  
3.     <property name="dataSource" ref="dynamicDataSource" />  
4.     <property name="configLocations" value="classpath:hibernate.cfg.xml" />  
5.     <property name="hibernateProperties">  
6.         <props>  
7.             <prop key="hibernate.dialect">${hibernate.dialect}</prop>     
8.     </property>  
9. </bean>  

    5.事务管理

使用动态数据源的时候，可以看出和使用单数据源的时候相比，在使用配置上几乎没有差别，在进行性事务管理配置的时候也没有差别：

使用Spring的JdbcTemplate的事务管理配置示例：

Xml代码  

1. <bean id="transactionManager" class="org.springframework.jdbc.datasource.DataSourceTransactionManager">  
2.     <property name="dataSource" ref="dynamicDataSource" />  
3. </bean>  
4.   
5. <bean id="sqlBaseDAOProxy" class="org.springframework.transaction.interceptor.TransactionProxyFactoryBean">  
6.     <property name="transactionManager" ref="transactionManager" />  
7.     <property name="target" ref="sqlBaseDAO" />  
8.     <property name="transactionAttributes">  
9.         <props>  
10.             <prop key="insert*">PROPAGATION_REQUIRED</prop>  
11.             <prop key="*">PROPAGATION_REQUIRED,readOnly</prop>  
12.         </props>  
13.     </property>  
14. </bean>  

使用Hibernate时的事务管理配置示例：

Xml代码  

1. <tx:annotation-driven transaction-manager="transactionManager"/>  
2.   
3. <bean id="transactionManager" class="org.springframework.orm.hibernate3.HibernateTransactionManager">  
4.     <property name="sessionFactory" ref="sessionFactory" />  
5. </bean>  

    6.动态数据源的管理控制

如何选择控制每个业务中需要的具体数据源，可是使用手动控制：

Java代码  

1. ApplicationContext context = new ClassPathXmlApplicationContext("applicationContext.xml");  
2. BaseDAO dao = (BaseDAO) context.getBean("sqlBaseDAO", BaseDAOImpl.class);  
3.   
4. try {  
5.     DBContextHolder.setCustomerType(DBContextHolder.DATA_SOURCE_FROM);  
6.     System.err.println(dao.select("select count(*) sum from TEST t ").get(0).get("SUM"));  
7.     DBContextHolder.setCustomerType(DBContextHolder.DATA_SOURCE_TO);  
8.     System.err.println(dao.select("select count(*) sum from TEST t ").get(0).get("SUM"));  
9.       
10. } catch (Exception e) {  
11.     e.printStackTrace();  
12. }  

也可以采用AOP的控制方式：

Java代码  

1. @Aspect  
2. public class DynamicDataSourceAspect {  
3.     @Pointcut("execution (public service.impl..*.*(..))")  
4.     public void serviceExecution(){}  
5.       
6.     @Before("serviceExecution()")  
7.     public void setDynamicDataSource(JoinPoint jp) {  
8.         for(Object o : jp.getArgs()) {  
9.             //处理具体的逻辑 ，根据具体的境况CustomerContextHolder.setCustomerType()选取DataSource  
10.         }  
11.     }  
12. }  

  7.总结

   通过扩展Spring的AbstractRoutingDataSource可以很好的实现多数据源的rout效果，而且对扩展更多的数据源有良好的伸缩性，只要增加数据源和修改DynamicDataSource的targetDataSources属性配置就好。在数据源选择控制上，可以采用手动控制(业务逻辑并不多的时候)，也可以很好的用AOP的@Aspect在Service的入口加入一个切面@Pointcut，在@Before里判断JoinPoint的类容选定特定的数据源。