@Transactional 之TransactionAspectSupport与TransactionManager：如何在不抛出异常的情况下实现事务回滚

在 Java 开发中，尤其是使用 Spring 框架时，@Transactional 注解是我们管理事务的利器。它可以帮助我们轻松地声明式地管理数据库事务。然而，默认情况下，Spring 的事务管理机制只会在方法抛出异常时触发回滚。如果我们希望在某些特定场景下，即使方法没有抛出异常，也能主动触发事务回滚，应该如何实现呢？本文将深入探讨 @Transactional 的工作原理，并提供实现这一需求的解决方案。

@Transactional 的默认行为

在 Spring 中，@Transactional 注解通过 AOP（面向切面编程）实现事务管理。它的默认行为是：

提交事务：如果方法正常执行完成（没有抛出异常），Spring 会自动提交事务。
回滚事务 ：如果方法抛出了未捕获的 RuntimeException 或 Error，Spring 会回滚事务。

这种默认行为适用于大多数场景，但有时我们需要在方法逻辑中主动决定是否回滚事务。例如，当业务逻辑检测到一个条件不满足时，即使没有异常抛出，我们也希望回滚之前的操作。

问题场景

假设我们有一个简单的服务方法，用于保存用户信息：

java 复制代码

@Service
public class UserService {

    @Autowired
    private UserRepository userRepository;

    @Transactional
    public void saveUser(String username) {
        userRepository.save(new User(username));
        if ("invalid".equals(username)) {
            // 检测到非法输入，希望回滚事务
            // 但这里不抛出异常
        }
    }
}

在这个例子中，如果 username 是 "invalid"，我们希望回滚事务，但又不想通过抛出异常来实现。默认情况下，Spring 不会回滚事务，因为方法正常结束。那么，如何解决这个问题呢？

实现方法：主动触发回滚

Spring 提供了 TransactionStatus 和 TransactionAspectSupport 来支持手动控制事务状态。以下是两种实现"方法不抛出异常也回滚"的方案：

方案一：使用 TransactionStatus.setRollbackOnly()

我们可以注入 TransactionManager，然后通过 TransactionStatus 对象手动设置回滚标志：

java 复制代码

@Service
public class UserService {

    @Autowired
    private UserRepository userRepository;

    @Autowired
    private PlatformTransactionManager transactionManager;

    @Transactional
    public void saveUser(String username) {
        TransactionStatus status = transactionManager.getTransaction(new DefaultTransactionDefinition());
        userRepository.save(new User(username));
        if ("invalid".equals(username)) {
            transactionManager.rollback(status); // 手动回滚
            return;
        }
        transactionManager.commit(status); // 手动提交
    }
}

缺点：这种方式需要手动管理事务的提交和回滚，失去了 @Transactional 声明式管理的简洁性，因此不推荐在大多数情况下使用。

方案二：使用 TransactionAspectSupport

更优雅的方式是利用 Spring 提供的 TransactionAspectSupport 类，它允许我们在 @Transactional 方法中直接控制当前事务的状态：

java 复制代码

@Service
public class UserService {

    @Autowired
    private UserRepository userRepository;

    @Transactional
    public void saveUser(String username) {
        userRepository.save(new User(username));
        if ("invalid".equals(username)) {
            TransactionAspectSupport.currentTransactionStatus().setRollbackOnly();
        }
    }
}

工作原理

TransactionAspectSupport.currentTransactionStatus() 获取当前事务的状态。
setRollbackOnly() 将事务标记为"仅回滚"，告诉 Spring 在方法结束时回滚事务，而不是提交。

优点

无需抛出异常，代码简洁。
仍然保留了 @Transactional 的声明式事务管理特性。
与 Spring 的 AOP 机制无缝集成。

注意事项

确保方法是在事务上下文中运行。如果 @Transactional 没有生效（例如方法是 private 或没有通过 Spring 代理调用），currentTransactionStatus() 会抛出异常。
如果需要验证事务是否生效，可以通过日志或调试确认。