mybatis 多数据源 以及 多数据源切换

mqzlp

　　最近做流程遇到需求：
　　一个流程引擎的数据源，所有流程数据都在此数据源
　　针对多个系统多个 用户相关数据源
　　流程中只保存  用户id,name信息，其数据来源都是从各个用户数据源取得的。
　　流程引擎配置不同系统的人员，部门等配置都需要切换用户数据源来取得，而流程引擎的数据源应该一直保持。
　　所以解决方案应该为： 
　　两套sqlSessionFactory          一个专门针对流程引擎使用
　　一个专门用于不同用户数据源
　　sqlsessionFactory1---->datasource_engine
　　sqlsessionFactory2 ------> datasource_business -------->(切换) userDataSource1,userDataSource2
　　1：实现数据源切换
　　DynamicDataSourceHolder.java

/**
* 切源数据源实现类
*
*/
public class DynamicDataSourceHolder {
private static final ThreadLocal<String> local = new ThreadLocal<String>();
/**
* 清空当前数据源
*/
public static void clear(){
local.remove();
}
/**
* 获取当前数据源
*/
public static String get(){
return local.get();
}
/**
* 设置当前数据源
*/
public static void set(String source){
local.set(source);
}
}
　　DynamicDataSource.java

import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
public class DynamicDataSource extends AbstractRoutingDataSource {
@Override
protected Object determineCurrentLookupKey() {
//返回当前数据源
return DynamicDataSourceHolder.get();
}
}

　　spring配置

<bean id="userDataSource1" class="org.springframework.jndi.JndiObjectFactoryBean">
<property name="jndiName" value="java:comp/env/jdbc/userDataSource1"></property>
</bean><!-- 用户数据源1  -->
<!-- 使用jndi数据源需要在web容器配置如下：
<Resource auth="Container" driverClassName="oracle.jdbc.OracleDriver"
maxActive="10" maxIdle="5" maxWait="15000" name="jdbc/userDataSource1"
password="123456" type="javax.sql.DataSource" url="jdbc:oracle:thin:@localhost:1521:ORCL"
username="user" />
-->
<bean id="userDataSource2" class="org.springframework.jndi.JndiObjectFactoryBean">
<property name="jndiName" value="java:comp/env/jdbc/userDataSource2"></property>
</bean><!-- 用户数据源2 -->
<bean id="dataSource_business" class="com.xxx.DynamicDataSource">
<property name="targetDataSources">
<map key-type="java.lang.String">
<entry key="userDataSource1" value-ref="userDataSource1"></entry>
<entry key="userDataSource2" value-ref="userDataSource2"></entry>
</map>
</property>
<property name="defaultTargetDataSource" ref="userDataSource1"></property>
</bean><!-- 自定义用户路由数据源 -->
<bean id="dataSource_engine" class="org.springframework.jndi.JndiObjectFactoryBean">
<property name="jndiName" value="java:comp/env/jdbc/engine"></property>
</bean><!-- 流程引擎数据源 -->
　　使用此种方式  切源数据源的话只需要设置代码：DynamicDataSourceHolder.set(“userDataSource1”);//将用户数据源切换为key为userDataSource1的数据源
　　如何动态切换呢？
　　可以使用aop来实现     aop拦截相应方法      动态获取key,更改数据源
　　比如
　　DataSourceAopProxy.java

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.aspectj.lang.annotation.Pointcut;
import org.springframework.stereotype.Service;
/**
*  aop实现动态切换数据源
*
*/
@Aspect
@Service
public class DataSourceAopProxy {
@Pointcut("execution(* com.xxx.*.*(..))")
private void setDataSource(JoinPoint jp){
//此处进行切换数据源   根据业务条件   获取数据源id
String dataSourceName = "userDataSource1";
DynamicDataSourceHolder.set(dataSourceName);
}
//@Before("setDataSource()")
//public void setDataSourceBefore(){
//String dataSourceName = "before";
//DynamicDataSourceHolder.set(dataSourceName);
//}
//
//
//@After("setDataSource()")
//public void setDataSourceAfter(JoinPoint jp){
//String dataSourceName = "after";
//DynamicDataSourceHolder.set(dataSourceName);
//}
}
　　不过上面没有具体获取数据源key的方式:
　　我的想法是         1 从切面的方法参数中取  （不过好像只能取来参数值，而取不到参数名称，再者也不一定之前写的方法有那个key参数，如果加上的话改动太大太累）
　　2 从session中取。针对各系统用户请求，可以知道系统id，也就可以知道key了。 针对管理员的话，对某个系统管理就需要将sesssion中的key改成此系统的
　　而且管理员一次只能管理一个系统，而不能管理多个（不同管理员，session不同，可以管理不同系统）
　　这种方式的话需要把session放入ThreadLocal中，在切面中取session也从threadLocal中取。不过总感觉这个方式太搓
　　暂时没有别的想法了。
　　2.：两套sqlSessionFactory实现
　　遇到问题： 系统中用户相关和流程引擎相关实体，mapper都放在一起，分包也不够清晰，做两套sqlSessionFactory的话扫包的basePackage 需要分包
　　分包的话各个service引用mapper的地方都要改，工作量太大
　　解决：     查看资料发现可以使用自定义注解来取代spring的repository设置到annotationClass中。这样每个mapper文件中注入相应的注解@EngineRepository，@BusinessRepository
　　<property name="annotationClass" value="org.springframework.stereotype.Repository"></property>
　　改成<property name="annotationClass" value="com.xxx.EngineRepository"></property>
　　配置文件如下：

<bean id="sqlSessionFactory_engine" class="org.mybatis.spring.SqlSessionFactoryBean">
<property name="configLocation" value="classpath:mybatis_engine.xml"></property>
<property name="dataSource" ref="dataSource_engine"></property>
<property name="databaseIdProvider" ref="databaseIdProvider"></property>
</bean><!-- 流程引擎所用sqlSessionFactory -->
<bean id="sqlSessionFactory_business" class="org.mybatis.spring.SqlSessionFactoryBean">
<property name="configLocation" value="classpath:mybatis_business.xml"></property>
<property name="dataSource" ref="dataSource_business"></property>
<property name="databaseIdProvider" ref="databaseIdProvider"></property>
</bean><!-- 用户相关sqlSessionFactory -->
<bean id="vendorProperties" class="org.springframework.beans.factory.config.PropertiesFactoryBean">
<property name="properties">
<props>
<prop key="SQL SERVER">sqlserver</prop>
<prop key="DB2">db2</prop>
<prop key="Oracle">oracle</prop>
</props>
</property>
</bean>
<bean id="databaseIdProvider" class="org.apache.ibatis.mapping.VendorDatabaseIdProvider">
<property name="properties" ref="vendorProperties"></property>
</bean>

<bean id="scanner_engine" class="org.mybatis.spring.MapperScannerConfigurer">
<property name="annotationClass" value="com.xxx.EngineRepository"></property>
<property name="basePackage" value="com.xxx.*.model"></property>
<property name="sqlSessionFactoryBeanName" value="sqlSessionFactory_engine"></property>
</bean><!-- 流程引擎扫mapper配置 -->
<bean id="scanner_business" class="org.mybatis.spring.MapperScannerConfigurer">
<property name="annotationClass" value="com.xxx.BusinessRepository"></property>
<property name="basePackage" value="com.xxx.*.model"></property>
<property name="sqlSessionFactoryBeanName" value="sqlSessionFactory_business"></property>
</bean><!-- 用户相关扫mapper配置 -->
　　如上所配置使用流程引擎的mapper文件就加上注解@EngineRepository 

// 使用dataSource_engine 数据源
@EngineRepository
public interface TaskAssignMapper {
TaskAssign getTaskAssignByDeployIdAndTaskKey(String deployId,String taskKey);
...
}
// 使用 dataSource_business 数据源
@BusinessRepository
public interface UserDaoMapper {
List<User> getAllUser();
...
}

public @interface EngineRepository {
}
public @interface BusinessRepository {
}