Spring Boot 应用优雅停机与JPA实体数据持久化策略

import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

@Service
public class ShutdownPersistenceService {

    private final MangaService mangaService; // 假设这是处理Manga实体的服务

    public ShutdownPersistenceService(MangaService mangaService) {
        this.mangaService = mangaService;
    }

    /**
     * 执行所有待保存的JPA实体持久化操作。
     * 确保此方法是幂等的，并且能够处理并发调用（如果需要）。
     */
    @Transactional // 确保在一个事务中完成所有保存操作
    public void persistAllEntitiesOnShutdown() {
        System.out.println("--- 收到停机前数据持久化请求 ---");
        try {
            // 这里可以添加更复杂的逻辑，例如根据特定条件选择要保存的实体
            mangaService.saveAll(); // 假设MangaService有saveAll方法来批量保存
            System.out.println("--- 所有JPA实体数据已成功持久化 ---");
        } catch (Exception e) {
            System.err.println("!!! 停机前数据持久化失败: " + e.getMessage());
            // 记录错误，可能需要告警
            throw new RuntimeException("数据持久化异常", e);
        }
    }
}

import org.springframework.http.HttpStatus;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

@RestController
@RequestMapping("/admin/lifecycle") // 建议放在一个受保护的admin路径下
public class ApplicationLifecycleController {

    private final ShutdownPersistenceService shutdownPersistenceService;

    public ApplicationLifecycleController(ShutdownPersistenceService shutdownPersistenceService) {
        this.shutdownPersistenceService = shutdownPersistenceService;
    }

    /**
     * 外部调用此端点以触发应用停机前的数据持久化。
     * 调用者应等待此请求完成后再发送应用终止信号。
     */
    @PostMapping("/prepare-shutdown")
    public ResponseEntity<String> prepareForShutdown() {
        try {
            shutdownPersistenceService.persistAllEntitiesOnShutdown();
            return ResponseEntity.ok("数据持久化完成。应用现在可以安全终止。");
        } catch (Exception e) {
            return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR)
                                 .body("数据持久化失败: " + e.getMessage());
        }
    }

    // 可以在此添加一个用于实际触发应用关闭的端点，但通常建议由外部系统在持久化完成后发送终止信号。
    // 例如，使用Spring Boot Actuator的 /actuator/shutdown
    /*
    @PostMapping("/initiate-shutdown")
    public ResponseEntity<String> initiateShutdown() {
        // 这会立即关闭应用，不等待持久化完成，因此通常在调用 /prepare-shutdown 之后由外部触发。
        // 或者，在此方法内部调用 prepareForShutdown()，然后立即关闭。
        // 但更好的做法是分离关注点，让外部协调。
        SpringApplication.exit(SpringApplication.run(Application.class));
        return ResponseEntity.ok("应用正在关闭...");
    }
    */
}