当项目中使用到读写分离的时候，我们就会遇到多数据源的问题。多数据源让人最头痛的，不是配置多个数据源，而是如何能灵活动态的切换数据源。例如在一个spring和Mybatis的框架的项目中，我们在spring配置中往往是配置一个dataSource来连接数据库，然后绑定给sessionFactory，在dao层代码中再指定sessionFactory来进行数据库操作。
正如上图所示，每一块都是指定绑死的，如果是多个数据源，也只能是下图中那种方式。
可看出在Dao层代码中写死了两个SessionFactory，这样日后如果再多一个数据源，还要改代码添加一个SessionFactory，显然这并不符合开闭原则。
那么正确的做法应该是：
具体代码与配置如下:
1、applicationContext-mgr.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:aop="http://www.springframework.org/schema/aop" xmlns:context="http://www.springframework.org/schema/context" xmlns:tx="http://www.springframework.org/schema/tx" xmlns:p="http://www.springframework.org/schema/p" xsi:schemaLocation="http://www.springframework.org/schema/beans http://www.springframework.org/schema/context/spring-context-2.5.xsd http://www.springframework.org/schema/tx/spring-tx-2.5.xsd"> <!-- use annotation --> <context:annotation-config /> <context:component-scan base-package="com.carl.o2o.**.mgr"> </context:component-scan> <!-- master --> <bean> <property value="${driverClassName_master}"/> <property value="${username_master}"/> <property value="${password_master}"/> <property value="${url_master}?Unicode=true&characterEncoding=UTF-8&allowMultiQueries=true"/> <property value="150"/> <property value="10"/> <property value="20"/> <property value="3600"/> <property value="10"/> <property value="1800"/> </bean> <!-- slave --> <bean> <property value="${driverClassName_slave}"/> <property value="${username_slave}"/> <property value="${password_slave}"/> <property value="${url_slave}?Unicode=true&characterEncoding=UTF-8"/> <property value="150"/> <property value="10"/> <property value="20"/> <property value="3600"/> <property value="10"/> <property value="1800"/> </bean> <!-- spring 动态数据源 --> <bean> <property> <map key-type="java.lang.String"> <entry key="slave" value-ref="slave" /> </map> </property> <property ref="master" /> </bean> <!-- mybatis mapper config --> <bean> <property ref="dynamicDataSource"/> <property value="classpath:o2o_mybatis_config.xml"/> <property > <list> <value>classpath:sqlMap/*.xml</value> <value>classpath*:/com/carl/o2o/**/*.xml</value> </list> </property> </bean> <bean> <constructor-arg index="0" ref="sqlSessionFactory"></constructor-arg> </bean> <bean> <property value="com.carl.o2o.**.mgr.dao" /> </bean> <!-- 多数据源 aop --> <bean /> <aop:config> <aop:advisor pointcut="execution(* com.carl.o2o.mgr.*.*(..))" advice-ref="DataSourceAspect" /> </aop:config> <!-- 事务 --> <bean> <property ref="dynamicDataSource"></property> </bean> </beans>
2、DynamicDataSource
DynamicDataSource使用Spring中的代码结合AOP实现多数据源切换.
public class DynamicDataSource extends AbstractRoutingDataSource { public DynamicDataSource() { } protected Object determineCurrentLookupKey() { return DBContextHolder.getDbType(); } public Logger getParentLogger() { return null; } }
3、DBContextHolder
DynamicDataSource的辅助类，用于实际的切换多数据源。
public class DBContextHolder { private static ThreadLocal<String> contextHolder = new ThreadLocal(); public static String MASTER = "master"; public static String SLAVE = "slave"; public DBContextHolder() { } public static String getDbType() { String db = (String)contextHolder.get(); if(db == null) { db = MASTER; } return db; } public static void setDbType(String str) { contextHolder.set(str); } public static void setMaster() { contextHolder.set(MASTER); } public static void setSlave() { contextHolder.set(SLAVE); } public static void clearDBType() { contextHolder.remove(); } }
4、DataSourceAspect
多数据源AOP切面编程实现。