动态切换数据源

/ 后端 / 无站内评论 / 367浏览

spring+mybatis整合时,有些业务场景会用到多数据源,这时候就要动态切换数据源了。
转自:http://x125858805.iteye.com/blog/2061713

一、配置文件

<!-- 数据源配置 --> </span> 
<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>
    <!-- 默认使用ds1的数据源 -->
    <property name="defaultTargetDataSource" ref="ds1"></property>     
</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");  
	   //注意这里在调用service前切换到ds2的数据源
	   DataSourceContextHolder.setDbType("ds2"); 
	   testService.add(user);  
	}  
}

四、总结

最主要的变化是DynamicDataSource 类,这个类继承了AbstractRoutingDataSource,我们再继续考察,发现这个类实现了datasource这个接口。 再仔细研究,发现我配置了多个数据源给DynamicDataSource,默认的数据源是ds1。 我们研究下AbstractRoutingDataSource类的代码,主要是两个实现datasource的方法

召唤蕾姆
琼ICP备18000156号

鄂公网安备 42011502000211号