【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数据库连接。