使用spring动态切换数据源

原理&＃xff1a;主要是调用目标方法时&＃xff0c;注入不同的数据源&＃xff0c;从而实现切换&＃xff0c;即利用aop&＃xff0c;而aop的实现是用代理实现的

1&＃xff0c;给工程添加一个获取数据源的路由&＃xff0c;并给它两个不同的数据源

&＃64;Beanpublic ThreadLocalRountingDataSource threadLocalRountingDataSource() {ThreadLocalRountingDataSource dataSource&＃61;new ThreadLocalRountingDataSource();Map targetDataSources&＃61;new HashMap();DataSource master &＃61; master();//master 数据源DataSource slave &＃61; slave();//slave 数据源targetDataSources.put(DataSources.MASTER, master);targetDataSources.put(DataSources.SLAVE, slave);dataSource.setTargetDataSources(targetDataSources);dataSource.setDefaultTargetDataSource(master);return dataSource;}

2&＃xff0c;ThreadLocalRountingDataSource类主要是有一个决定数据源键的方法

public class ThreadLocalRountingDataSource extends AbstractRoutingDataSource {&＃64;Overrideprotected Object determineCurrentLookupKey() {return DataSourceTypeManager.get();}
}

3&＃xff0c;DataSourceTypeManager实际是一个对ThreadLocal调用的一个工具类&＃xff0c;主要是给当前线程设置期数据源的路由键&＃xff0c;代码如下&＃xff1a;

import com.example.demo.enums.DataSources;public class DataSourceTypeManager {private static final ThreadLocal dataSourceTypes &＃61; new ThreadLocal(){&＃64;Overrideprotected DataSources initialValue(){return DataSources.MASTER;}};public static DataSources get(){return dataSourceTypes.get();}public static void set(DataSources dataSourceType){dataSourceTypes.set(dataSourceType);}public static void reset(){dataSourceTypes.set(DataSources.MASTER);}
}

4&＃xff0c;编写一个aop类&＃xff0c;决定在哪个方法需要切换数据源

&＃64;Aspect // for aop
&＃64;Component // for auto scan
&＃64;Order(0) // execute before &＃64;Transactional
public class DataSourceInterceptor { &＃64;Pointcut("execution(public * com.example.demo.service..*.getUser(..))")public void dataSourceSlave(){};&＃64;Before("dataSourceSlave()")public void before(JoinPoint jp) {DataSourceTypeManager.set(DataSources.SLAVE);}
}

彩蛋&＃xff1a;

事务拦截器在啥时候注入了ioc容器&＃xff1f;

观察&＃64;EnableTransactionManagement&＃xff0c;它import了 TransactionManagementConfigurationSelector类&＃xff0c;然后给容器注入了

AutoProxyRegistrar和ProxyTransactionManagementConfiguration&＃xff0c;这里我们重点关注第二个类&＃xff0c;点进去发现如下代码

给&＃64;Transactional标注的方法添加了 TransactionInterceptor 拦截器