如何使用Spring解決ibatis多數據源的問題

本篇內容介紹了“如何使用Spring解決ibatis多數據源的問題”的有關知識，在實際案例的操作過程中，不少人都會遇到這樣的困境，接下來就讓小編帶領大家學習一下如何處理這些情況吧！希望大家仔細閱讀，能夠學有所成！

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多數據源的問題”的內容就介紹到這里了，感謝大家的閱讀。如果想了解更多行業相關的知識可以關注億速云網站，小編將為大家輸出更多高質量的實用文章！