直接看spring的配置吧

<!-- 數據源配置 --> ?
<bean id="ds1" class="org.apache.commons.dbcp.BasicDataSource" destroy-method="close"> ?
? ? <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="test" /> ?
? ? <property name="password" value="test" /> ?
? ? <property name="defaultAutoCommit" value="false"></property> ?
</bean> ?
??
<bean id="ds2" class="org.apache.commons.dbcp.BasicDataSource" destroy-method="close"> ?
? ? <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="test" /> ?
? ? <property name="password" value="test" /> ?
? ? <property name="defaultAutoCommit" value="false"></property> ?
</bean> ?
??
?<bean id="dataSource" class="app.platform.mybatis.DynamicDataSource"> ?
? ? <property name="targetDataSources"> ?
? ? ? ? <map key-type="java.lang.String"> ?
? ? ? ? ? ? <entry value-ref="ds1" key="ds1"></entry> ?
? ? ? ? ? ? <entry value-ref="ds2" key="ds2"></entry> ?
? ? ? ? </map> ?
? ? </property> ?
? ? <property name="defaultTargetDataSource" ref="ds1"></property> ? ? ?<!-- 默認使用ds1的數據源 -->
</bean> ?


<!-- mybatis配置 -->
<bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">
<property name="dataSource" ref="dataSource" />
<property name="configLocation" value="classpath:/mybatis/mybatis-config.xml" />
<property name="mapperLocations" value="classpath*:app/mapper/**/*.xml"/>
</bean>

<!-- 事務管理器配置,單數據源事務 -->
<bean id="transactionManager"
class="org.springframework.jdbc.datasource.DataSourceTransactionManager">
<property name="dataSource" ref="dataSource" />
</bean>

<!-- 事務攔截的類及方法 -->
<tx:advice id="txAdvice" transaction-manager="transactionManager">
<tx:attributes>
<tx:method name="get*" read-only="true" />
<tx:method name="find*" read-only="true" />
<tx:method name="query*" read-only="true" />
<tx:method name="is*" read-only="true" />
<tx:method name="do*" propagation="REQUIRES_NEW" rollback-for="java.lang.Exception"/>
<tx:method name="save*" propagation="REQUIRED" rollback-for="java.lang.Exception" />
<tx:method name="update*" propagation="REQUIRED" rollback-for="java.lang.Exception" />
<tx:method name="delete*" propagation="REQUIRED" rollback-for="java.lang.Exception" />
</tx:attributes>
</tx:advice>
<aop:config proxy-target-class="true">
<aop:advisor pointcut="execution(* app..service..*.*(..))" advice-ref="txAdvice" />
</aop:config>

重點是上面的數據源配置

下面的類是在數據源中用到的

package app.platform.mybatis;

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

public class DynamicDataSource extends AbstractRoutingDataSource { ?
? ? @Override ?
? ? protected Object determineCurrentLookupKey() { ?
? ? ? ? return DataSourceContextHolder.getDbType(); ?
? ? } ?
}

package app.platform.mybatis;

public class DataSourceContextHolder { ?
?
? ? private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>(); ?
??
? ? public static void setDbType(String dbType) { ?
? ? ? ? contextHolder.set(dbType); ?
? ? } ?
??
? ? public static String getDbType() { ?
? ? ? ? return ((String) contextHolder.get()); ?
? ? } ?
??
? ? public static void clearDbType() { ?
? ? ? ? contextHolder.remove(); ?
? ? } ?
} ?

這樣就可以了，下面是一個單元測試的例子

public class Test{

@Autowired
private TestService testService;


@Test ?
public void addTest() throws Exception { ?
? ?User user = new User(); ?
? ?user.setUsername("admin"); ?
? ?user.setPassword("123456"); ?
? ?
? ?DataSourceContextHolder.setDbType("ds2"); ? ? //注意這里在調用service前切換到ds2的數據源
?
? ?testService.add(user); ?
} ?
}

最主要的變化是DynamicDataSource 類，這個類繼承了AbstractRoutingDataSource，我們再繼續考察，發現這個類實現了datasource這個接口。?

再仔細研究，發現我配置了多個數據源給DynamicDataSource，默認的數據源是ds1。?
我們研究下AbstractRoutingDataSource類的代碼，主要是兩個實現datasource的方法?

Java代碼??

public?Connection?getConnection()?throws?SQLException?{??

????return?determineTargetDataSource().getConnection();??

}??

??

public?Connection?getConnection(String?username,?String?password)?throws?SQLException?{??

????return?determineTargetDataSource().getConnection(username,?password);??

}??

根據這段代碼發現主要玄機都在determineTargetDataSource()方法上。?

Java代碼??

protected?DataSource?determineTargetDataSource()?{??

????????Assert.notNull(this.resolvedDataSources,?"DataSource?router?not?initialized");??

????????Object?lookupKey?=?determineCurrentLookupKey();??

????????DataSource?dataSource?=?this.resolvedDataSources.get(lookupKey);??

????????if?(dataSource?==?null?&&?(this.lenientFallback?||?lookupKey?==?null))?{??

????????????dataSource?=?this.resolvedDefaultDataSource;??

????????}??

????????if?(dataSource?==?null)?{??

????????????throw?new?IllegalStateException("Cannot?determine?target?DataSource?for?lookup?key?["?+?lookupKey?+?"]");??

????????}??

????????return?dataSource;??

????}??

根據這段代碼發現，首先在使用數據源之前，首先判斷使用數據源的key，也就是我們配置給?
Java代碼??

private?Map<Object,?Object>?targetDataSources;??

這個map中的key值，找到key值之后再找到對應的datasource然后并使用這個數據源。?


從上面我們發現，實際上DynamicDataSource只是在內部封裝了數據源，然后調用它，只不過在內部他加了一些控制而已。（此處不知道是否可以理解為代理模式）?

再深一步想想，此處使用的orm層是mybatis，如果換成hibernate呢，或者jdbctemplate呢。?
實際上這個方法都適用。

總結

以上是生活随笔為你收集整理的spring+mybatis 多数据源整合的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。