Spring Boot+Hibernate：单次Repository调用多数据源插入方案咨询

针对你不想修改现有customerRepository.save(customer)调用、同时要向两个数据源插入数据的需求，以下是几个简便可行的方案：

方案1：AOP切面拦截（最轻量化） #

直接通过AOP拦截Repository的save方法，在主数据源写入完成后，自动同步到第二个数据源，完全不需要修改原有业务代码。

实现步骤 #

1. 配置第二个数据源对应的Repository（比如命名为CustomerSecondaryRepository，确保它绑定到第二个数据源的EntityManager/JdbcTemplate）
2. 编写AOP切面类，拦截save操作并触发二次写入：

@Aspect
@Component
public class DualDatasourceSyncAspect {
    private final CustomerSecondaryRepository secondaryRepo;

    // 构造注入第二个数据源的Repository
    public DualDatasourceSyncAspect(CustomerSecondaryRepository secondaryRepo) {
        this.secondaryRepo = secondaryRepo;
    }

    // 拦截CustomerRepository的save方法，返回结果后执行同步
    @AfterReturning(
        pointcut = "execution(* com.your.project.repository.CustomerRepository.save(..))",
        returning = "savedCustomer"
    )
    public void syncToSecondaryDatasource(Customer savedCustomer) {
        // 同步到第二个数据源，可根据需求添加事务控制或异步处理
        secondaryRepo.save(savedCustomer);
    }
}

注意事项 #

• 事务一致性：如果需要强一致，可将切面方法加入原有事务（通过@Transactional(propagation = Propagation.REQUIRED)）；若允许最终一致，可改用@Async异步执行，避免阻塞主流程。
• 对象冲突：如果是JPA，注意复制对象后再传入第二个Repository，避免同一个实体被两个EntityManager管理导致的异常。

方案2：自定义Repository实现类 #

如果是Spring Data JPA，可通过自定义Repository的实现类，在save方法内部同时操作两个数据源的EntityManager，完全封装写入逻辑。

实现步骤 #

1. 自定义Repository实现类，注入两个数据源的EntityManager：

public class CustomerRepositoryImpl implements CustomerRepository {
    private final EntityManager primaryEntityManager;
    private final EntityManager secondaryEntityManager;

    public CustomerRepositoryImpl(
        @Qualifier("primaryEntityManager") EntityManager primaryEntityManager,
        @Qualifier("secondaryEntityManager") EntityManager secondaryEntityManager
    ) {
        this.primaryEntityManager = primaryEntityManager;
        this.secondaryEntityManager = secondaryEntityManager;
    }

    @Override
    public <S extends Customer> S save(S customer) {
        // 写入主数据源
        primaryEntityManager.persist(customer);
        // 复制对象避免EntityManager上下文冲突，写入第二个数据源
        S secondaryCustomer = BeanUtils.copyProperties(customer, customer.getClass());
        secondaryEntityManager.persist(secondaryCustomer);
        return customer;
    }

    // 实现其他继承自JpaRepository的方法（若需要）
}

2. 在Repository接口上指定实现类（Spring Data会自动识别Impl后缀的实现类）：

@Repository
public interface CustomerRepository extends JpaRepository<Customer, Long> {
    // 自定义查询方法...
}

方案3：扩展AbstractRoutingDataSource（适配你的原有思路） #

如果你坚持用AbstractRoutingDataSource的路由逻辑，可扩展它实现多数据源写入：

核心思路 #

重写determineTargetDataSource()方法，当识别到write类型的查找键时，遍历所有目标数据源执行写入操作，最终返回主数据源供原有流程使用。结合JDBC拦截器或JPA的PostPersist事件实现实际写入逻辑。

简化实现示例 #

public class MultiTargetRoutingDataSource extends AbstractRoutingDataSource {
    private Map<Object, List<DataSource>> multiTargetDataSources;

    @Override
    protected DataSource determineTargetDataSource() {
        Object lookupKey = determineCurrentLookupKey();
        if ("write".equals(lookupKey)) {
            List<DataSource> dataSources = multiTargetDataSources.get(lookupKey);
            // 通过事务同步器在提交后触发多数据源写入
            TransactionSynchronizationManager.registerSynchronization(new TransactionSynchronization() {
                @Override
                public void afterCommit() {
                    // 此处获取当前待保存对象，遍历数据源执行写入逻辑
                }
            });
        }
        return super.determineTargetDataSource();
    }

    // 注入多目标数据源映射
    public void setMultiTargetDataSources(Map<Object, List<DataSource>> multiTargetDataSources) {
        this.multiTargetDataSources = multiTargetDataSources;
    }
}

这个方案相对复杂，适合需要统一路由逻辑的场景，否则优先考虑前两个方案。

内容的提问来源于stack exchange，提问作者Goutam Kundu