spring配置多个数据源

 

Spring多数据源配置

1、在spring配置文件中添加几个数据源：
<bean id="TestDataSource1" class="org.apache.commons.dbcp.BasicDataSource">
   <property name="driverClassName" value="oracle.jdbc.driver.OracleDriver" />
   <property name="url" value="jdbc:oracle:thin:@//127.0.0.1:1521/orcl" />
   <property name="username" value="test1" />
   <property name="password" value="123456" />
   <property name="maxActive" value="300" />
   <property name="maxIdle" value="10" />
   <property name="maxWait" value="1000" />
   <property name="defaultAutoCommit" value="true" />
</bean>

<bean id="TestDataSource2" class="org.apache.commons.dbcp.BasicDataSource">
   <property name="driverClassName" value="oracle.jdbc.driver.OracleDriver" />
   <property name="url" value="jdbc:oracle:thin:@//127.0.0.1:1521/orcl" />
   <property name="username" value="test2" />
   <property name="password" value="123456" />
   <property name="maxActive" value="300" />
   <property name="maxIdle" value="10" />
   <property name="maxWait" value="1000" />
   <property name="defaultAutoCommit" value="true" />
</bean>


2、然后将数据源进行组合：

<!-- 组合数据源 -->
<bean id="dataSource" class="com.baoan.project.dao.hibernate.DynamicDataSource">
   <property name="targetDataSources">
      <map key-type="java.lang.String">
         <entry key="TestDataSource1" value-ref="TestDataSource1" />

　　　　　<entry key="TestDataSource2" value-ref="TestDataSource2" /> 

 </map>
 </property>
 <!-- 给一个默认的数据源配置 -->
 <property name="defaultTargetDataSource" ref="TestDataSource1" />

</bean>

3. 接着扩展一个Spring提供的AbstractRoutingDataSource，Override 其中的 determineCurrentLookupKey方法实现数据源的route. 

 

 

package datasource;

import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;

public class DynamicDataSource extends AbstractRoutingDataSource{

    @Override
    protected Object determineCurrentLookupKey() {
        return CustomerContextHolder.getCustomerType();
    }
}

而其中的CustomerContextHolder这是开发人员自己实现的一个封装了ThreadLocal类型的 ContextHolder。

 

 

package datasource;

public class CustomerContextHolder {
    public static final String DATA_SOURCE_A = "TestDataSource1";
    public static final String DATA_SOURCE_A = "TestDataSource2";

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

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

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

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

 

5. 如何是用这个动态的多数据源呢？ 其实很简单，因为我们的DynamicDataSource 是继承与AbstractRoutingDataSource，而AbstractRoutingDataSource又是继承于org.springframework.jdbc.datasource.AbstractDataSource，显然的AbstractDataSource实现了统一的DataSource接口，所以我们的DynamicDataSource 同样可以方便的当一个DataSource使用，下面拿Hibernate做例子： 

 

 

<bean id="sessionFactory" class="org.springframework.orm.hibernate3.annotation.AnnotationSessionFactoryBean">
     <!-- 可以看到和 普通的dataSource用法一样 -->
    <property name="dataSource" ref="dynamicDataSource" />
    <property name="configLocations" value="classpath:hibernate.cfg.xml" />
    <property name="hibernateProperties">
         <props>
                <prop key="hibernate.dialect">${hibernate.dialect}</prop>
         </props>
    </property>
</bean>

可以看到我们用的仍然是一个sessionFactory，这样看来事务管理的配置也和以前一样。

 

 

<tx:annotation-driven transaction-manager="transactionManager"/>

<bean id="transactionManager" class="org.springframework.orm.hibernate3.HibernateTransactionManager">
    <property name="sessionFactory" ref="sessionFactory" />
</bean>

 

DynamicDataSource Bean也在容器中了，现在剩下的就在程序中如何控制，选择某一个想要的数据源该怎么做：

 

//这样就将数据源动态的设置成了dataSourceB.
CustomerContextHolder.setCustomerType(CustomerContextHolder.DATA_SOURCE_B);

或者使用AOP实现 

package datasource;

import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.annotation.Pointcut;

@Aspect
public class DynamicDataSourceAspect {
    @Pointcut("execution (public service.impl..*.*(..))")
    public void serviceExecution(){}

    @Before("serviceExecution()")
    public void setDynamicDataSource(JoinPoint jp) {
        for(Object o : jp.getArgs()) {
            //处理具体的逻辑 ，根据具体的境况CustomerContextHolder.setCustomerType()选取DataSource
        }
    }
}

6. 总结： 我们可以看到运用AbstractRoutingDataSource可以很好的实现多数据源的，而且以后扩展更多的数据源时也非常容易，只要增加数据源和修改DynamicDataSource Bean的targetDataSources 配置就好。关于选择某一个数据源，其实可以很好的用@Aspect在Service的入口加入一个切面@Pointcut，在@Before里判断JoinPoint的类容选定特定的数据源（例如更加JoinPoint的某个key来判断在设置CustomerContextHolder.setCustomerType）。
根究实际的应用来确定。个人看法： 以前有很多应用部署了writeDataSource和readDataSource和slaveDataBase的实现，但现在的大部分应用在构架上已经不太适合了，越来越注重和用户的交互性使得数据库间他同步变得日益复杂和难以维护，所以在架构系统时不妨考虑使用水平切割的方法来切割数据库，当然这种开发需要需要更多的时间分析业务领域，选取如何的配置数据源其实也是和业务息息相关的。