<font size="1">版权声明:本文为博主原创文章,未经博主允许不得转载。https://blog.csdn.net/qq_36523638/article/details/84571304</font>
Spring+mybatis动态切换数据源
这是基于mybatis的动态切换数据源
数据源连接池使用的为Druid
bean工厂使用的是mybatis的SqlSessionFactoryBean
1、配置第一个数据源:
#配置第一个数据源(具体配置视情况而定):
<bean id="dataSourceA" 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="filters" value="stat,log4j"/>-->
<property name="maxActive" value="20"/>
<property name="initialSize" value="1"/>
<property name="maxWait" value="60000"/>
<property name="minIdle" value="1"/>
<property name="timeBetweenEvictionRunsMillis" value="3000"/>
<property name="minEvictableIdleTimeMillis" value="300000"/>
<property name="validationQuery" value="SELECT 'x' FROM DUAL"/>
<property name="testWhileIdle" value="true"/>
<property name="testOnBorrow" value="false"/>
<property name="testOnReturn" value="false"/>
<property name="poolPreparedStatements" value="true"/>
<property name="maxPoolPreparedStatementPerConnectionSize" value="20"/>
</bean>
2、配置第二个数据源:
#配置第二个数据源(具体配置视情况而定):
<bean id="dataSourceB" 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}"/>
</bean>
#配置第三个数据源
#配置第四个数据源
...
...
#可以配置多个数据源,只要不超出连接池的限制就可以
3、配置动态的数据源(关键)这里的开关:MultipleDataSource在第5步说明
<bean id="multipleDataSource" class="包名.MultipleDataSource">
<!-- 这里可以设置一个默认的数据源 >
<property name="defaultTargetDataSource" ref="dataSource" />
<property name="targetDataSources">
<map>
<entry key="dataSourceA" value-ref="dataSourceA" />
<entry key="dataSourceB" value-ref="dataSourceB" />
</map>
</property>
</bean>
4、然后在配置sqlSessionFactory的配置文件中添加属性:
<property name="dataSource" ref="multipleDataSource"/>
添加完后是这样:
<bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">
<property name="dataSource" ref="multipleDataSource"/>
<一些插件的配置>
...
</>
</bean>
5、包名.MultipleDataSource(动态切换开关):
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();
}
public static void removeDataSourceKey() {
dataSourceKey.remove();
}
}
6、最后自己写一个切面AOP来做数据源的切换:
/**
* 数据源切换AOP
*/
@Aspect
@Order(Ordered.LOWEST_PRECEDENCE-1)
@Component
public class MultipleDataSourceInterceptor {
/**
* 拦截器对所有的业务实现类请求之前进行数据源切换 特别注意,由于用到了多数据源,Mapper的调用最好只在*ServiceImpl,不然调用到非默认数据源的表时,会报表不存在的异常
*
* @throws
*/
@Pointcut("execution(* com..*Impl.*(..)) || execution(* com..*Mapper.*(..))")
public void log(){
}
@Before("log()")
public void setDataSoruce(JoinPoint joinPoint)
throws Throwable {
//int i=8;
//int j=i/0;
Class<?> clazz = joinPoint.getTarget().getClass();
String className = clazz.getName();
if (ClassUtils.isAssignable(clazz, Proxy.class)) {
className = joinPoint.getSignature().getDeclaringTypeName();
}
// 对包名含有serverwpms的设置为wpms数据源,否则默认为后台的数据源
if (className.contains(".serverwpms.")) {
MultipleDataSource.setDataSourceKey("dataSourceA");
}else if (className.contains(".server.") || className.contains(".mapper.")) {
MultipleDataSource.setDataSourceKey("dataSourceB");
}
}
/**
* 当操作完成时,释放当前的数据源 如果不释放,频繁点击时会发生数据源冲突,本是另一个数据源的表,结果跑到另外一个数据源去,报表不存在
*
* @param joinPoint
* @throws Throwable
*/
@After("log()")
public void removeDataSoruce(JoinPoint joinPoint)
throws Throwable {
MultipleDataSource.removeDataSourceKey();
}
}
aop在springmvc的配置文件中定义下:
<!-- 定义aspect类 -->
<bean name="multipleDataSourceInterceptor" class="包名.MultipleDataSourceInterceptor"/>
<font size="4" color="red"> ----到这,系统中如果没有用到事务,多数据源的配置就完成了,aop会根据你想要的需求去动态切换数据源,如果用到事务的话,仅仅到这是不够的</font>
查阅资料说事务管理器是从connection级别来处理的,尤其是非常方便的DataSourceTransactionManager管理器,方法上用一个@Transactional注解就可以完成事务管理。
DataSourceTransactionManager管理器会对connection进行缓存(大概是这个意思),导致我们的aop MultipleDataSourceInterceptor虽然将数据源A的字符串切换为了数据源B,但是根本上的connection是并没有切换回来的,这时候如果本该对数据源A操作的sql去操作数据源B了,会报表不存在的错误。
<font size = "4" color="blue">这时候应该怎么解决呢,其实主要是事务管理器执行的优先级比我们的aop优先级高罢了,每次都是事务管理器先执行,然后aop才去切换数据源,发现并没有切换过来,手动设置下aop的优先级高于事务管理器就好了,而事务管理器的优先级默认为Order(1),也就是我之前所见过的最高的优先级了</font>
<font size="5" color="green"></font>
上面<font size="5" color="red">@Order(Ordered.LOWEST_PRECEDENCE-1)</font>为关键,意思是给aop定一个优先级,这个优先级为多少呢, Ordered.LOWEST_PRECEDENCE为最低的order,大家都知道,Order越低,优先级越高,事务管理器默认优先级为1,这个aop的order比最低的Order还要小一个单位,代表优先级是最高的。
注:事务管理器我是这么配置的:
<!-- 配置事务管理器,注意这里的dataSource和SqlSessionFactoryBean的dataSource要一致,不然事务就没有作用了 -->
<bean id="transactionManager" class="org.springframework.jdbc.datasource.DataSourceTransactionManager">
<property name="dataSource" ref="multipleDataSource"/>
</bean>
<!-- 用注解来实现事务管理 -->
<tx:annotation-driven transaction-manager="transactionManager"/>