Spring與Mybatis相結合實現多數據源切換功能

廢話不多說，關鍵代碼如下所示：

1. 代碼: DbContextHolder

 public class DbContextHolder {//線程安全的ThreadLocalprivate 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();}}

2. 代碼: DynamicDataSource

 import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;public class DynamicDataSource extends AbstractRoutingDataSource {@Overridepublic Object determineCurrentLookupKey() {return DbContextHolder.getDbType();}}

3.代碼: spring.xml

 <!-- 數據源屬性配置文件--><context:property-placeholder location="classpath:ibatis.properties" /><bean id="jksh"destroy-method="close"><!-- Connection Info --><property name="driverClassName" value="${driver.jksh}" /><property name="url" value="${url.jksh}" /><property name="username" value="${username.jksh}" /><property name="password" value="${password.jksh}" /><!-- Connection Pooling Info --><property name="maxIdle" value="${maxIdle.jksh}" /><property name="maxActive" value="${maxActive.jksh}" /><property name="defaultAutoCommit" value="false" /><property name="timeBetweenEvictionRunsMillis"value="${timeBetweenEvictionRunsMillis.jksh}" /><property name="minEvictableIdleTimeMillis" value="${minEvictableIdleTimeMillis.jksh}" /></bean><bean id="jclt"destroy-method="close"><!-- Connection Info --><property name="driverClassName" value="${driver.jclt}" /><property name="url" value="${url.jclt}" /><property name="username" value="${username.jclt}" /><property name="password" value="${password.jclt}" /><!-- Connection Pooling Info --><property name="maxIdle" value="${maxIdle.jclt}" /><property name="maxActive" value="${maxActive.jclt}" /><property name="defaultAutoCommit" value="false" /><property name="timeBetweenEvictionRunsMillis" value="${timeBetweenEvictionRunsMillis.jclt}" /><property name="minEvictableIdleTimeMillis" value="${minEvictableIdleTimeMillis.jclt}" /></bean><bean id="dataSource"><property name="targetDataSources"><map key-type="java.lang.String"><entry key="jksh" value-ref="jksh" /><entry key="jclt" value-ref="jclt" /></map></property><property name="defaultTargetDataSource" ref="jksh" /></bean>

4. 代碼:main方法

import javax.sql.DataSource;import org.springframework.context.ApplicationContext;import org.springframework.context.support.ClassPathXmlApplicationContext; import org.springframework.core.io.FileSystemResource;import org.springframework.core.io.Resource;import com.jclt.service.commons.DbContextHolder;import com.jclt.service.model.User;import org.apache.ibatis.session.SqlSession;import org.apache.ibatis.session.SqlSessionFactory;import org.mybatis.spring.SqlSessionFactoryBean;public class Text {/*** @param args*/public static void main(String[] args) {ApplicationContext appContext = new ClassPathXmlApplicationContext("client-beans.xml");DbContextHolder.setDbType("jclt");String res="src/main/resources/ibatis-config.xml";DataSource datasource=(DataSource) appContext.getBean("dataSource");SqlSessionFactoryBean bean=new SqlSessionFactoryBean();bean.setDataSource(datasource);Resource resource=new FileSystemResource(res);bean.setConfigLocation(resource);try {SqlSessionFactory sessionfactory = bean.getObject();SqlSession session=sessionfactory.openSession();User user=session.selectOne("com.jclt.service.Dao.readJKSH.findOne");System.out.println(user.getName());} catch (Exception e) {e.printStackTrace();}DbContextHolder.setDbType("jksh");String res1="src/main/resources/ibatis-config.xml";DataSource datasource1=(DataSource) appContext.getBean("dataSource");SqlSessionFactoryBean bean1=new SqlSessionFactoryBean();bean1.setDataSource(datasource1);Resource resource1=new FileSystemResource(res1);bean1.setConfigLocation(resource1);try {SqlSessionFactory sessionfactory = bean.getObject();SqlSession session=sessionfactory.openSession();User user=session.selectOne("com.jclt.service.Dao.readJKSH.findOne");System.out.println(user.getName());} catch (Exception e) {e.printStackTrace();}}}

以上所述是小編給大家介紹的Spring與Mybatis相結合實現多數據源切換功能，希望對大家有所幫助，如果大家有任何疑問請給我留言，小編會及時回復大家的。在此也非常感謝大家對武林網網站的支持！