如何在 Spring Boot 中实现多数据源的事务管理？

在 Spring Boot 中实现多数据源的事务管理可以通过以下几种方式：

一、使用编程式事务管理

1. 配置多个数据源

   • 如同前面提到的，在 application.properties 或 application.yml 文件中配置多个数据源的连接信息，并创建对应的数据源 bean。

2. 手动开启和提交事务

   • 在需要使用多数据源事务的方法中，手动获取事务管理器并开启事务，执行数据库操作后提交事务。如果出现异常，则回滚事务。
    

   例如：

@Service
public class MultiDataSourceService {

    @Autowired
    private DataSourceOne dataSourceOne;

    @Autowired
    private DataSourceTwo dataSourceTwo;

    @Autowired
    private PlatformTransactionManager transactionManagerOne;

    @Autowired
    private PlatformTransactionManager transactionManagerTwo;

    public void multiDataSourceOperation() {
        TransactionStatus statusOne = transactionManagerOne.getTransaction(new DefaultTransactionDefinition());
        TransactionStatus statusTwo = transactionManagerTwo.getTransaction(new DefaultTransactionDefinition());

        try {
            // 对数据源一进行操作
            JdbcTemplate jdbcTemplateOne = new JdbcTemplate(dataSourceOne);
            jdbcTemplateOne.update("INSERT INTO table1...");

            // 对数据源二进行操作
            JdbcTemplate jdbcTemplateTwo = new JdbcTemplate(dataSourceTwo);
            jdbcTemplateTwo.update("INSERT INTO table2...");

            transactionManagerOne.commit(statusOne);
            transactionManagerTwo.commit(statusTwo);
        } catch (Exception e) {
            transactionManagerOne.rollback(statusOne);
            transactionManagerTwo.rollback(statusTwo);
            throw e;
        }
    }
}

二、使用声明式事务管理

1. 配置事务管理器
   • 创建多个事务管理器 bean，分别对应不同的数据源。

@Configuration
public class TransactionConfig {

    @Bean(name = "transactionManagerOne")
    public PlatformTransactionManager transactionManagerOne(@Qualifier("dataSourceOne") DataSource dataSourceOne) {
        return new DataSourceTransactionManager(dataSourceOne);
    }

    @Bean(name = "transactionManagerTwo")
    public PlatformTransactionManager transactionManagerTwo(@Qualifier("dataSourceTwo") DataSource dataSourceTwo) {
        return new DataSourceTransactionManager(dataSourceTwo);
    }
}

1. 使用 @Transactional 注解
   • 在需要进行事务管理的方法上添加 @Transactional 注解，并指定事务管理器。

@Service
public class MultiDataSourceService {

    @Autowired
    private DataSourceOne dataSourceOne;

    @Autowired
    private DataSourceTwo dataSourceTwo;

    @Autowired
    @Qualifier("transactionManagerOne")
    private PlatformTransactionManager transactionManagerOne;

    @Autowired
    @Qualifier("transactionManagerTwo")
    private PlatformTransactionManager transactionManagerTwo;

    @Transactional("transactionManagerOne")
    public void operationOnDataSourceOne() {
        // 对数据源一进行操作
        JdbcTemplate jdbcTemplateOne = new JdbcTemplate(dataSourceOne);
        jdbcTemplateOne.update("INSERT INTO table1...");
    }

    @Transactional("transactionManagerTwo")
    public void operationOnDataSourceTwo() {
        // 对数据源二进行操作
        JdbcTemplate jdbcTemplateTwo = new JdbcTemplate(dataSourceTwo);
        jdbcTemplateTwo.update("INSERT INTO table2...");
    }
}

三、注意事项

1. 异常处理：确保在事务方法中正确处理异常，以便事务能够正确回滚。
2. 事务传播行为：根据实际需求设置 @Transactional 注解的事务传播行为，例如 REQUIRED、REQUIRES_NEW 等。
3. 数据源切换：在多数据源环境下，要注意在事务方法中正确切换数据源，以确保操作在正确的数据源上执行。

通过以上方法，你可以在 Spring Boot 中实现多数据源的事务管理，确保数据的一致性和完整性。