Spring多数据源的配置

1.业务需求

在项目开发中，经常会遇到需要操作多个数据源的情况，比如数据同步，在数据库A中增加数据的同时写入数据库B。这就需要做到可以配置多个数据源，并做到动态切换数据源。

2.多数据源配置

首先是多数据源的配置，在applicationContext.xml文件中。
主要配置代码如下：

    <!-- 定义数据源的信息 -->      <bean id="parentDataSource" class="com.mchange.v2.c3p0.ComboPooledDataSource"          destroy-method="close">          <property name="driverClass">              <value>com.mysql.jdbc.Driver</value>          </property>        <property name="maxPoolSize">              <value>80</value>          </property>          <property name="minPoolSize">              <value>1</value>          </property>          <property name="initialPoolSize">              <value>1</value>          </property>          <property name="maxIdleTime">              <value>20</value>          </property>      </bean>       <bean id="dataSource_local" parent="parentDataSource">        <property name="jdbcUrl">              <value>jdbcUrl_local</value>          </property>          <property name="user">              <value>root</value>          </property>          <property name="password">              <value>password</value>        </property>       </bean>    <bean id="dataSource_remote" parent="parentDataSource">           <property name="jdbcUrl">              <value>jdbcUrl_remote</value>          </property>          <property name="user">              <value>root</value>          </property>          <property name="password">              <value>password</value>          </property>      </bean>     <!-- 动态DataSource配置 -->      <bean id="dynamicDataSource" class="gov.pbc.util.DynamicDataSource">          <property name="targetDataSources">              <map key-type="java.lang.String">                  <entry key="dataSource_local" value-ref="dataSource_local"/>                  <entry key="dataSource_remote" value-ref="dataSource_remote"/>              </map>          </property>          <property name="defaultTargetDataSource" ref="dataSource_local"/>      </bean>    <!--定义Hibernate的SessionFactory -->      <!-- SessionFactory使用的数据源为上面的数据源 -->      <!-- 指定了Hibernate的映射文件和配置信息 -->      <bean id="sessionFactory"          class="org.springframework.orm.hibernate3.annotation.AnnotationSessionFactoryBean">          <property name="dataSource" ref="dynamicDataSource" />        <!-- 设置自动扫描，这样就不用一一注册model的ORM映射了  -->        <property name="packagesToScan">            <list>                <value>com.test.model</value>            </list>        </property>          <!-- 设置Hibernate的相关属性 -->        <property name="hibernateProperties">            <props>                <!-- 设置Hibernate的数据库方言 -->                <prop key="hibernate.dialect">org.hibernate.dialect.MySQLDialect</prop>                <!-- 设置Hibernate是否在控制台输出SOL语句，开发调试阶段通常设置为true -->                <prop key="hibernate.show_sql">true</prop>                <prop key="hibernate.format_sql">true</prop>                <prop key="hibernate.current_session_context_class">thread</prop>                <prop key="hbm2ddl.auto">update</prop>                <prop key="hibernate.connection.release_mode">after_statement</prop>            </props>        </property>     </bean>    <bean id="hbnTemplate" class="org.springframework.orm.hibernate3.HibernateTemplate">        <property name="sessionFactory" ref="sessionFactory"></property>    </bean>    <bean id="txManager"          class="org.springframework.orm.hibernate3.HibernateTransactionManager">          <property name="sessionFactory" ref="sessionFactory" />      </bean>    <!-- 开启基于@Transactional注解方式的事务管理 -->    <tx:annotation-driven transaction-manager="txManager"/>

3.数据源动态切换

首先，建立一个获得和设置上下文环境的类，主要负责改变上下文数据源的名称。代码如下：

public class CustomContextHolder {    private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>();      public static final String DATA_SOURCE_LOCAL = "dataSource_local";//对应动态数据源配置中的key      public static final String DATA_SOURCE_REMOTE = "dataSource_remote";      public static void setCustomerType(String customerType) {           contextHolder.set(customerType);      }      public static String getCustomerType() {          return contextHolder.get();      }      public static void clearCustomerType() {          contextHolder.remove();      }}

然后，建立动态数据源类，注意，这个类必须继承AbstractRoutingDataSource，且实现方法determineCurrentLookupKey。代码如下：

import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;public class DynamicDataSource extends AbstractRoutingDataSource{    @Override      protected Object determineCurrentLookupKey() {          String sourceType = CustomContextHolder.getCustomerType();          //System.out.println("DataSourceType: "+sourceType);          return sourceType;      } }

4.执行数据源的切换

在需要对某个数据库进行操作前，执行切换数据源的代码，个人习惯在service类中执行切换。示例代码如下：

import javax.annotation.Resource;import com.test.dao.LoTeacherDAO;import com.test.dao.ReTeacherDAO;import com.test.model.LoTeacher;import com.test.model.ReTeacher;import com.test.util.CustomContextHolder;public class TeacherServiceImpl implements TeacherService{    @Resource    private LoTeacherDAO loTeacherDAO;    @Resource    private ReTeacherDAO reTeacherDAO;    public void addLoTeacher(LoTeacher loTea){        CustomContextHolder.setCustomerType(CustomContextHolder.DATA_SOURCE_LOCAL);        loTeacherDAO.save(loTea);    }    public void addReTeacher(ReTeacher reTea){        CustomContextHolder.setCustomerType(CustomContextHolder.DATA_SOURCE_REMOTE);        reTeacherDAO.save(reTea);    }}