Spring动态切换数据源及事务
前段时间花了几天来解决公司框架ssm上事务问题。如果不动态切换数据源话，直接使用spring的事务配置，是完全没有问题的。由于框架用于各个项目的快速搭建，少去配置各个数据源配置xml文件等。采用了动态切换数据源方式。在解决问题的时候查看了相关源代码等。接下来对动态数据源切换、事务相关的核心源代码个分析总结，总结不到位，请谅解。
第一、实现动态切换数据源
思路大概如下：具体切换到哪个数据源通过包名来控制，写一个类实现使用spring提供MethodInterceper接口，再通过aop来切面到service，这样来确定我们需要的数据源。
我们通过在properties文件中配置相关数据连接配置：如：
在公用工程中只需简单配置sys系统库的数据源配置，因为任何工程都需要这个基础的数据源，如下图：
在properties文件中，我们配置了3个数据库的连接，其中sys系统库已经配置了，那么其他两个库数据源没有在xml中配置，我们可以通过启动项目后由spring给我们自动创建相关的bean，spring的强大之处哦。如下面这段代码是创建相关数据源bean的代码：
/**
* <Description>
* 动态生成其他数据源bean，并且注册到spring容器中。
*
* 实现spring ApplicationContextAware接口，spring在实例化此bean时候会自动调用setApplicationContext方法，
* 这样此bean就具有拿到容器，那么你想怎么搞就怎么搞了。
*
* 实现InitializingBean接口，我们知道在初始化bean的时候，会自动调用afterPropertiesSet方法，在spring源码中有大量
* 实现了此接口的类。可以看出spring的为我们提供了强大的扩展性
* @author 兰伟
* @CreateDate 2018年6月9日 上午12:07:36
* @since V1.0
*/
public class DataSources implements ApplicationContextAware, InitializingBean {
private static Logger logger = Logger.getLogger(DataSources.class);
public static String DEFAULT_DATASOURCE="ds.sys";
private ApplicationContext context;
@Override
public void setApplicationContext(ApplicationContext context)
throws BeansException {
this.context = context;
}
@Override
public void afterPropertiesSet() {
try {
regDynamicBean();
}
catch (Exception e) {
e.printStackTrace();
}
}
private void regDynamicBean() throws IOException{
//CustomPropertyConfigurer中读取了properties配置文件
Map<String, Object> pmap = CustomPropertyConfigurer.getctxPropertiesMap();
Map<String, DataSourceInfo> mapCustom = getDbSourceInfo(pmap);
// 把数据源bean注册到容器中
addSourceBeanToApp(mapCustom);
}
@SuppressWarnings("unchecked")
private void addSourceBeanToApp(Map<String, DataSourceInfo> mapCustom) {
DefaultListableBeanFactory acf = (DefaultListableBeanFactory) context.getAutowireCapableBeanFactory();
BeanDefinition beanDefinition;
Iterator<String> iter = mapCustom.keySet().iterator();
while(iter.hasNext()){
String dataSourceId = iter.next();
// 得到Bean定义，并添加到容器中
beanDefinition = new ChildBeanDefinition(DEFAULT_DATASOURCE);
// 注意：必须先注册到容器中，再得到Bean进行修改，否则数据源属性不能有效修改
acf.registerBeanDefinition(dataSourceId, beanDefinition);
// 再得到数据源Bean定义，并修改连接相关的属性
DruidDataSource cpds = (DruidDataSource)context.getBean( dataSourceId);
cpds.setUrl(mapCustom.get(dataSourceId).connUrl);
cpds.setUsername(mapCustom.get(dataSourceId).userName);
cpds.setPassword(mapCustom.get(dataSourceId).password);
cpds.setDriverClassName(mapCustom.get(dataSourceId).driverClass);
if(cpds.getDriverClassName().indexOf( "jtds" )!=-1||cpds.getDriverClassName().indexOf( "sqlserver" )!=-1){
cpds.setValidationQuery("select 1");
}else if( cpds.getDriverClassName().indexOf( "oracle" )!=-1 ){
cpds.setValidationQuery("SELECT 'x' FROM dual");
}
((Map<String, Object>) SpringUtils.getBean("targetDataSources")).put(dataSourceId, cpds);
}
}
public Map<String, DataSourceInfo> getDbSourceInfo(Map<String, Object> pmap,boolean includeSys) throws IOException {
Map<String, DataSourceInfo> mapDataSource = new HashMap<String,DataSourceInfo>();
String source = (String) pmap.get("db.source");
if(source==null || source.equals("jndi")){
return mapDataSource;
}
Matcher matcher;
Pattern pattern = Pattern.compile("^(\\w+\\.\\w+)\\.jdbc\\.(url|username|password|driverclass|validationQuery)$");
for(Map.Entry<String, Object> entry : pmap.entrySet()) {
String keyProp = entry.getKey();
String valueProp = entry.getValue().toString();
matcher = pattern.matcher(keyProp );
if(matcher.find()){
String dsName = matcher.group(1);
if(StringUtils.equals(dsName, DEFAULT_DATASOURCE) && !includeSys){
continue;
}
String dsPropName = matcher.group(2);
DataSourceInfo dsi;
if(mapDataSource.containsKey(dsName)){
dsi = mapDataSource.get(dsName);
}
else{
dsi = new DataSourceInfo();
}
// 根据属性名给数据源属性赋值
if("url".equals(dsPropName)){
dsi.connUrl = valueProp;
}else if("username".equals(dsPropName)){
dsi.userName = valueProp;
}else if("password".equals(dsPropName)){
dsi.password = valueProp;
}else if("driverclass".equals(dsPropName)){
dsi.driverClass = valueProp;
}else if("validationQuery".equals(dsPropName)){
dsi.validationQuery = valueProp;
}
mapDataSource.put(dsName, dsi);
}
}
return mapDataSource;
}
public Map<String, DataSourceInfo> getDbSourceInfo(Map<String, Object> pmap) throws IOException {
return getDbSourceInfo(pmap,false);
}
public class DataSourceInfo{
public String connUrl;
public String userName;
public String password;
public String driverClass;
public String validationQuery;
}
}
在上面我们已经实现了各个数据源。接下来就是根据包名来动态切换数据源的问题了。如何切换，我们最好不要在业务代码中来写什么DataSourceContextHolder.setDbType（”ds.sms”）这样的代码。我们可以通过spring提供MethodIntercepter接口，采用aop的方式来。
及spring的相关配置
Spring中提供AbstractRoutingDataSource抽象类，重写determineCurrentLookupKey方法，当需要查询数据的时候会自动切换到指定数据库，核心代码如下
DataSourceContextHolder核心代码，在这里有个局部线程变量，这是和线程绑定起来。我们在写代码的时候一定要注意，不要随意开启新线程，否则是不起作用的哦，除非重新在设置一把。
动态切换数据源主要工作已完成.
第二、接下来看看事务以及mybatis与spring整合核心配置
在配置中我们可以看到自己去实现了一套DynamicSqlSessionTemplate、DynamicSqlSessionFactoryBean、CustomDataSourceTransactionManager。这几个是修改后的。在之前的代码中，我们还是采用自带的SqlSessionTemplate、SqlSessionFactoryBean、DataSourceTransactionManager,如果使用之前的，在看下datasource的配置，这个是系统库，那么切换数据源后，两个DataSource就不一致的。这就会导致同一个线程中始终是无法获取到由spring管理的事务相关设置。所以想要保证事务的话，必须要datasource是同一个。所以自己就实现了一个DataSourceTransactionManager，修改了getDataSource方法等。
Mybatis与spring整合事务相关的核心源码可以查看
TransactionInterceptor事务拦截器，继承了TransactionAspectSupport类
TransactionAspectSupport
AbstractPlatformTransactionManager
SqlSessionUtils 有个核心方法getSqlSession，这个就需要从当前线程中去获取sqlSesson
DataSourceTransactionManager 事务管理 继承了AbstractPlatformTransactionManager。
SqlSessionFactoryBean
SqlSessionTemplate这里面有个核心SqlSessionInterceptor拦截器，其实也是个代理通过代理模式来。
这么几个核心类。
有篇博客转门对DataSourceTransactionManager 核心的几个类做了分析
做了源码分析https://www.cnblogs.com/chihirotan/p/6739748.html，大家可以参考下。
在解决问题之前没有找到他的这篇文章，害的我挨着撸了一把spring及mybatis-spring的这块的源码。如果早看到的话，估计时间会少花点。
