关于spring动态切换数据源

项目中有一个需求是要把测试的数据导入到生产环境，导入前要检查生产中是否 有该数据，这就需要多配置一个生产数据源，在查询生产的时候切换过来，不用的时候再切换回来，这就需要用到动态数据源的技术了。

       动态切换数据源的例子网上有很多，我也是参考了很多，按照我的需求从中提取了我需要的东西。但是中间遇到了些问题，最终定位到是和事物有关的（本人对事物理解不是太深，导致搞了好几天才定位到！）

       那么是什么问题呢，就是明明调用了切换数据源的方法但是就是没切换过来！感觉和网上写的一模一样呀！，很头大，然后试了很多办法，比如用AspectJ注解实现aop，以及xml配置aop通知，下面都会介绍到，其实方法都是对的，就是事物配置的问题，下面我们进入正题。

本文中所用的是SSH（spring+springmvc+Hibernate）Mybatis是否管用还没验证！

        1.创建数据源操作类

可以让我们手动设置我们当前想要的数据源（当然我们要用aop去设置，不用手动），sourceDS是我测试库的数据源，shengchanDS是生产数据库的数据源

package com.ecps.ssh.util;public class DataSourceHolder {public static final String sourceDS = "sourceDS";public static final String shenchanDs = "shengchanDS";//线程本地环境private static final ThreadLocal<String> dataSources = new ThreadLocal<String>();//设置数据源public static void setDataSource(String customerType){dataSources.set(customerType);}//获取数据源public static String getDataSource(){return dataSources.get();}//清除数据源public static void clearDataSource(){dataSources.remove();}}

      2.创建动态数据源类

用上面数据源操作类将目前线程中的数据源名字给determineCurrentLookupKey这个方法，它就可以改变数据源的方法

package com.ecps.ssh.util;import java.sql.SQLFeatureNotSupportedException;import java.util.logging.Logger;import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;public class DynamicDataSource extends AbstractRoutingDataSource{public Logger getParentLogger() throws SQLFeatureNotSupportedException {return null;}@Overrideprotected Object determineCurrentLookupKey() {return DataSourceHolder.getDataSource();}}

    3.注解方式实现aop通知

创建@Aspect注解的aop切面类

要使用@Aspect注解必须在applicationContext.xml中配置

<!-- 支持 @AspectJ 标记--><aop:aspectj-autoproxy expose-proxy="true"/>

expos-proxy="true" 是本项目特殊需求必须要的属性，它支持在同一个service直接的方法调用也能触发事物，例如

public class  UserServiceImpl  implements UserService{

public void findA(){

............

       }

     public void queryB(){ 

...........

        this.((UserService)AopContext.currentProxy()).findA();//只有配置了expos-proxy="true"才能这么写，并且触发配置给a方法的切面通知(@Before、@After)

      }

}

import org.aspectj.lang.annotation.After;import org.aspectj.lang.annotation.Aspect;import org.aspectj.lang.annotation.Before;import org.springframework.core.annotation.Order;@Aspect@Order(1)public class AspectJAdvice {@Before("execution(* com.ecps.ssh.service..*.checkSql(..))")public void before(){DataSourceHolder.setDataSource(DataSourceHolder.shenchanDs);//调用方法前就设置为生产数据源}@After("execution(* com.ecps.ssh.service..*.checkSql(..))")public void after(){DataSourceHolder.clearDataSource();//调用后清除数据源，系统然后就调用默认数据源了}}

4.xml配置的方式实现aop通知

<!-- 支持 @AspectJ 标记--><aop:aspectj-autoproxy expose-proxy="true"/><!-- 以AspectJ方式 定义 AOP execution(* com.wu.service..*.*(..))--><bean id="AopAdvice" class="com.ecps.ssh.util.AopAdvice"/> <aop:config proxy-target-class="true" > <aop:advisor pointcut="execution(* com.ecps.ssh.service..*.*(..))" advice-ref="txAdvice" order="2"/> <aop:aspect ref="aspectJAdvice" order="1"><aop:pointcut expression="execution(* com.ecps.ssh.service..*.checkSql(..))" id="shengchanPointCut"/><aop:before method="before" pointcut-ref="shengchanPointCut"/><aop:after method="after" pointcut-ref="shengchanPointCut"/></aop:aspect></aop:config> 

这种方式需要创建通知类，就是上面配置中的AopAdvice类

package com.ecps.ssh.util;public class AopAdvice {public void before(){DataSourceHolder.setDataSource(DataSourceHolder.shenchanDs);System.out.println("=====切换为生产数据源");}public void after(){DataSourceHolder.clearDataSource();System.out.println("=====清除生产数据源");}}

这样就是我们的实现方法，上面的1-2-3  和1-2-4两种都可以

下面讲讲遇到的事物问题

一般的事物配置如下

<tx:advice id="txAdvice"><tx:attributes><tx:method name="find*"   read-only="true"   propagation="SUPPORTS" /><tx:method name="get*"    read-only="true"   propagation="SUPPORTS" /><tx:method name="update*" propagation="REQUIRED" /><tx:method name="delete*" propagation="REQUIRED" /><tx:method name="add*"    propagation="REQUIRED" /><tx:method name="check*"    propagation="REQUIRED" /><tx:method name="*"       read-only="false" propagation="REQUIRED" /></tx:attributes></tx:advice> 

如果我们的queryB方法调用findA方法，queryB方法是对应上面的*，意思就是要开启一个新事物，然后queryB方法再调用findA方法，findA方法发现已经有事物了，那么它就不开启新事物，所以就没有切换成功。

上面只是个例子，要说明的就是，切换数据源是需要一个新的事物！