@Component
class SimpleIdempotencyInterceptor implements HandlerInterceptor {

    @Autowired
    private final IdempotencyDataRepositoryImpl idempotencyRepository;

    @Override
    public boolean preHandle(
        HttpServletRequest request,
        HttpServletResponse response,
        Object handler
    ) throws Exception {
        if (request.getMethod().equals(HttpMethod.GET.name()) || !isPathEnabled(request)) {
            return true;
        }

        String idempotencyKey = request.getHeader("myapp-idempotency-key");
        if (idempotencyKey != null) {
            return handleIdempotency(idempotencyKey, response);
        }

        return true;
    }

    @Override
    public void afterCompletion(
        HttpServletRequest request,
        HttpServletResponse response,
        Object handler,
        Exception ex
    ) throws Exception {
        String idempotencyKey = request.getHeader("myapp-idempotency-key");
        if (idempotencyKey != null && response.getStatus() >= 200 && response.getStatus() <= 299) {
            cacheResponse(idempotencyKey, response);
        }
    }

    private boolean handleIdempotency(String idempotencyKey, HttpServletResponse response) {
        IdempotencyDataEntity existing = idempotencyRepository.findByIdempotencyKey(idempotencyKey);

        switch (existing == null ? "null" : "exists") {
            case "null":
                idempotencyRepository.save(
                    new IdempotencyDataEntity(
                        idempotencyKey,
                        0,
                        ""
                    )
                );
                return true;
            default:
                sendCachedResponse(response, existing);
                return false;
        }
    }

    private void cacheResponse(String idempotencyKey, HttpServletResponse response) {
        try {
            IdempotencyDataEntity existing = idempotencyRepository.findByIdempotencyKey(idempotencyKey);
            if (existing != null) {
                String responseBody = "/* extracted response body */";
                idempotencyRepository.save(
                    new IdempotencyDataEntity(
                        idempotencyKey,
                        response.getStatus(),
                        responseBody
                    )
                );
            }
        } catch (Exception e) {
            log.warn("Failed to cache idempotency response for key: " + idempotencyKey, e);
        }
    }
}

/**
     * Dọn dẹp các bản ghi idempotency cũ hơn 7 ngày
     * Chạy hàng ngày lúc 2 giờ sáng
     */
    @Scheduled(cron = "0 0 2 * * *")
    public void cleanupOldIdempotencyRecords() {
        try {
            LocalDateTime cutoffDate = LocalDateTime.now().minusDays(7);
            int deletedCount = idempotencyRepository.deleteByCreatedDateBefore(cutoffDate);
            logger.info("Đã dọn dẹp {} bản ghi idempotency cũ (cũ hơn {})", deletedCount, cutoffDate);
        } catch (Exception e) {
            logger.error("Dọn dẹp bản ghi idempotency cũ không thành công", e);
        }
    }