有时候因为项目的需要,可能一个项目需要连接多个数据库,比如后台管理系统,每个用户的权限管理什么的可能就是自己独有的一个数据源,然后业务操作又是另外一个数据源。如果仅仅是管理一个项目的业务,那可能偷懒点,干脆把表挪到一起,凑合着用,但是可能一个管理后台还管理着多个项目的业务配置,每个项目又是独立的数据库,因此为这样一个后台管理系统配置一套多数据源切换还是很有必要的。
虽然有时候没有明确要求说要分开数据库来开发,但本着折腾的心态,还是自己动手实践了一下,一番周折后,终于把事情解决,期间也遇到了些问题,觉得还是有必要记录一下。
application.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:context="http://www.springframework.org/schema/context"
xmlns:tx="http://www.springframework.org/schema/tx"
xmlns:task="http://www.springframework.org/schema/task"
xmlns:p="http://www.springframework.org/schema/p"
xmlns:util="http://www.springframework.org/schema/util"
xmlns:aop="http://www.springframework.org/schema/aop"
xsi:schemaLocation="http://www.springframework.org/schema/beans
http://www.springframework.org/schema/beans/spring-beans-4.1.xsd
http://www.springframework.org/schema/context
http://www.springframework.org/schema/context/spring-context-4.1.xsd
http://www.springframework.org/schema/tx
http://www.springframework.org/schema/tx/spring-tx-4.1.xsd
http://www.springframework.org/schema/task
http://www.springframework.org/schema/task/spring-task-4.1.xsd
http://www.springframework.org/schema/util
http://www.springframework.org/schema/util/spring-util-4.2.xsd
http://www.springframework.org/schema/aop
http://www.springframework.org/schema/aop/spring-aop-4.1.xsd"
default-lazy-init="false">
<aop:aspectj-autoproxy/>
<context:component-scan base-package="xxx.xxx.xxx"/>
<!-- 定时器开关 开始 -->
<task:annotation-driven />
<!-- 引入配置文件 -->
<bean id="propertyConfigurer"
class="org.springframework.beans.factory.config.PropertyPlaceholderConfigurer">
<property name="location" value="classpath:jdbc.properties" />
</bean>
<bean id="adminDataSource" class="com.alibaba.druid.pool.DruidDataSource" destroy-method="close">
<property name="driverClassName" value="${admin.jdbc.driver}" />
<property name="url" value="${admin.jdbc.url}" />
<property name="username" value="${admin.jdbc.username}" />
<property name="password" value="${admin.jdbc.password}" />
<!-- 初始化连接大小 -->
<property name="initialSize" value="${admin.jdbc.initialSize}"></property>
<!-- 连接池最大数量 -->
<property name="maxActive" value="${admin.jdbc.maxActive}"></property>
<!-- 连接池最大空闲 -->
<property name="maxIdle" value="${admin.jdbc.maxIdle}"></property>
<!-- 连接池最小空闲 -->
<property name="minIdle" value="${admin.jdbc.minIdle}"></property>
<!-- 获取连接最大等待时间 -->
<property name="maxWait" value="${admin.jdbc.maxWait}"></property>
<!-- 配置间隔多久才进行一次检测,检测需要关闭的空闲连接,单位是毫秒 -->
<property name="timeBetweenEvictionRunsMillis" value="60000" />
<!-- 配置一个连接在池中最小生存的时间,单位是毫秒 -->
<property name="minEvictableIdleTimeMillis" value="300000" />
<property name="validationQuery" value="SELECT 'x'" />
<property name="testWhileIdle" value="true" />
<property name="testOnBorrow" value="true" />
<property name="testOnReturn" value="true" />
<!-- 打开PSCache,并且指定每个连接上PSCache的大小 -->
<property name="poolPreparedStatements" value="true" />
<property name="maxPoolPreparedStatementPerConnectionSize" value="20" />
<!-- 配置监控统计拦截的filters -->
<property name="filters" value="stat,log4j" />
</bean>
<bean id="appDataSource" class="com.alibaba.druid.pool.DruidDataSource" destroy-method="close">
<property name="driverClassName" value="${app.jdbc.driver}" />
<property name="url" value="${app.jdbc.url}" />
<property name="username" value="${app.jdbc.username}" />
<property name="password" value="${app.jdbc.password}" />
<!-- 初始化连接大小 -->
<property name="initialSize" value="${app.jdbc.initialSize}"></property>
<!-- 连接池最大数量 -->
<property name="maxActive" value="${app.jdbc.maxActive}"></property>
<!-- 连接池最大空闲 -->
<property name="maxIdle" value="${app.jdbc.maxIdle}"></property>
<!-- 连接池最小空闲 -->
<property name="minIdle" value="${app.jdbc.minIdle}"></property>
<!-- 获取连接最大等待时间 -->
<property name="maxWait" value="${app.jdbc.maxWait}"></property>
<!-- 配置间隔多久才进行一次检测,检测需要关闭的空闲连接,单位是毫秒 -->
<property name="timeBetweenEvictionRunsMillis" value="60000" />
<!-- 配置一个连接在池中最小生存的时间,单位是毫秒 -->
<property name="minEvictableIdleTimeMillis" value="300000" />
<property name="validationQuery" value="SELECT 'x'" />
<property name="testWhileIdle" value="true" />
<property name="testOnBorrow" value="true" />
<property name="testOnReturn" value="true" />
<!-- 打开PSCache,并且指定每个连接上PSCache的大小 -->
<property name="poolPreparedStatements" value="true" />
<property name="maxPoolPreparedStatementPerConnectionSize" value="20" />
<!-- 配置监控统计拦截的filters -->
<property name="filters" value="stat,log4j" />
</bean>
<bean id="multipleDataSource" class="xxx.xxx.xxx....MultipleDataSource">
<property name="defaultTargetDataSource" ref="adminDataSource"/>
<property name="targetDataSources">
<map>
<entry key="adminDataSource" value-ref="adminDataSource"/>
<entry key="appDataSource" value-ref="appDataSource"/>
</map>
</property>
</bean>
<bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">
<property name="dataSource" ref="multipleDataSource" />
<property name="mapperLocations" value="classpath*:mapper/*.xml"></property>
<property name="configLocation" value="classpath:mybatis-config.xml"></property>
<property name="plugins">
<array>
<bean class="com.github.pagehelper.PageHelper">
<property name="properties">
<value>
dialect=mysql
reasonable=true
</value>
</property>
</bean>
</array>
</property>
</bean>
<bean class="org.mybatis.spring.mapper.MapperScannerConfigurer">
<property name="basePackage" value="xxx.xxx.xxx.mapper,yyy.yyy.yyy.yyy.mapper" />
<property name="sqlSessionFactoryBeanName" value="sqlSessionFactory"></property>
</bean>
<tx:annotation-driven transaction-manager="transactionManager" />
<bean id="transactionManager"
class="org.springframework.jdbc.datasource.DataSourceTransactionManager"
p:dataSource-ref="multipleDataSource"/>
</beans>
MultipleDataSource类
import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
/**
* 设置切换数据库数据源
*/
public class MultipleDataSource extends AbstractRoutingDataSource {
public static final String DATA_SOURCE_ADMIN = "adminDataSource";
public static final String DATA_SOURCE_APP = "appDataSource";
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();
}
public static void removeDataSourceKey(){
dataSourceKey.remove();
}
}
以上,我们就简单的配置好了两个数据源,一个属于admin的,一个属于app,默认是接入admin的数据源。在MultipleDataSource 中的静态属性值分别对应在application.xml中配置好的bean id。因为可能具体的业务操作是通过引入已经写好dao、service层,包括sql映射文件的jar包,因此mybatis的mapper扫描我们可以通过,分割来配置扫描多个包,xml的扫描也配置扫描全部jar包下的所有文件,classpath*:mapper/*.xml。这样后台管理其实只要写个Controller来接受请求,所有的业务调用都引用相应工程的jar包,相对来说还是挺方便的。
以上的配置已经可以让项目跑起来,对于操作admin数据源自有的表来说不会报什么错,当操作到app数据源的表就会报在admin数据库中不存在某某表,这是因为我们还没有对数据源的切换进行配置,本来以为一路下来会顺风顺水,但最重要的也是最折腾的地方在于下面的数据源切换上。
import java.lang.reflect.Proxy;
import org.apache.commons.lang.ClassUtils;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.After;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.springframework.stereotype.Component;
/**
* 数据源切换AOP
*
*/
@Component
@Aspect
public class MultipleDataSourceInterceptor {
/**
* 拦截器对所有的业务实现类请求之前进行数据源切换
* 特别注意,由于用到了多数据源,Mapper的调用最好只在*ServiceImpl,不然调用到非默认数据源的表时,会报表不存在的异常
* @param joinPoint
* @throws Throwable
*/
@Before( "execution(* xxx.xxx.xxx..*.*ServiceImpl.*(..))")
public void setDataSoruce(JoinPoint joinPoint) throws Throwable {
Class<?> clazz = joinPoint.getTarget().getClass();
String className = clazz.getName();
if (ClassUtils.isAssignable(clazz, Proxy.class)) {
className = joinPoint.getSignature().getDeclaringTypeName();
}
//对类名含有business的设置为app数据源,否则默认为后台的数据源
if(className.contains(".business.")){
MultipleDataSource.setDataSourceKey(MultipleDataSource.DATA_SOURCE_APP);
}else{
MultipleDataSource.setDataSourceKey(MultipleDataSource.DATA_SOURCE_ADMIN);
}
}
/**
* 当操作完成时,释放当前的数据源
* 如果不释放,频繁点击时会发生数据源冲突,本是另一个数据源的表,结果跑到另外一个数据源去,报表不存在
* @param joinPoint
* @throws Throwable
*/
@After( "execution(* xxx.xxx.xxx..*.*ServiceImpl.*(..))")
public void removeDataSoruce(JoinPoint joinPoint) throws Throwable {
MultipleDataSource.removeDataSourceKey();
}
}
一开始只配置Before拦截,在操作完成后没有进行移除数据源操作,在进行测试时偶尔会出现两个数据源混掉的情况,后来也是折腾了一段时间,后面想想每个业务完成后还是把线程局部变量的值给清空一下,于是有了After拦截进行移除操作,终于不会再发生数据源混掉的情况了。大概原因是业务操作完成之后把数据源锁定在某一个数据源后,然后该线程在其他地方又直接调用了Mapper的方法,这个Mapper可能操作的又是另外一个数据源了,比如说操作日志。。。,所以说用完了最好还是把它给还回去,从哪里来到哪里去,省得哪里又搞出事情来。
OVER。