【Java】mybatis用AbstractRoutingDataSource实现动态数据源时的事务管理

第一阶段

最近遇到一个用到多数据源的项目，它是通过实现AbstractRoutingDataSource类来动态切换数据源的。而当我想为这个项目加入事务时，发现程序报错。就是数据源切换失败。

经过探索后发现原因是spring默认的事务管理器DataSourceTransactionManager在开启事务时会缓存连接。如图



恰好我当时对分布式事务有点兴趣，所有我首先想到的是尝试通过改用 JtaTransactionManager来解决这个问题。最终我用的是jta中无需服务器支持的分布式事务组件 Atomikos 。配置就不说了，网上一大堆，我就说下我遇到的问题。

第二阶段

接下来问题具体是出在SpringManagedTransaction类，它是Mybatis的Transaction接口的实现，接口的作用是封装一个数据库连接并处理连接生命周期，包括：创建，准备，提交/回滚和关闭。

在SpringManagedTransaction打开一个数据库连接的时候它会调用DataSourceUtils数据源工具类的getConnection(this.dataSource);方法，在这里会把AbstractRoutingDataSource的对象作为参数传进去。



而当运行到doGetConnection()方法时，他会根据为每个数据源缓存通过这个数据源获得的连接。但是AbstractRoutingDataSource虽然可以动态切换数据源，但是它表现出来的只是一个数据源，所有这个也会导致数据源切换失败。



解决办法就是自己实现一个Mybatis的事务工厂

public class DynamicTransactionFactory implements TransactionFactory {

public DynamicTransactionFactory( DynamicDataSourceHolder holder) {
this.holder = holder;
}

@Override
public void setProperties(Properties props) {

}

@Override
public Transaction newTransaction(Connection conn) {
throw new UnsupportedOperationException("New Spring transactions require a DataSource");
}

@Override
public Transaction newTransaction(DataSource dataSource, TransactionIsolationLevel level, boolean autoCommit) {
// 必须保证 AbstractRoutingDataSource的实现类也是用同一个数据源容器对象
return new SpringManagedTransaction(CustomerContextHolder.getCustomerType());//从容器中获取当前切换到的数据源
}
}

这样每个事务对象就不是通过同一个AbstractRoutingDataSource来构建，而是和AbstractRoutingDataSource的实现类一样通过数据源容器来动态生成。如下

public class DynamicDataSource extends AbstractRoutingDataSource {

@Override
protected Object determineCurrentLookupKey() {
return CustomerContextHolder.getCustomerType();
}

}

public class CustomerContextHolder {
//用ThreadLocal来设置当前线程使用哪个dataSource
private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>();

public static void setCustomerType(String customerType) {
contextHolder.set(customerType);
}
public static String getCustomerType() {
return contextHolder.get();
}

public static void clearCustomerType() {
contextHolder.remove();
}
}

第三阶段

最后也是最重要的普通的数据源类的对象是不支持分布式事务的。所以即使你解决了所有分布式事务下数据源切换的问题，但如果你没使用AtomikosDataSourceBean（它也是DataSource接口的实现类）来构建数据源对象，而使用普通的数据源类的话。即使你的程序运行没有报任何错误，事务的回滚操作也会失败。原因是普通的数据源类不提供XA数据库连接。