package com.example.demo.scheduler;

import java.util.Map;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;

import com.example.demo.service.TaskCleanupService;
import com.example.demo.service.TaskQueueService;

/**
 * 任务队列定时调度器
 * 每2分钟检查一次任务状态
 */
@Component
public class TaskQueueScheduler {

    private static final Logger logger = LoggerFactory.getLogger(TaskQueueScheduler.class);

    @Autowired
    private TaskQueueService taskQueueService;

    @Autowired
    private TaskCleanupService taskCleanupService;

    /**
     * 处理待处理任务
     * 每2分钟执行一次，处理队列中的待处理任务
     */
    @Scheduled(fixedRate = 120000) // 2分钟 = 120000毫秒
    public void processPendingTasks() {
        try {
            logger.debug("开始处理待处理任务");
            taskQueueService.processPendingTasks();
        } catch (Exception e) {
            logger.error("处理待处理任务失败", e);
        }
    }

    /**
     * 检查任务状态 - 每2分钟执行一次轮询查询
     * 固定间隔：120000毫秒 = 2分钟
     * 查询正在处理的任务状态，更新完成/失败状态
     */
    @Scheduled(fixedRate = 120000) // 2分钟 = 120000毫秒
    public void checkTaskStatuses() {
        try {
            logger.info("=== 开始执行任务队列状态轮询查询 (每2分钟) ===");
            taskQueueService.checkTaskStatuses();
            logger.info("=== 任务队列状态轮询查询完成 ===");
        } catch (Exception e) {
            logger.error("检查任务状态失败", e);
        }
    }

    /**
     * 清理过期任务
     * 每天凌晨2点执行一次
     */
    @Scheduled(cron = "0 0 2 * * ?")
    public void cleanupExpiredTasks() {
        try {
            logger.info("开始清理过期任务");
            int cleanedCount = taskQueueService.cleanupExpiredTasks();
            logger.info("清理过期任务完成，清理数量: {}", cleanedCount);
        } catch (Exception e) {
            logger.error("清理过期任务失败", e);
        }
    }

    /**
     * 定期清理任务
     * 每天凌晨4点执行一次，清理已完成和失败的任务
     */
    @Scheduled(cron = "0 0 4 * * ?")
    public void performTaskCleanup() {
        try {
            logger.info("开始执行定期任务清理");
            Map<String, Object> result = taskCleanupService.performFullCleanup();
            logger.info("定期任务清理完成: {}", result);
        } catch (Exception e) {
            logger.error("定期任务清理失败", e);
        }
    }
}