package com.gunshi.project.xyt.schedule;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
import com.gunshi.project.xyt.model.ForecastPlan;
import com.gunshi.project.xyt.service.ForecastPlanService;
import jakarta.annotation.Resource;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.CommandLineRunner;
import org.springframework.context.annotation.Profile;
import org.springframework.stereotype.Component;

import java.util.List;
import java.util.Map;
import java.util.function.Function;
import java.util.stream.Collectors;

/**
 * @author cxw
 * @description 任务预热，预先加载数据库已经配置好的任务
 * @classname TaskGroupJobRunner.java
 * @create 2024-07-30, 星期二, 9:42:39
 */
@Component
@Slf4j
@Profile("prod")
public class TaskGroupJobRunner implements CommandLineRunner {

    @Resource
    private TaskGroupHandler taskGroupHandler;

    @Autowired
    private ForecastPlanService forecastPlanService;

    @Override
    public void run(String... args) {
        // 初始加载数据库里状态为自动、正常的定时任务
        List<ForecastPlan> planList = forecastPlanService.list(new QueryWrapper<ForecastPlan>().eq("type", "1").eq("status", "0"));
        if (CollectionUtils.isNotEmpty(planList)) {
            Map<Integer, ForecastPlan> jobMap = planList.stream()
                    .collect(Collectors.toMap(ForecastPlan::getId, Function.identity(), (key1, key2)->key2));
            for (Map.Entry<Integer, ForecastPlan> entry : jobMap.entrySet()) {
                taskGroupHandler.addCronJob(String.valueOf(entry.getKey()), entry.getValue());
            }
        }
    }
}