SpringBoot集成jta-atomikos实现分布式事务

1、依赖

    org.springframework.boot
    spring-boot-starter-jta-atomikos

2、application.yml 配置双数据源

server:
  port: 8004
spring:
  application: jta-atomikos
  datasource:
    db1:
      driver-class-name: com.mysql.jdbc.Driver
      jdbc-url: jdbc:mysql://localhost:3306/orders?useSSL=false&serverTimezone=GMT%2b8&characterEncoding=utf8&connectTimeout=1000&socketTimeout=3000&autoReconnect=true
      username: root
      password: 123456
    db2:
      driver-class-name: com.mysql.jdbc.Driver
      jdbc-url: jdbc:mysql://localhost:3306/orderdetails?useSSL=false&serverTimezone=GMT%2b8&characterEncoding=utf8&connectTimeout=1000&socketTimeout=3000&autoReconnect=true
      username: root
      password: 123456

3、添加配置读取类

1、db1数据源配置读取类
@Configuration
@ConfigurationProperties(prefix = "spring.datasource.db1")
@Data
public class DB1Config {
    private  String username;
    private String password;
    private String jdbcUrl;
}

2、db2数据源配置读取类
@Configuration
@ConfigurationProperties(prefix = "spring.datasource.db2")
@Data
public class DB2Config {
    private  String username;
    private String password;
    private String jdbcUrl;
}

4、创建多数据源,管理事务

1、db1的数据源配置
@Configuration
@MapperScan(basePackages = {"com.jtaatomikos.mapper.db1"}, sqlSessionFactoryRef = "sqlSessionFactoryDb1")
public class DbSessionFactory1 {

    @Bean(name = "db1")
    public DataSource businessDbDataSource(@Qualifier("DB1Config") Db1Config db1Conf) {
        MysqlXADataSource mysqlXaDataSource = new MysqlXADataSource();
        mysqlXaDataSource.setUrl(db1Conf.getJdbcUrl());
        mysqlXaDataSource.setPassword(db1Conf.getPassword());
        mysqlXaDataSource.setUser(db1Conf.getUsername());
        mysqlXaDataSource.setPinGlobalTxToPhysicalConnection(true);

        AtomikosDataSourceBean xaDataSource = new AtomikosDataSourceBean();
        xaDataSource.setXaDataSource(mysqlXaDataSource);
        xaDataSource.setUniqueResourceName("db1");
        return xaDataSource;
    }

    @Bean(name="sqlSessionFactoryDb1")
    public SqlSessionFactory sqlSessionFactoryDb1(
@Qualifier("db1")DataSource DataSource) throws Exception {
        SqlSessionFactoryBean factoryBean = new SqlSessionFactoryBean();
        factoryBean.setDataSource(DataSource);
        return factoryBean.getObject();
    }

    @Bean(name="sqlSessionTemplateDb1")
    public SqlSessionTemplate sqlSessionTemplateDb1(
@Qualifier("sqlSessionFactoryDb1") SqlSessionFactory sqlSessionFactory) throws Exception {
        return new SqlSessionTemplate(sqlSessionFactory);
    }
}
2、db2数据源配置
@MapperScan(basePackages = {"com.jtaatomikos.mapper.db2"}, sqlSessionFactoryRef = "sqlSessionFactoryDb2")
public class DbSessionFactory2 {

    @Bean(name = "db2")
    public DataSource newhomeDbDataSource(@Qualifier("DB2Config") DB2Config db2Conf) {
        MysqlXADataSource mysqlXaDataSource = new MysqlXADataSource();
        mysqlXaDataSource.setUrl(db2Conf.getJdbcUrl());
        mysqlXaDataSource.setPassword(db2Conf.getPassword());
        mysqlXaDataSource.setUser(db2Conf.getUsername());
        mysqlXaDataSource.setPinGlobalTxToPhysicalConnection(true);

        AtomikosDataSourceBean xaDataSource = new AtomikosDataSourceBean();
        xaDataSource.setXaDataSource(mysqlXaDataSource);
        xaDataSource.setUniqueResourceName("db2");
        return xaDataSource;
    }


    @Bean(name="sqlSessionFactoryDb2")
    public SqlSessionFactory sqlSessionFactoryDb2(
@Qualifier("db2")DataSource DataSource ) throws Exception {
        SqlSessionFactoryBean factoryBean = new SqlSessionFactoryBean();
        factoryBean.setDataSource(DataSource);
        return factoryBean.getObject();
    }

    @Bean(name="sqlSessionTemplateDb2")
    public SqlSessionTemplate sqlSessionTemplateDb2(
@Qualifier("sqlSessionFactoryDb2") SqlSessionFactory sqlSessionFactory) throws Exception {
        return new SqlSessionTemplate(sqlSessionFactory);
    }
}

5、数据持久化

1、db1数据持久化操作
package com.jtaatomikos.mapper.db1;
@Mapper
public interface OrderMapper {
    @Insert("INSERT INTO orders(id, name) VALUES(#{id}, #{name})")
    int insert(@Param("id") String id, @Param("name") String name);
}
2、db2数据持久化操作
package com.jtaatomikos.mapper.db2;
@Mapper
public interface OrderDetailMapper {
    @Insert("INSERT INTO orderdetails(id, orderId) VALUES(#{id}, #{orderId})")
    int insert(@Param("id") int id, @Param("orderId") String orderId);
}

6、数据库DLL文件

create database orders;
use orders;
create table orders(
id varchar(40),
name varchar(25)
);

create database orderdetails;
use orderdetails;
create table orderdetails(
id int,
ordcerId  varchar(40)
);

7、模拟事务回滚

@Service
public class OrderService{

    @Resource
    private OrderMapper orderMapper;

    @Resource
    private OrderDetailMapper orderDetailMapper;

    @Transactional
    public String insertDbaAndDbB(boolean flag) {
        //实际生产环境订单id应该通关雪花算法生成(生产的id是数字类型,b+树索引查询效率会很高),uuid没有规律,会严重影响索引查找速度
        String orderId = UUID.randomUUID().toString();
        //插入订单
        orderMapper.insert(orderId, "重入门到放弃");
        //模拟报错,事务会回滚
        if(flag){
            //flag=true的时候,触发异常,事务回滚,为false的话,两个数据库都能正确插入数据
            System.out.println(1 / 0);
        }
        //插入订单明细
        orderDetailMapper.insert(1, orderId);
        return "success";
    }
}