spring 多数据源的配置和使用（转）

1. 首先在配置文件中配置多个dataSource

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

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

publicclass DynamicDataSource extends AbstractRoutingDataSource{

 @Override
 protected Object determineCurrentLookupKey() {
 return DBContextHolder.getDBType();
 }
}

public class DynamicDataSource extends AbstractRoutingDataSource{   
   
    @Override   
    protected Object determineCurrentLookupKey() {   
        return DBContextHolder.getDBType();   
    }   
}  

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

publicclass DBContextHolder{
 publicstaticfinal String DATA_SOURCE_FROM = "dataSourceFrom";
 publicstaticfinal String DATA_SOURCE_TO = "dataSourceTo";

 privatestaticfinal ThreadLocal<String> contextHolder = new ThreadLocal<String>();

 publicstaticvoid setDBType(String dbType) {
 contextHolder.set(dbType);
 }

 publicstatic String getDBType() {
 return contextHolder.get();
 }

 publicstaticvoid clearDBType() {
 contextHolder.remove();
 }
}

public class DBContextHolder{   
    public static final String DATA_SOURCE_FROM = "dataSourceFrom";   
    public static final String DATA_SOURCE_TO = "dataSourceTo";   
       
    private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>();   
       
    public static void setDBType(String dbType) {   
        contextHolder.set(dbType);   
    }   
       
    public static String getDBType() {   
        return contextHolder.get();   
    }   
       
    public static void clearDBType() {   
        contextHolder.remove();   
    }   
}  

3. 配置动态数据源

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

<bean id="dynamicDataSource"class="datasource.DynamicDataSource" >
 <!-- 通过key-value的形式来关联数据源 -->
 <property name="targetDataSources">
 <map>
 <entry value-ref="dataSourceFrom" key="dataSourceFrom"></entry>
 <entry value-ref="dataSourceTo" key="dataSourceTo"></entry>
 </map>
 </property>
 <property name="defaultTargetDataSource" ref="dataSourceFrom" />
</bean>

<bean id="dynamicDataSource" class="datasource.DynamicDataSource" >   
    <!-- 通过key-value的形式来关联数据源 -->   
    <property name="targetDataSources">   
        <map>   
            <entry value-ref="dataSourceFrom" key="dataSourceFrom"></entry>   
            <entry value-ref="dataSourceTo" key="dataSourceTo"></entry>   
        </map>   
    </property>   
    <property name="defaultTargetDataSource" ref="dataSourceFrom" />   
</bean>   

 

 

 

4. 使用动态数据源

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

<!-- JdbcTemplate使用动态数据源的配置 -->
<bean id="jdbcTemplate"class="org.springframework.jdbc.core.JdbcTemplate">
 <property name="dataSource">
 <ref bean="dynamicDataSource" />
 </property>
</bean>

<!-- 对JdbcTemplate的应用封装类 -->
<bean id="sqlBaseDAO"class="com.whty.dao.BaseDAOImpl">
 <property name="jdbcTemplate">
 <ref bean="jdbcTemplate" />
 </property>
</bean>

<!-- JdbcTemplate使用动态数据源的配置 -->   
<bean id="jdbcTemplate" class="org.springframework.jdbc.core.JdbcTemplate">   
    <property name="dataSource">   
        <ref bean="dynamicDataSource" />   
    </property>   
</bean>   
   
<!-- 对JdbcTemplate的应用封装类 -->   
<bean id="sqlBaseDAO" class="com.whty.dao.BaseDAOImpl">   
    <property name="jdbcTemplate">   
        <ref bean="jdbcTemplate" />   
    </property>   
</bean>

5. 动态数据源的管理

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

ApplicationContext context = new ClassPathXmlApplicationContext("applicationContext.xml");
BaseDAO dao = (BaseDAO) context.getBean("sqlBaseDAO", BaseDAOImpl.class);

try {
 DBContextHolder.setCustomerType(DBContextHolder.DATA_SOURCE_FROM);
 System.err.println(dao.select("select count(*) sum from TEST t ").get(0).get("SUM"));
 DBContextHolder.setCustomerType(DBContextHolder.DATA_SOURCE_TO);
 System.err.println(dao.select("select count(*) sum from TEST t ").get(0).get("SUM"));

} catch (Exception e) {
 e.printStackTrace();
}

ApplicationContext context = new ClassPathXmlApplicationContext("applicationContext.xml");   
BaseDAO dao = (BaseDAO) context.getBean("sqlBaseDAO", BaseDAOImpl.class);   
   
try {   
    DBContextHolder.setCustomerType(DBContextHolder.DATA_SOURCE_FROM);   
    System.err.println(dao.select("select count(*) sum from TEST t ").get(0).get("SUM"));   
    DBContextHolder.setCustomerType(DBContextHolder.DATA_SOURCE_TO);   
    System.err.println(dao.select("select count(*) sum from TEST t ").get(0).get("SUM"));   
       
} catch (Exception e) {   
    e.printStackTrace();   
}  

如果在service层有比较统一的规则的话，也可以使用aop设置数据源使用。

6. 这里一般都是一个service一个数据源，所以最好使用aop在service层执行完之后统一调用

DBContextHolder.clearDBType();

DBContextHolder.clearDBType();

清空数据源信息。

当然，在上面配置里面有个参数defaultTargetDataSource为默认数据源，就是不设置数据源的话，就是用这个数据源。