mybatis多数据库切换,(动态数据源)。
项目中将一个库的某些标的某些数据保存到另一个库。
使用spring的aop编程动态切换数据源,代码如下,以备下次用到!
1.先将两个数据库连接,创建两个数据源,交于spring管理!
<bean id="dataSource" 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.user}" /> <property name="password" value="${jdbc.password}" /> </bean> <bean id="dataSource2" 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.user}" /> <property name="password" value="${jdbc.password}" /> </bean>
2.将动态切换数据源的类交给spring
<bean id="dynamicDataSource" class="cn.**.**.common.db.DynamicDataSource">
<property name="targetDataSources">
<map key-type="java.lang.String">
<!-- 指定lookupKey和与之对应的数据源 -->
<entry key="dataSource" value-ref="dataSource"></entry>
<entry key="dataSource2" value-ref="dataSource2"></entry>
</map>
</property>
<!-- 这里可以指定默认的数据源 -->
<property name="defaultTargetDataSource" ref="dataSource" />
</bean>
3.创建数据源切换的类,必须继承 AbstractRoutingDataSource 这个类(数据源动态切换必须继承该类)
public class DynamicDataSource extends AbstractRoutingDataSource { @Override protected Object determineCurrentLookupKey() { // 从自定义的位置获取数据源标识 return DynamicDataSourceHolder.getDataSource(); } }
4.将数据源放进线程中,避免相互干扰
public class DynamicDataSourceHolder { /** * 注意:数据源标识保存在线程变量中,避免多线程操作数据源时互相干扰 */ private static final ThreadLocal<String> THREAD_DATA_SOURCE = new ThreadLocal<String>(); public static String getDataSource() { String dataSource = THREAD_DATA_SOURCE.get(); if(StringUtils.isEmpty(dataSource)){ return DataSourceKey.DATA_SOURCE; }else{ return dataSource; } } public static void setDataSource(String dataSource) { THREAD_DATA_SOURCE.set(dataSource); } public static void clearDataSource() { THREAD_DATA_SOURCE.remove(); } }
5.数据源的key
DataSourceKey 这个类是多个数据源在spring管理中的name标识,可通过key取得数据源。
6.设置默认的数据源。
@Target({ElementType.TYPE, ElementType.METHOD}) @Retention(RetentionPolicy.RUNTIME) @Documented public @interface DataSource { /** * 默认是某个数据源 */ String value() default DataSourceKey.DATA_SOURCE; }
7.根据spring的aop在执行之前将数据源放进线程中,动态切换数据源。
@Aspect @Component @Order(0) public class DataSourceAspect{ private Logger logger = Logger.getLogger(getClass()); @Pointcut("execution(public * cn.**.**.service..*Service.*(..))") public void dataSourceAspect(){}; /** * 拦截目标方法,获取由@DataSource指定的数据源标识,设置到线程存储中以便切换数据源 */ @Before("dataSourceAspect()") public void before(JoinPoint point) throws Exception{ Class<?> target = point.getTarget().getClass(); MethodSignature signature = (MethodSignature) point.getSignature(); // 默认使用目标类型的注解,如果没有则使用其实现接口的注解 for (Class<?> clazz : target.getInterfaces()) { resolveDataSource(clazz, signature.getMethod()); } resolveDataSource(target, signature.getMethod()); } private void resolveDataSource(Class<?> clazz, Method method) { try { Class<?>[] types = method.getParameterTypes(); // 先使用类型注解 if (clazz.isAnnotationPresent(DataSource.class)) { DataSource source = clazz.getAnnotation(DataSource.class); DynamicDataSourceHolder.setDataSource(source.value()); }else{ DynamicDataSourceHolder.clearDataSource(); } // 方法注解可以覆盖类型注解 Method m = clazz.getMethod(method.getName(), types); if (m != null && m.isAnnotationPresent(DataSource.class)) { DataSource source = m.getAnnotation(DataSource.class); DynamicDataSourceHolder.setDataSource(source.value()); } logger.info("选择连接加载的DB :"+DynamicDataSourceHolder.getDataSource()); } catch (Exception e) { //如果出现异常不能断了所有连接,默认还是某个数据库连接 logger.error("选择连接加载DB ERROR "+clazz+":" +e.getMessage()); } } }
8.service层动态切换
默认的数据源不用标明,另一个数据源用@DataSource(DataSourceKey.LYD_DATA_SOURCE) 在service上标明即可。
9.另一个库的查询也可用@Results注解封装返回的实体(或map),sql用@Select注解
10.在多数据源切换中,不要在一个事务方法里面用到两个数据源,不然会切换不过去。因为事务一般定义在service层,所以如果用到两个数据源,可以将两个数据源放在controller进行。