LOGO
Yutuo

最近一个项目,要用到多个数据库,具体一点的需求就是把分析好的数据同时保存到多个数据库。如果没有用MyBaits的话,应该还是不难的,用代码切换一下就行。MyBaits的多数据源切换不太会,在网上找了一堆资料,然后自己实验,总算是成功了。

根据我测试的结果,MyBaits的多数据源切换是以事务为单位的,也就说,如果想切换数据源,必须选提交前一个事务,并开启下一下事务。

下面用代码来说明一下。

首先,是Java文件

一个DynamicDataSource.java是继承AbstractRoutingDataSource的。

Java
package net.yutuo.test.common;

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

public class DynamicDataSource extends AbstractRoutingDataSource {
    @Override
    protected Object determineCurrentLookupKey() {
        return DynamicContextHolder.getDataSourceId();
    }
}

另一个DynamicContextHolder.java是用于线程安全的,为了外部使用更简单。

Java
package net.yutuo.test.common;

public class DynamicContextHolder {
    private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>();

    public static void setDataSourceId(String dataSourceId) {
        contextHolder.set(dataSourceId);
    }

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

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

这两个文件也可能写到同一个文件,分成两个更容易理解而已。

然后,定义XML

db-context.xml的定义里我们要加上多个DataSource的定义,以及DynamicDataSource的定义,并把sqlSessionFactorydataSource值修改为我们定义的动态数据源dynamicDataSource

XML
<?xml version="1.0" encoding="UTF-8"?>
<beans xmlns="http://www.springframework.org/schema/beans"
	xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xmlns:p="http://www.springframework.org/schema/p"
	xmlns:c="http://www.springframework.org/schema/c" xmlns:context="http://www.springframework.org/schema/context"
	xsi:schemaLocation="
        http://www.springframework.org/schema/beans http://www.springframework.org/schema/beans/spring-beans-3.2.xsd
        http://www.springframework.org/schema/context http://www.springframework.org/schema/context/spring-context-3.2.xsd">

	<bean id="dataSource_1" class="oracle.jdbc.pool.OracleDataSource"
		destroy-method="close">
		<property name="URL"
			value="jdbc:oracle:thin:@127.0.0.1:1521:TESTDB" />
		<property name="user" value="testuser" />
		<property name="password" value="testpassword" />
	</bean>


	<bean id="dataSource_2" class="oracle.jdbc.pool.OracleDataSource"
		destroy-method="close">
		<property name="URL"
			value="jdbc:oracle:thin:@127.0.0.2:1521:TESTDB" />
		<property name="user" value="testuser" />
		<property name="password" value="testpassword" />
	</bean>

	<bean id="dynamicDataSource" class="net.yutuo.test.common.DynamicDataSource">
		<property name="targetDataSources">
			<map key-type="java.lang.String">
				<entry value-ref="dataSource_1" key="dataSource_1"></entry>
				<entry value-ref="dataSource_2" key="dataSource_2"></entry>
			</map>
		</property>
		<property name="defaultTargetDataSource" ref="dataSource_1"></property>
	</bean>

	<bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">
		<property name="dataSource" ref="dynamicDataSource" />
		<property name="configurationProperties">
			<props>
				<prop key="callSettersOnNulls">true</prop>
			</props>
		</property>
		<property name="configLocation" value="classpath:mybatis-config.xml" />
	</bean>
    <!-- .......... --->
</beans>

最后,使用方法

因为代码量比较大,所以只添加了实际使用的代码,其它代码请自己添加。

Java
DefaultTransactionDefinition definition = new DefaultTransactionDefinition();
definition.setPropagationBehavior(TransactionDefinition.PROPAGATION_REQUIRES_NEW);
TransactionStatus status = null;

for (String dataSource : dataSourceList) {
    try {
        DynamicContextHolder.setDataSourceId(dataSource);
        status = transactionManager.getTransaction(definition);

        // .................

        transactionManager.commit(status);

    } finally {
        if (status != null && !status.isCompleted()) {
            transactionManager.rollback(status);
        }
    }
}

转载请注明:宇托的狗窝 » MyBaits多数据源的设置和切换方法

上一篇 下一篇
发表我的评论
取消评论

Hi,您需要填写昵称和邮箱!

  • 昵称 (必填)
  • 邮箱 (必填)
  • 网址