spring+mybatis多数据源配置

今天尝试了下多数据源配置，在这里记录下。

1.首先在spring配置文件中配置多个数据源

<!-- 用户中心数据源 -->
<bean id="membercenterDataSource" class="com.alibaba.druid.pool.DruidDataSource" init-method="init" destroy-method="close">
   <property name="driverClassName" value="${jdbc.driver}" />
   <property name="url" value="${jdbc.url}" />
   <property name="username" value="${jdbc.username}" />
   <property name="password" value="${jdbc.password}" />
   <!-- 配置初始化大小、最小、最大 -->
   <property name="initialSize" value="1" />
   <property name="minIdle" value="1" />
   <property name="maxActive" value="10" />
   <!-- 配置获取连接等待超时的时间 -->
   <property name="maxWait" value="10000" />
   <!-- 配置间隔多久才进行一次检测，检测需要关闭的空闲连接，单位是毫秒 -->
   <property name="timeBetweenEvictionRunsMillis" value="60000" />
   <!-- 配置一个连接在池中最小生存的时间，单位是毫秒 -->
   <property name="minEvictableIdleTimeMillis" value="300000" />
   <property name="testWhileIdle" value="true" />
   <!-- 这里建议配置为TRUE，防止取到的连接不可用 -->
   <property name="testOnBorrow" value="true" />
   <property name="testOnReturn" value="false" />
   <!-- 打开PSCache，并且指定每个连接上PSCache的大小 -->
   <property name="poolPreparedStatements" value="true" />
   <property name="maxPoolPreparedStatementPerConnectionSize" value="20" />
   <!-- 这里配置提交方式，默认就是TRUE，可以不用配置 -->
   <property name="defaultAutoCommit" value="true" />
   <!-- 验证连接有效与否的SQL，不同的数据配置不同 -->
   <property name="validationQuery" value="select 1 " />
   <property name="filters" value="stat" />
</bean>
<!-- crm数据源 -->
<bean id="crmDataSource" class="com.alibaba.druid.pool.DruidDataSource" init-method="init" destroy-method="close">
   <property name="driverClassName" value="${crm.jdbc.driver}" />
   <property name="url" value="${crm.jdbc.url}" />
   <property name="username" value="${crm.jdbc.username}" />
   <property name="password" value="${crm.jdbc.password}" />
   <!-- 配置初始化大小、最小、最大 -->
   <property name="initialSize" value="1" />
   <property name="minIdle" value="1" />
   <property name="maxActive" value="10" />
   <!-- 配置获取连接等待超时的时间 -->
   <property name="maxWait" value="10000" />
   <!-- 配置间隔多久才进行一次检测，检测需要关闭的空闲连接，单位是毫秒 -->
   <property name="timeBetweenEvictionRunsMillis" value="60000" />
   <!-- 配置一个连接在池中最小生存的时间，单位是毫秒 -->
   <property name="minEvictableIdleTimeMillis" value="300000" />
   <property name="testWhileIdle" value="true" />
   <!-- 这里建议配置为TRUE，防止取到的连接不可用 -->
   <property name="testOnBorrow" value="true" />
   <property name="testOnReturn" value="false" />
   <!-- 打开PSCache，并且指定每个连接上PSCache的大小 -->
   <property name="poolPreparedStatements" value="true" />
   <property name="maxPoolPreparedStatementPerConnectionSize" value="20" />
   <!-- 这里配置提交方式，默认就是TRUE，可以不用配置 -->
   <property name="defaultAutoCommit" value="true" />
   <!-- 验证连接有效与否的SQL，不同的数据配置不同 -->
   <property name="validationQuery" value="select 1 " />
   <property name="filters" value="stat" />
</bean>
2.定义一个DataSource
package com.alog.membercenter.dal;

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

public class MultipleDataSource extends AbstractRoutingDataSource {

    private static final ThreadLocal<String> dataSourceKey = new InheritableThreadLocal<String>();

    public static void setDataSourceKey(String dataSource) {
        dataSourceKey.set(dataSource);
    }

    @Override
    protected Object determineCurrentLookupKey() {
        return dataSourceKey.get();
    }
}
3.又回到spring配置文件
<bean id="membercenterSqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">
		<property name="dataSource" ref="dataSource"/>
	</bean>

	<bean id="dataSource" class="com.alog.membercenter.dal.MultipleDataSource">
		<property name="defaultTargetDataSource" ref="membercenterDataSource"/>
		<property name="targetDataSources">
			<map>
				<entry key="membercenterDataSource" value-ref="membercenterDataSource"/>
				<entry key="crmDataSource" value-ref="crmDataSource"/>
			</map>
		</property>
	</bean>

	<!-- scan for mappers and let them be autowired -->
	<bean class="org.mybatis.spring.mapper.MapperScannerConfigurer">
		<property name="basePackage" value="com.alog.membercenter.dal.*.*" />
		<property name="sqlSessionFactoryBeanName" value="membercenterSqlSessionFactory"/>
	</bean>

	<!-- 配置Spring的事务管理器,只对主库做事务,slave库为只读DB -->
    <bean id="transactionManager" class="org.springframework.jdbc.datasource.DataSourceTransactionManager">
        <property name="dataSource" ref="membercenterDataSource"/>
    </bean>
4.配置aop数据源切换，我这边不同数据源的dao放在不同的目录下，所以以方法目录做判断
package com.alog.membercenter.biz.aspect;

import com.alog.membercenter.dal.MultipleDataSource;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.springframework.aop.AfterReturningAdvice;
import org.springframework.aop.MethodBeforeAdvice;
import org.springframework.stereotype.Component;

import java.lang.reflect.Method;

@Component
@Aspect
public class MultipleDataSourceAspect{
    @Around("execution(* com.alog.membercenter.dal..*.*(..))")
    public Object doAround(ProceedingJoinPoint jp) throws Throwable {
        String methodName=jp.getSignature().toLongString();
        if(methodName.contains("com.alog.membercenter.dal.crm")){
            MultipleDataSource.setDataSourceKey("crmDataSource");
        }else {
            MultipleDataSource.setDataSourceKey("membercenterDataSource");
        }
        return jp.proceed();
    }
}