小码农的代码(三)----------SpringJDBC多数据源应用

前面项目中基础的SpringJdbc的配置说明已经简单总结,接下去就是项目中使用到多数据源这个问题了。
对于多数据源的实现,我的想法是首先配置多个数据源,然后在数据操作时通过切换数据源的方式来达到目的,那么如何在保持jdbctmplate只有一个的时候进行数据源切换呢,我考虑到可以起一个线程来实现在操作中的自由切换。
首先配置3个数据源,为方便测试就直接在本机新建了三个mysql数据库(实际过程中是使用mysql与多个sql server)来写测试用例
<bean id="defaultDateSource" class="com.mchange.v2.c3p0.ComboPooledDataSource"
		destroy-method="close">
		<property name="driverClass" value="${jdbc.driverClass}" />
		<property name="jdbcUrl" value="${jdbc.jdbcUrl}" />
		<property name="user" value="${jdbc.user}" />
		<property name="password" value="${jdbc.password}" />
	</bean>
<bean id="dataSource01" class="com.mchange.v2.c3p0.ComboPooledDataSource"
		destroy-method="close">
		<property name="driverClass" value="${jdbc.driverClass01}" />
		<property name="jdbcUrl" value="${jdbc.jdbcUrl01}" />
		<property name="user" value="${jdbc.user01}" />
		<property name="password" value="${jdbc.password01}" />
	</bean>
<bean id="dataSource02" class="com.mchange.v2.c3p0.ComboPooledDataSource"
		destroy-method="close">
		<property name="driverClass" value="${jdbc.driverClass02}" />
		<property name="jdbcUrl" value="${jdbc.jdbcUrl02}" />
		<property name="user" value="${jdbc.user02}" />
		<property name="password" value="${jdbc.password02}" />
	</bean>
习惯使用c3p0数据库连接池,下次尝试试用下阿里DruidDataSource,据说功能更加强大,但是一直没有使用过-,-跑题了 新建三个数据库还是只有一个表user(id,name,age;id逐渐自增)。
然后新建一个单例的线程ThreadLocal(这是一个很好用的东西,下次专门写个测试用例总结下,做项目总是能够不断的引出新的东西,这点真的很好),能够set/get/remove数据源
public class DatabaseContextHolder {
	private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>();

	public static void setDateBaseType(String database) {
		Assert.notNull(database, "DatabaseType cannot be null");
		contextHolder.set(database);
	}

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

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

接下来是Spring支持多数据源切换的核心类AbstractRoutingDataSource,新建一个继承这个抽象类的子类DynamicDataSource,并实现determineCurrentLookupKey方法(抽象类与接口的相似性
public class DynamicDataSource extends AbstractRoutingDataSource {

	@Override
	protected Object determineCurrentLookupKey() {
		return DatabaseContextHolder.getDateBaseType();
	}
}

继续回到spring的配置文件,通过DynamicDataSource管理多数据源,并设置默认的数据源
<bean id="dynamicDataSource" class="com.zh.demo.base.tool.DynamicDataSource">
		<property name="targetDataSources">
			<map key-type="java.lang.String">
				<entry value-ref="defaultDateSource" key="defaultDateSource"></entry>
				<entry value-ref="dataSource01" key="dataSource01"></entry>
				<entry value-ref="dataSource02" key="dataSource02"></entry>
			</map>
		</property>
		<property name="defaultTargetDataSource" ref="defaultDateSource">
		</property>
	</bean>
当然jdbctemplate中指向的数据源也需要切换成它
<bean id="jdbcTemplate" class="org.springframework.jdbc.core.JdbcTemplate">
		<property name="dataSource" ref="dynamicDataSource" />
最后新建一个DataBaseType类与配置中的数据源名称相关联
public class DataBaseType {
	public static final String DATABASE_DEFAULT = "defaultDateSource";
	public static final String DATABASE_01 = "dataSource01";
	public static final String DATABASE_02 = "dataSource02";
}
这样就可以写测试用例了。
简单的新增语句在测试类中进行数据库切换
public void test(){
		userService.saveUser(("insert into user (name,age)values('张三',13)");
		DatabaseContextHolder.clearDatabaseType();
		DatabaseContextHolder.setDateBaseType(DataBaseType.DATABASE_01);
		userService.saveUser("insert into user (name,age)values('张三',13)");
		DatabaseContextHolder.clearDatabaseType();
		DatabaseContextHolder.setDateBaseType(DataBaseType.DATABASE_02);
		userService.saveUser("insert into user (name,age)values('张三',13)");
	}
结果是三个数据库中都插入了数据,多数据源配置成功。
注意点与总结:
一、实际项目中发现一个问题,就是在同一个事务中不能够进行数据库切换,该问题暂时未解决,有方法的朋友可以的话烦请指导下。
二、在使用过程中扩展想到能否通过请求来动态的切换数据源,目前考虑到还是需要在请求中带参数来让后台识别进行数据库切换,不知道有没有更好的方法。
以下内容如存在错误之处烦请斧正以达到共同进步的效果,多谢。

你可能感兴趣的:(java,spring)