mybatis手动切换数据库_mybatis动态切换数据源

(#)背景:由于业务的需求，导致需要随时切换15个数据源，此时不能low逼的去写十几个mapper，所以想到了实现一个数据源的动态切换

首先要想重写多数据源，那么你应该理解数据源的一个概念是什么，DataSourceTransactionManager这个类就是spring中对于数据源的封装，其中DataSource做为

他的一个成员.接下来我们要介绍一下我们切换动态数据源需要使用的类,AbstractRoutingDataSource,先来看看这个类的源码

首先看看这几个变量，targetDataSources代表的就是备选的数据源了，用一个map存储，显然是为了在关键的时候快速的查找这些数据源，defaultTargetDataSource

就是我们在配置的时候一般都会制定一个默认的数据源就是它了，

程序运行的时候在加载配置文件的时候，首先会执行setTargetDataSources方法，这个方法会加载配置文件中配置的数据源，存储在上面说的targetDataSource中,

然后是设置setDefaultTergetDataSource，这个就是上面说的默认的数据源,

接下来会执行这个方法，在其中会将所有的数据源用来初始化resolvedDataSources,而当实际上和数据库产生交互的时候那么会调用到下面的方法:

如果看到这里大概你也知道怎么回事了，在我们实际和数据库产生交互的时候那么就需要使用到数据源了，那么这个时候我们只是需要重写这determineCurrentLookupKey

方法，而用这个方法具体干了什么呢？就是用它来寻找数据源啊，也就是说我们在mapper中的配置:

(#)下面看看代码上怎么来写

首先为了线程安全我们使用一个ThreadLocal来做存储

public class ContextHolder {

public static final String DATASOURCE_1="dataSource";

public static final String DATASOURCE_2="dataSource2";

private static final ThreadLocal context = new ThreadLocal();

public static void setConsumerType(String consumerType){

context.set(consumerType);

}

public static String getConsumerType(){

return context.get();

}

public static void clearConsumerType(){

context.remove();

}

}

public class DynamicDataSource extends AbstractRoutingDataSource {

@Override

protected Object determineCurrentLookupKey() {

return ContextHolder.getConsumerType();

}

}

其实剩下的怎么实现已经很简单了，基本思路就是使用一个动态代理，在执行这个方法的时候，我们动态的给ContextHoler设置值，我建议使用spring aop,或者如果

不是spring的项目，那么直接只用动态代理也是很好的我觉，下面贴出一些我的简陋的代码

@Aspect

@Component

public class DataSourceAspect {

@Pointcut("execution(* com.wang.route.DynamicPersonService.*(..))")

public void pointCut() {

}

@Before(value = "pointCut()")

public void before(JoinPoint joinPoint) {

ContextHolder.setConsumerTyp(“”);

}

}