記錄學習的點滴(Spring多數據源配置)

applicationContext.xml

	<!-- 引入數據源信息的properties文件 -->
	<context:property-placeholder location="classpath:db.properties" />
	<!-- ORACLE數據源 -->
	<bean id="oracleDataSource" class="org.springframework.jdbc.datasource.DriverManagerDataSource">
		<property name="driverClassName" value="${oracle.driver}" />
		<property name="url" value="${oracle.url}" />
		<property name="username" value="${oracle.username}" />
		<property name="password" value="${oracle.password}" />
	</bean>
	
	<!-- MYSQL數據源 -->
	<bean id="mysqlDataSource" class="org.springframework.jdbc.datasource.DriverManagerDataSource">
		<property name="driverClassName" value="${mysql.driver}" />
		<property name="url" value="${mysql.url}" />
		<property name="username" value="${mysql.username}" />
		<property name="password" value="${mysql.password}" />
	</bean>

	<!-- 多個數據源的配置 -->
	<bean id="multipleDataSource" class="com.springmybatis.system.db.MyDataSource">
		<property name="defaultTargetDataSource" ref="oracleDataSource"></property>
		<property name="targetDataSources">
			<map key-type="java.lang.String">
				<entry key="oracleDataSource" value-ref="oracleDataSource" />
				<entry key="mysqlDataSource" value-ref="mysqlDataSource" />
			</map>
		</property>
	</bean>
	
	<bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">  
	    <property name="dataSource" ref="multipleDataSource" />  
	    <property name="configLocation" value="classpath:mybatis-config.xml"></property>  
	</bean>

com.springmybatis.system.db.MyDataSource.java

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

public class MyDataSource extends AbstractRoutingDataSource {

    /** 
     * 獲取當前使用的那個數據源。 
     */ 
	@Override
	protected Object determineCurrentLookupKey() {
		return MyDbContextHolder.getDbType();
	}
}

com.springmybatis.system.db.MyDbContextHolder.java

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

	/**
	 * 設置當前數據庫。
	 * @param dbType
	 */
	public static void setDbType(String dbType) {
		contextHolder.set(dbType);
	}

	/**
	 * 取得當前數據源。
	 * @return
	 */
	public static String getDbType() {
		String str = (String) contextHolder.get();
		return str;
	}

	/**
	 * 清除上下文數據
	 */
	public static void clearDbType() {
		contextHolder.remove();
	}
}

調用實例,親測OK!

	@Transactional
	public int addAndDelUser(TestUser user) throws Exception {
		int addCount=0;int delCount=0;
		// 切換到MYSQL數據源
		MyDbContextHolder.setDbType("mysqlDataSource");
		// 添加新會員信息
		addCount = testUserDao.addUser(user);
		String[] arr = new String[]{"1"};
		for (String str : arr) {
			// 設定舊會員ID
			user = new TestUser();user.setUserId(str);
			// 刪除舊會員信息
			delCount = testUserDao.delUser(user);			
		}
		// 恢復到原先數據源
		MyDbContextHolder.clearDbType();
		return addCount + delCount;
	}
切換數據庫的時候,切記不要再Mapper層進行切換。在Mapper切換的話,數據庫還是默認數據庫。





發佈了37 篇原創文章 · 獲贊 2 · 訪問量 1萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章