基于spring和ibatis的多数据源切换方案

在仅使用ibatis时，多数据源简直就是梦魇，每多一个数据源就需要多一份sql-map-config配置文件。

 

采用spring的AbstractRoutingDataSource就可以简单的解决这个问题。

 

AbstractRoutingDataSource实现了javax.sql.DataSource接口，因此可以理解为一个虚拟的动态DataSource，在需要的时候根据上下文Context动态决定使用哪个数据源。

 

下面是一个完整的配置：

<beans xmlns="http://www.springframework.org/schema/beans"
		xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
		xmlns:aop="http://www.springframework.org/schema/aop"
		xmlns:tx="http://www.springframework.org/schema/tx"
		xmlns:jee="http://www.springframework.org/schema/jee"
		xsi:schemaLocation="
			http://www.springframework.org/schema/beans http://www.springframework.org/schema/beans/spring-beans-2.5.xsd
			http://www.springframework.org/schema/aop http://www.springframework.org/schema/aop/spring-aop-2.5.xsd
			http://www.springframework.org/schema/tx http://www.springframework.org/schema/tx/spring-tx-2.5.xsd
			http://www.springframework.org/schema/jee http://www.springframework.org/schema/jee/spring-jee-2.5.xsd">


	<!-- ========================= GENERAL DEFINITIONS ========================= -->
	<jee:jndi-lookup id="ds0" jndi-name="jdbc/ds0"/>
	<jee:jndi-lookup id="ds1" jndi-name="jdbc/ds1"/>
	<jee:jndi-lookup id="ds2" jndi-name="jdbc/ds2"/>
	
	<bean id="dataSource" class="com.xxx.xxx.util.DynamicDataSource">  
	        <property name="targetDataSources">  
	           <map key-type="java.lang.String">  
	              <entry key="0" value-ref="ds0"/>
	              <entry key="1" value-ref="ds1"/>
	              <entry key="2" value-ref="ds2"/>  
	           </map>  
	        </property>  
	        <property name="defaultTargetDataSource" ref="1"/>  
	</bean>
        <!-- SqlMap setup for iBATIS Database Layer -->
	<bean id="sqlMapClient" class="org.springframework.orm.ibatis.SqlMapClientFactoryBean">
		<property name="dataSource" ref="dataSource"/>
		<property name="configLocation" value="classpath:com/xxx/xxx/dao/sqlmap/sql-map-config.xml"/>
	</bean>
        <bean id="testDAO" class="com.xxx.xxx.dao.impl.TestDAO">
		<property name="sqlMapClient" ref="sqlMapClient"/>
	</bean>
        <bean id="testService" class="com.xxx.xxx.service.impl.TestService">
		<property name="testDAO" ref="testDAO"/>
	</bean>
</beans>

 

其核心是DynamicDataSource，代码如下

package com.xxx.xxx.util;

import org.apache.log4j.Logger;
import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;

public class DynamicDataSource extends AbstractRoutingDataSource {

	static Logger log = Logger.getLogger("DynamicDataSource");
	@Override
	protected Object determineCurrentLookupKey() {
		// TODO Auto-generated method stub
		return DbContextHolder.getDbType();
	}

}

 

上下文DbContextHolder为一线程安全的ThreadLocal，如下

package com.xxx.xxx.util;

public class DbContextHolder {
	private static final ThreadLocal contextHolder = new ThreadLocal();

	public static void setDbType(String dbType) {
		contextHolder.set(dbType);
	}

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

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

 

sql-map-config.xml如下

<?xml version="1.0" encoding="UTF-8" standalone="no"?>
<!DOCTYPE sqlMapConfig PUBLIC "-//iBATIS.com//DTD SQL Map Config 2.0//EN"
        "http://www.ibatis.com/dtd/sql-map-config-2.dtd">

<sqlMapConfig>

    <sqlMap resource="com/xxx/xxx/dao/sqlmap/Object.xml"/>

</sqlMapConfig>

 

这样在调用service之前只需要设置一下上下文即可调用相应的数据源，如下：

DbContextHolder.setDbType("2");
//execute services
//.........

dao如下

package com.xxx.xxx.dao.impl;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.apache.log4j.Logger;
import org.springframework.orm.ibatis.support.SqlMapClientDaoSupport;

import com.xxx.xxx.vo.TestObj;

public class TestDAO extends SqlMapClientDaoSupport implements ITestDAO {

    static Logger log = Logger.getLogger(TestDAO.class);

    public TestObj getTestObj(String objID) throws Exception {
        return (TestObj) getSqlMapClientTemplate().queryForObject("getTestObj", objID);
    }
}

 

 

参考http://blog.springsource.com/2007/01/23/dynamic-datasource-routing/

评论

2 楼 bo_hai 2012-12-16  

请问：这样配置可以解决事务回滚的问题吗？

1 楼 rockydang 2012-07-02  

多谢分享。
第一个配置文件中有一块配置错了，估计是手误啊

<property name="defaultTargetDataSource" ref="1"/>  应该改成
<property name="defaultTargetDataSource" ref="ds1"/>