package cn.iocoder.yudao.module.mes.job.pro;

import cn.iocoder.yudao.module.mes.service.pro.qrexport.MesProQrExportTaskService;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.boot.context.event.ApplicationReadyEvent;
import org.springframework.context.event.EventListener;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;

/**
 * 二维码导出临时文件清理
 * <p>
 * 大批量导出的临时 ZIP 可达上 GB，正常下载完成后即删，本任务只兜底两类残留：
 * <ol>
 *     <li>下载中断或生成完成后始终未下载的文件，超过保留期后删除；</li>
 *     <li>因服务重启而失去执行线程、永远停在「生成中」的任务记录。</li>
 * </ol>
 * 用 Spring 自带的 {@code @Scheduled} 而非 Quartz JobHandler：后者需要在管理后台手工登记任务，
 * 漏登记就会静默不执行，而这里的清理属于「不执行就会占满磁盘」的兜底逻辑，必须开箱即跑。
 */
@Slf4j
@Component
@RequiredArgsConstructor
public class MesProQrExportCleanupTask {

    private final MesProQrExportTaskService taskService;

    /**
     * 每 5 分钟清理一次：与默认 10 分钟的临时文件保留期匹配，避免残留文件长时间占用磁盘
     */
    @Scheduled(cron = "0 */5 * * * ?")
    public void cleanup() {
        try {
            int files = taskService.cleanupExpiredFiles();
            int tasks = taskService.recycleStaleTasks();
            if (files > 0 || tasks > 0) {
                log.info("[cleanup][二维码导出清理完成：删除临时文件 {} 个，回收残留任务 {} 个]", files, tasks);
            }
        } catch (Exception e) {
            log.error("[cleanup][二维码导出清理失败]", e);
        }
    }

    /**
     * 启动时立刻回收一次残留任务
     * <p>
     * 进程重启后运行态注册表为空，此时表中的非终态记录必定没有执行线程在推进，
     * 立刻回收可避免它们以「生成中」的状态一直挂在界面上（否则要等到下一次定时清理）。
     */
    @EventListener(ApplicationReadyEvent.class)
    public void recycleOnStartup() {
        try {
            int tasks = taskService.recycleStaleTasks();
            if (tasks > 0) {
                log.info("[recycleOnStartup][启动回收残留二维码导出任务 {} 个]", tasks);
            }
        } catch (Exception e) {
            log.error("[recycleOnStartup][启动回收二维码导出任务失败]", e);
        }
    }

}