SpringBoot 集成 atomikos 实现分布式事务

<dependency>    <groupId>org.springframework.boot</groupId>    <artifactId>spring-boot-starter-jta-atomikos</artifactId></dependency>

@Configuration//表示通过 aop 框架暴露改代理对象，AopContext 能够访问@EnableAspectJAutoProxy(exposeProxy = true)@MapperScan(basePackages = "com.practice.thinkindynamicsource.多数据源动态实现.mapper" , sqlSessionTemplateRef = "sqlSessionTemplate")public class ApplicationConfig {}

@Configurationpublic class AtomikosConfig {
    @Bean(name = "userTransaction")    public UserTransaction userTransaction() throws SystemException {        UserTransactionImp userTransactionImp = new UserTransactionImp();        userTransactionImp.setTransactionTimeout(10000);        return userTransactionImp;    }
    @Bean(name = "atomikosTransactionManager", initMethod = "init", destroyMethod = "close")    public TransactionManager atomikosTransactionManager() {        UserTransactionManager userTransactionManager = new UserTransactionManager();        userTransactionManager.setForceShutdown(false);        return userTransactionManager;    }
    @Bean(name = "transactionManager")    @DependsOn({"userTransaction", "atomikosTransactionManager"})    public PlatformTransactionManager transactionManager() throws SystemException {        UserTransaction userTransaction = userTransaction();        TransactionManager atomikosTransactionManager = atomikosTransactionManager();        return new JtaTransactionManager(userTransaction, atomikosTransactionManager);    }
}

    @Bean    @ConfigurationProperties("spring.datasource.druid.master")    public DataSource masterDataSource(Environment env)    {        String prefix = "spring.datasource.druid.master.";        return getDataSource(env, prefix, MASTER);    }
    @Bean    @ConfigurationProperties("spring.datasource.druid.slave")    @ConditionalOnProperty(prefix = "spring.datasource.druid.slave", name = "enabled", havingValue = "true")    public DataSource slaveDataSource(Environment env)    {        String prefix = "spring.datasource.druid.slave.";        return getDataSource(env, prefix, SLAVE);    }    protected DataSource getDataSource(Environment env, String prefix, String dataSourceName)    {        Properties prop = build(env, prefix);//获取配置信息        AtomikosDataSourceBean ds = new AtomikosDataSourceBean();        ds.setXaDataSourceClassName("com.alibaba.druid.pool.xa.DruidXADataSource");        ds.setUniqueResourceName(dataSourceName);        ds.setXaProperties(prop);        return ds;    }
    @Bean(name = "dynamicDataSource")    @Primary    public DynamicDataSource dataSource(DataSource masterDataSource)    {        Map<Object, Object> targetDataSources = new HashMap<>();        targetDataSources.put(MASTER, masterDataSource);        setDataSource(targetDataSources, SLAVE, "slaveDataSource");        return new DynamicDataSource(masterDataSource, targetDataSources);    }

//下面是结合 AbstractRoutingDataSource 实现多数据源的代码，具体参考源码
    /**     * 设置数据源     *     * @param targetDataSources 备选数据源集合     * @param sourceName 数据源名称     * @param beanName bean名称     */    public void setDataSource(Map<Object, Object> targetDataSources, String sourceName, String beanName)    {        try        {            DataSource dataSource = SpringUtils.getBean(beanName);            targetDataSources.put(sourceName, dataSource);        }        catch (Exception e)        {        }    }

    public SqlSessionFactory createSqlSessionFactory(Environment env, DataSource dataSource) throws Exception    {//        String typeAliasesPackage = env.getProperty("mybatis.typeAliasesPackage");        String mapperLocations = env.getProperty("mybatis.mapperLocations");        String configLocation = env.getProperty("mybatis.config-location");//        typeAliasesPackage = setTypeAliasesPackage(typeAliasesPackage);        VFS.addImplClass(SpringBootVFS.class);
        final SqlSessionFactoryBean sessionFactory = new SqlSessionFactoryBean();        sessionFactory.setDataSource(dataSource);//        sessionFactory.setTypeAliasesPackage(typeAliasesPackage);        sessionFactory.setMapperLocations(resolveMapperLocations(StringUtils.split(mapperLocations, ",")));        sessionFactory.setConfigLocation(new DefaultResourceLoader().getResource(configLocation));        return sessionFactory.getObject();    }
    @Bean(name = "sqlSessionFactoryMaster")    public SqlSessionFactory sqlSessionFactoryMaster(Environment env, @Qualifier("masterDataSource") DataSource dataSource) throws Exception    {        return createSqlSessionFactory(env, dataSource);    }
    @Bean(name = "sqlSessionFactorySlave")    public SqlSessionFactory sqlSessionFactorySlave(Environment env, @Qualifier("slaveDataSource") DataSource dataSource) throws Exception    {        return createSqlSessionFactory(env, dataSource);    }
    @Bean(name = "sqlSessionTemplate")    public DynamicSqlSessionTemplate sqlSessionTemplate(@Qualifier("sqlSessionFactoryMaster") SqlSessionFactory factoryMaster,                                                        @Qualifier("sqlSessionFactorySlave") SqlSessionFactory factorySlave) throws Exception    {        Map<Object, SqlSessionFactory> sqlSessionFactoryMap = new HashMap<>();        sqlSessionFactoryMap.put(DruidConfig.MASTER, factoryMaster);        sqlSessionFactoryMap.put(DruidConfig.SLAVE, factorySlave);
        DynamicSqlSessionTemplate customSqlSessionTemplate = new DynamicSqlSessionTemplate(factoryMaster);        customSqlSessionTemplate.setTargetSqlSessionFactorys(sqlSessionFactoryMap);        return customSqlSessionTemplate;    }

public class DynamicSqlSessionTemplate extends SqlSessionTemplate {
    private final SqlSessionFactory sqlSessionFactory;    private final ExecutorType executorType;    private final SqlSession sqlSessionProxy;    private final PersistenceExceptionTranslator exceptionTranslator;    private Map<Object, SqlSessionFactory> targetSqlSessionFactorys;    private SqlSessionFactory defaultTargetSqlSessionFactory;

    @Override    public SqlSessionFactory getSqlSessionFactory()    {        SqlSessionFactory targetSqlSessionFactory = targetSqlSessionFactorys                .get(DynamicDataSourceContextHolder.getDataSourceType());        if (targetSqlSessionFactory != null)        {            return targetSqlSessionFactory;        }        else if (defaultTargetSqlSessionFactory != null)        {            return defaultTargetSqlSessionFactory;        }        return this.sqlSessionFactory;    }

    private class SqlSessionInterceptor implements InvocationHandler    {        public Object invoke(Object proxy, Method method, Object[] args) throws Throwable        {            final SqlSession sqlSession = getSqlSession(DynamicSqlSessionTemplate.this.getSqlSessionFactory(),                    DynamicSqlSessionTemplate.this.executorType, DynamicSqlSessionTemplate.this.exceptionTranslator);            try            {                Object result = method.invoke(sqlSession, args);                if (!isSqlSessionTransactional(sqlSession, DynamicSqlSessionTemplate.this.getSqlSessionFactory()))                {                    sqlSession.commit(true);                }                return result;            }            catch (Throwable t)            {                Throwable unwrapped = unwrapThrowable(t);                if (DynamicSqlSessionTemplate.this.exceptionTranslator != null                        && unwrapped instanceof PersistenceException)                {                    Throwable translated = DynamicSqlSessionTemplate.this.exceptionTranslator                            .translateExceptionIfPossible((PersistenceException) unwrapped);                    if (translated != null)                    {                        unwrapped = translated;                    }                }                throw unwrapped;            }            finally            {                closeSqlSession(sqlSession, DynamicSqlSessionTemplate.this.getSqlSessionFactory());            }        }    }}复制代码

    @Override    @Transactional    public Map testDataSource() {        SpringUtils.getAopProxy(this).insertA();        SpringUtils.getAopProxy(this).insertB();        int i=10/0;        return new HashMap();    }
    @DataSource(value = DataSourceType.slave)    public void insertA(){        dynamicTestMapper.dynamicTestA();    }
    @DataSource(value = DataSourceType.master)    public void insertB(){        dynamicTestMapper.dynamicTestB();    }