如何配置Atomikos实现单Spring Repository操作双异地同步数据库？

可以实现你需求的场景：通过注解标记服务方法，让同一个Spring Data JPA Repository在分布式事务场景下同时写入两个物理隔离的同构数据库，无注解时仅写入单库。核心思路是通过自定义事务注解+AOP切面+Atomikos分布式事务管理器的组合来实现，具体步骤如下：

1. 配置双XA数据源与事务管理器 #

• 为两个异地同构数据库分别配置AtomikosDataSourceBean，确保启用XA事务支持（需数据库本身支持XA协议）。
• 对应配置两个EntityManagerFactory，分别绑定主、备数据源，保持实体映射一致（因为是同构库）。
• 配置两类事务管理器：
  • 分布式事务管理器：使用JtaTransactionManager整合Atomikos的UserTransaction和TransactionManager，用于双库的两阶段提交。
  • 单库事务管理器：使用DataSourceTransactionManager绑定主数据源，用于普通单库操作。

2. 自定义分布式事务注解 #

创建一个标记性注解，用于标识需要触发双库写入的服务方法：

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface DistributedTransactional {
}

3. AOP切面实现双库同步逻辑 #

通过AOP拦截标记了@DistributedTransactional的服务方法，在分布式事务上下文内，让Repository的操作同时同步到两个数据库：

@Aspect
@Component
public class DistributedTxAspect {

    @Autowired
    private EntityManager primaryEntityManager;
    @Autowired
    private EntityManager secondaryEntityManager;

    @Autowired
    private JtaTransactionManager jtaTransactionManager;

    @Around("@annotation(com.yourpackage.DistributedTransactional)")
    public Object executeDualDbTx(ProceedingJoinPoint joinPoint) throws Throwable {
        // 开启全局分布式事务
        TransactionDefinition txDef = new DefaultTransactionDefinition();
        TransactionStatus txStatus = jtaTransactionManager.getTransaction(txDef);

        try {
            // 执行原方法（写入主库）
            Object result = joinPoint.proceed();

            // 将主库操作同步到备库
            syncEntityToSecondaryDb(result);

            // 提交全局事务
            jtaTransactionManager.commit(txStatus);
            return result;
        } catch (Exception e) {
            // 全局回滚
            jtaTransactionManager.rollback(txStatus);
            throw e;
        }
    }

    private void syncEntityToSecondaryDb(Object entityOrIterable) {
        if (entityOrIterable == null) return;

        if (entityOrIterable instanceof Iterable) {
            ((Iterable<?>) entityOrIterable).forEach(entity -> secondaryEntityManager.merge(entity));
        } else {
            secondaryEntityManager.merge(entityOrIterable);
        }
        secondaryEntityManager.flush();
    }
}

4. 事务路由与Repository使用 #

• 在需要双库写入的服务方法上添加@DistributedTransactional注解，该方法内调用Repository的save/saveAll等操作会先写入主库，再通过切面同步到备库，且由Atomikos保证两阶段提交的原子性。
• 未标记该注解的服务方法，默认使用单库事务管理器，Repository仅操作主库，不触发分布式事务。

关键注意事项 #

• 确保两个数据库结构完全一致，实体类的JPA映射能同时适配两个库，避免同步时出现映射错误。
• 数据库需支持XA协议：比如MySQL需启用XA存储引擎，PostgreSQL需安装对应的XA扩展。
• 性能优化：分布式事务会增加操作延迟，建议仅在必须同步的业务方法上使用@DistributedTransactional注解。
• 异常一致性：切面需保证任何一步操作失败时，Atomikos会触发全局回滚，确保双库数据一致。

内容的提问来源于stack exchange，提问作者Haris Hajdarevic