springboot+jta+atomic实现多数据源分布式事务管理

引入依赖

<dependency>
	<groupId>mysql</groupId>
	<artifactId>mysql-connector-java</artifactId>
	<version>8.0.26</version>
</dependency>
<dependency>
    <groupId>org.postgresql</groupId>
    <artifactId>postgresql</artifactId>
    <version>42.2.23</version>
</dependency>
<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-jta-atomikos</artifactId>
    <version>2.5.4</version>
</dependency>

appliaction.properties配置两个数据源和jta

app.datasource.ecs.url=jdbc:postgresql://localhost:5432/test?useUnicode=true&characterEncoding=UTF-8
app.datasource.ecs.username=postgres
app.datasource.ecs.password=root
app.datasource.ecs.driverClassName=org.postgresql.Driver
app.datasource.ecs.validationQuery=select 'x'
app.datasource.ecs.testWhileIdle=true

app.datasource.bd.url=jdbc:mysql://127.0.0.1:3306/test?characterEncoding=utf8&useSSL=false&serverTimezone=UTC&allowPublicKeyRetrieval=true
app.datasource.bd.username=root
app.datasource.bd.password=root
app.datasource.bd.driverClassName=com.mysql.cj.jdbc.Driver
app.datasource.bd.validationQuery=select 'x'
app.datasource.bd.testWhileIdle=true

spring.jta.log-dir=classpath:tx-logs
spring.jta.transaction-manager-id=txManager
spring.jta.atomikos.properties.log-base-dir=./double-sql-log

创建两个数据源的配置文件

@Configuration
public class DataSourceConfig {
    @Bean
    @Primary
    @ConfigurationProperties("app.datasource.ecs")
    public DataSourceProperties ecsDataSourceProperties() {
        return new DataSourceProperties();
    }

    @Bean
    @Primary
    @ConfigurationProperties("app.datasource.ecs")
    public DataSource ecsDataSource() throws SQLException {
        PGXADataSource pgxaDataSource = new PGXADataSource();
        pgxaDataSource.setUrl(ecsDataSourceProperties().getUrl());
        pgxaDataSource.setPassword(ecsDataSourceProperties().getPassword());
        pgxaDataSource.setUser(ecsDataSourceProperties().getUsername());

        // 将本地事务注册到创 Atomikos全局事务
        AtomikosDataSourceBean xaDataSource = new AtomikosDataSourceBean();
        xaDataSource.setXaDataSource(pgxaDataSource);
        xaDataSource.setUniqueResourceName("ecsDataSource");

        return xaDataSource;
    }

    /*@Bean
    @Primary
    @ConfigurationProperties("app.datasource.ecs")
    public DataSource ecsDataSource() {
        return bdDataSourceProperties().initializeDataSourceBuilder().build();
    }*/

    @Bean
    @ConfigurationProperties("app.datasource.bd")
    public DataSourceProperties bdDataSourceProperties() {
        return new DataSourceProperties();
    }

    @Bean
    @ConfigurationProperties("app.datasource.bd")
    public DataSource bdDataSource() throws SQLException {
        MysqlXADataSource mysqlXaDataSource = new MysqlXADataSource();
        mysqlXaDataSource.setUrl(bdDataSourceProperties().getUrl());
        mysqlXaDataSource.setPinGlobalTxToPhysicalConnection(true);
        mysqlXaDataSource.setPassword(bdDataSourceProperties().getPassword());
        mysqlXaDataSource.setUser(bdDataSourceProperties().getUsername());
        mysqlXaDataSource.setPinGlobalTxToPhysicalConnection(true);

        // 将本地事务注册到创 Atomikos全局事务
        AtomikosDataSourceBean xaDataSource = new AtomikosDataSourceBean();
        xaDataSource.setXaDataSource(mysqlXaDataSource);
        xaDataSource.setUniqueResourceName("bdDataSource");
        return xaDataSource;
    }

    /*@Bean
    @ConfigurationProperties("app.datasource.bd")
    public DataSource bdDataSource() {
        return bdDataSourceProperties().initializeDataSourceBuilder().build();
    }*/
}

@Configuration
@MapperScan(basePackages = "priv.fjh.doublesql.repository.bd", sqlSessionFactoryRef = "sqlSessionFactoryBd")
public class MyBatisConfigBd {
    @Autowired
    private DataSourceConfig dataSourceConfig;

    @Bean("dataSourcBd")
    public DataSource dataSource() throws SQLException {
        return dataSourceConfig.bdDataSource();
    }

    /*@Bean("txManagerBd")
    public PlatformTransactionManager txManager(@Qualifier("dataSourcBd") DataSource dataSource) {
        return new DataSourceTransactionManager(dataSource);
    }*/

    @Bean("sqlSessionFactoryBd")
    public SqlSessionFactory sqlSessionFactory(@Qualifier("dataSourcBd") DataSource dataSource) throws Exception {
        final SqlSessionFactoryBean sqlSessionFactoryBean = new SqlSessionFactoryBean();
        sqlSessionFactoryBean.setDataSource(dataSource);
        sqlSessionFactoryBean.setMapperLocations(new PathMatchingResourcePatternResolver().getResources("mappers/mapper_bd/*.xml"));
        Resource configLocation = new ClassPathResource("mybatis-config.xml");
        sqlSessionFactoryBean.setConfigLocation(configLocation);
        return sqlSessionFactoryBean.getObject();
    }
}

@Configuration
@MapperScan(basePackages = "priv.fjh.doublesql.repository.ecs", sqlSessionFactoryRef = "sqlSessionFactoryEcs")
public class MyBatisConfigEcs {
    @Autowired
    private DataSourceConfig dataSourceConfig;

    @Bean("dataSourceEcs")
    public DataSource dataSource() throws SQLException {
        return dataSourceConfig.ecsDataSource();
    }


    /*@Bean("txManagerEcs")
    public PlatformTransactionManager txManager(@Qualifier("dataSourceEcs") DataSource dataSource) {
        return new DataSourceTransactionManager(dataSource);
    }*/

    @Primary
    @Bean("sqlSessionFactoryEcs")
    public SqlSessionFactory sqlSessionFactory(@Qualifier("dataSourceEcs") DataSource dataSource) throws Exception {
        final SqlSessionFactoryBean sqlSessionFactoryBean = new SqlSessionFactoryBean();
        sqlSessionFactoryBean.setDataSource(dataSource);
        sqlSessionFactoryBean.setMapperLocations(new PathMatchingResourcePatternResolver().getResources("mappers/mapper_ecs/*.xml"));
        Resource configLocation = new ClassPathResource("mybatis-config.xml");
        sqlSessionFactoryBean.setConfigLocation(configLocation);
        return sqlSessionFactoryBean.getObject();
    }
}

完成以上配置@Transactional注解就可以实现多数据源分布式事务管理了

额外知识点

从下图的运行结果可以发现：我们知道，同一个事务中同一个数据源的多次请求在同一个sqlSession中；但是同一个事务中多数据源的情况下，同数据源在同一个sqlSession中，不同数据源在不同sqlSession中。