如何拦截JPA事务的启动/提交/回滚以触发自定义行为？

要覆盖**注解式（@Transactional）和编程式（TransactionOperations.execute()）**两种事务场景，最可靠的方式是扩展Spring的PlatformTransactionManager——所有Spring管理的事务最终都会通过这个管理器执行操作，因此自定义它能实现全场景的事务拦截。

核心思路 #

通过包装或继承现有的JpaTransactionManager，重写事务启动、提交、回滚的核心方法，在这些节点记录时间戳并计算耗时，当超过阈值时输出警告日志。

具体实现步骤 #

1. 自定义带耗时监控的TransactionManager

创建一个继承JpaTransactionManager的子类，在事务启动时记录开始时间，提交/回滚时计算耗时并判断是否触发警告：

import org.springframework.orm.jpa.JpaTransactionManager;
import org.springframework.transaction.DefaultTransactionStatus;
import org.springframework.transaction.TransactionDefinition;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.time.Duration;
import java.time.Instant;

public class TimedJpaTransactionManager extends JpaTransactionManager {
    private static final Logger logger = LoggerFactory.getLogger(TimedJpaTransactionManager.class);
    // 定义耗时警告阈值（1秒）
    private static final long WARNING_THRESHOLD_MS = 1000;

    @Override
    protected void doBegin(Object transaction, TransactionDefinition definition) {
        super.doBegin(transaction, definition);
        // 将事务开始时间存入TransactionStatus的附属属性中
        DefaultTransactionStatus status = (DefaultTransactionStatus) getTransaction(definition);
        status.setAttribute("transactionStartTime", Instant.now());
    }

    @Override
    protected void doCommit(DefaultTransactionStatus status) {
        Instant startTime = (Instant) status.getAttribute("transactionStartTime");
        super.doCommit(status);
        logTransactionDuration(startTime, status, "提交");
    }

    @Override
    protected void doRollback(DefaultTransactionStatus status) {
        Instant startTime = (Instant) status.getAttribute("transactionStartTime");
        super.doRollback(status);
        logTransactionDuration(startTime, status, "回滚");
    }

    private void logTransactionDuration(Instant startTime, DefaultTransactionStatus status, String operation) {
        if (startTime == null) return;

        long durationMs = Duration.between(startTime, Instant.now()).toMillis();
        if (durationMs > WARNING_THRESHOLD_MS) {
            String transactionName = status.getTransactionDefinition().getName() != null
                    ? status.getTransactionDefinition().getName()
                    : "未命名事务";
            logger.warn("事务{}耗时过长: {}ms，事务名称: {}", operation, durationMs, transactionName);
        }
    }
}

2. 替换默认的TransactionManager

在Spring配置中注册自定义的事务管理器，替代默认的JpaTransactionManager：

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;

import javax.persistence.EntityManagerFactory;

@Configuration
public class TransactionConfig {

    @Primary
    @Bean
    public TimedJpaTransactionManager transactionManager(EntityManagerFactory entityManagerFactory) {
        TimedJpaTransactionManager transactionManager = new TimedJpaTransactionManager();
        transactionManager.setEntityManagerFactory(entityManagerFactory);
        // 按需配置其他属性（如JpaVendorAdapter、事务超时时间等）
        return transactionManager;
    }
}

方案优势 #

• 全场景覆盖：不管是@Transactional注解还是TransactionOperations.execute()发起的事务，都会经过自定义的管理器，不会遗漏任何场景。
• 低侵入性：无需修改业务代码，仅通过扩展Spring事务组件实现监控。
• 可扩展性：可以轻松扩展功能，比如记录事务隔离级别、只读属性，或者将耗时数据上报到监控系统。

内容的提问来源于stack exchange，提问作者Christophe L