iBatis多數據源的苦惱

在僅使用ibatis時，多數據源簡直就是夢魘，每多一個數據源就需要多一份sql-map-config配置文件。

采用spring的AbstractRoutingDataSource就可以簡單的解決這個問題。

AbstractRoutingDataSource實現了javax.sql.DataSource接口，因此可以理解為一個虛擬的動態DataSource，在需要的時候根據上下文Context動態決定使用哪個數據源。

Spring+iBatis多數據源的配置

下面是一個完整的配置：

< 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);  
    }  
}  

以上就介紹了Spring+iBatis多數據源的配置方案。本文來自eddysheng的博客：《基于spring和ibatis的多數據源切換方案》。

【編輯推薦】

1. 操作更簡單 iBatis配置文件方法
2. 詳細：iBatis配置實例
3. iBATIS ParameterMap配置實例淺析
4. iBATIS的多對多映射配置淺析
5. iBATIS中添加DAO的配置淺析