<dependencies>
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-web</artifactId>
    </dependency>
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-aop</artifactId>
    </dependency>
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-jdbc</artifactId>
    </dependency>
    <dependency>
        <groupId>com.h2database</groupId>
        <artifactId>h2</artifactId>
        <scope>runtime</scope>
    </dependency>
</dependencies>

// User.java
public class User {
    private Long id;
    private String name;
}

// Order.java
public class Order {
    private Long id;
    private String productName;
    private Long userId;
}

@Service
public class UserService {
    @Autowired
    private JdbcTemplate jdbcTemplate;
    // 自定義注解，表示此方法需要事務(wù)
    @TransactionalAspect
    public void createUserWithOrder(String userName, String productName) {
        // 1. 保存用戶
        String userSql = "INSERT INTO users(name) VALUES(?)";
        jdbcTemplate.update(userSql, userName);
        // 模擬異常：如果 productName 是 "fail"，則拋出異常
        if ("fail".equals(productName)) {
            throw new RuntimeException("訂單創(chuàng)建失??！");
        }
        // 2. 保存訂單
        String orderSql = "INSERT INTO orders(product_name, user_id) VALUES(?, ?)";
        // 假設(shè)用戶ID是自增，這里簡(jiǎn)化為查出來(lái)
        Long userId = jdbcTemplate.queryForObject("SELECT MAX(id) FROM users", Long.class);
        jdbcTemplate.update(orderSql, productName, userId);
    }
}

// TransactionalAspect.java
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface TransactionalAspect {
}

@Aspect
@Component
public class TransactionAspect {
    @Autowired
    private PlatformTransactionManager transactionManager;
    @Around("@annotation(transactionalAspect)")
    public Object manageTransaction(ProceedingJoinPoint joinPoint, TransactionalAspect transactionalAspect) throws Throwable {
        // 1. 定義事務(wù)屬性
        DefaultTransactionDefinition def = new DefaultTransactionDefinition();
        def.setPropagationBehavior(TransactionDefinition.PROPAGATION_REQUIRED); // 默認(rèn)傳播行為
        // 2. 開啟事務(wù)
        TransactionStatus status = transactionManager.getTransaction(def);
        try {
            // 3. 執(zhí)行目標(biāo)方法
            Object result = joinPoint.proceed();
            // 4. 方法成功執(zhí)行，提交事務(wù)
            transactionManager.commit(status);
            System.out.println("事務(wù)提交成功");
            return result;
        } catch (Exception e) {
            // 5. 發(fā)生異常，回滾事務(wù)
            if (status != null && !status.isCompleted()) {
                transactionManager.rollback(status);
                System.out.println("事務(wù)已回滾，原因: " + e.getMessage());
            }
            throw e; // 重新拋出異常
        }
    }
}

@RestController
public class TestController {
    @Autowired
    private UserService userService;
    // 測(cè)試：成功場(chǎng)景
    @GetMapping("/test/success")
    public String testSuccess() {
        userService.createUserWithOrder("張三", "iPhone");
        return "用戶和訂單創(chuàng)建成功！";
    }
    // 測(cè)試：失敗場(chǎng)景（應(yīng)回滾）
    @GetMapping("/test/fail")
    public String testFail() {
        try {
            userService.createUserWithOrder("李四", "fail");
        } catch (Exception e) {
            return "創(chuàng)建失敗，但事務(wù)已回滾！";
        }
        return "success";
    }
}