backend_code/src/main/java/com/template/common/utils/QuartzJobUtils.java

package com.template.common.utils;

import com.template.model.pojo.SmartDataTask;
import com.template.model.result.CommonResult;
import org.quartz.*;

import java.util.Map;

public class QuartzJobUtils {

    private static final String JobClass = "com.template.api.Task";

    /**
     * 创建定时任务
     *
     * @param scheduler     调度器
     * @param smartDataTask 定时任务信息类
     */
    public static Map<String, Object> createScheduleJob(Scheduler scheduler, SmartDataTask smartDataTask) {
        Map<String, Object> exists = exists(scheduler, smartDataTask.getTkName());
        if ("0".equals(exists.get("code"))) {
            return exists;
        }
        try {
            Class<? extends Job> jobClass = (Class<? extends Job>) Class.forName(JobClass);
            // 构建定时任务信息
            JobDetail jobDetail = JobBuilder.newJob(jobClass)
                    .withIdentity(smartDataTask.getTkName())
                    .withDescription(smartDataTask.getTkDescrition())
                    .build();
            // 设置定时任务的执行方式
            CronScheduleBuilder cronScheduleBuilder = CronScheduleBuilder.cronSchedule(smartDataTask.getTkCron());
            // 构建触发器
            CronTrigger cronTrigger = TriggerBuilder.newTrigger()
                    .withIdentity(smartDataTask.getTkName())
                    .withSchedule(cronScheduleBuilder)
                    .build();
            //把触发器与任务进行绑定
            scheduler.scheduleJob(jobDetail, cronTrigger);

            return CommonUtil.getReturnMap(String.valueOf(0), "定时任务【" + smartDataTask.getTkName() + "】创建成功！");
        } catch (ClassNotFoundException e) {
            return CommonUtil.getReturnMap(String.valueOf(1), "找不到定时任务类!");
        } catch (SchedulerException e) {
            return CommonUtil.getReturnMap(String.valueOf(1), "创建定时任务出错：" + e.getMessage());
        }
    }

    /**
     * 根据任务暂停定时任务
     *
     * @param scheduler 调度器
     * @param jobName   任务名称
     */
    public static Map<String, Object> pauseScheduleJob(Scheduler scheduler, String jobName) {
        Map<String, Object> exists = exists(scheduler, jobName);
        if ("1".equals(exists.get("code"))) {
            return exists;
        }

        JobKey jobKey = JobKey.jobKey(jobName);
        try {
            scheduler.pauseJob(jobKey);
            return CommonUtil.getReturnMap("0", "定时任务【" + jobName + "】暂停成功！");
        } catch (SchedulerException e) {
            return CommonUtil.getReturnMap("0", "定时任务【" + jobName + "】暂停出错：" + e.getMessage());
        }
    }

    public static Map<String, Object> exists(Scheduler scheduler, String jobName) {
        JobKey jobKey = new JobKey(jobName);
        try {
            if (scheduler.getJobDetail(jobKey) != null) {
                return CommonUtil.getReturnMap("0", "定时任务【" + jobName + "】已在运行！");
            } else {
                return CommonUtil.getReturnMap("1", "定时任务【" + jobName + "】不存在！");
            }
        } catch (SchedulerException e) {
            return CommonUtil.getReturnMap("1", "异常：" + e.getMessage());
        }
    }

    /**
     * 根据任务恢复定时任务
     *
     * @param scheduler 调度器
     * @param jobName   任务名称
     */
    public static Map<String, Object> resumeScheduleJob(Scheduler scheduler, String jobName) {
        Map<String, Object> exists = exists(scheduler, jobName);
        if ("1".equals(exists.get("code"))) {
            return exists;
        }

        JobKey jobKey = JobKey.jobKey(jobName);
        try {
            scheduler.resumeJob(jobKey);
            return CommonUtil.getReturnMap("0", "定时任务【" + jobName + "】已恢复成功！");
        } catch (SchedulerException e) {
            return CommonUtil.getReturnMap("1", "定时任务【" + jobName + "】恢复出错：" + e.getMessage());
        }
    }

    /**
     * 根据任务名称立即运行一次定时任务
     *
     * @param scheduler 调度器
     * @param jobName   任务名称
     */
    public static Map<String, Object> runOnce(Scheduler scheduler, String jobName) {
        Map<String, Object> exists = exists(scheduler, jobName);
        if ("1".equals(exists.get("code"))) {
            return exists;
        }

        JobKey jobKey = JobKey.jobKey(jobName);
        try {
            scheduler.triggerJob(jobKey);
            return CommonUtil.getReturnMap("0","定时任务【" + jobName + "】立即运行1次成功！");
        } catch (SchedulerException e) {
            return CommonUtil.getReturnMap("1","定时任务【" + jobName + "】立即运行1次出错：" + e.getMessage());
        }
    }

    /**
     * 根据任务更新定时任务
     *
     * @param scheduler     调度器
     * @param smartDataTask 定时任务信息类
     */
    public static Map<String, Object> updateScheduleJob(Scheduler scheduler, SmartDataTask smartDataTask) {
        Map<String, Object> exists = exists(scheduler, smartDataTask.getTkName());
        if ("1".equals(exists.get("code"))) {
            return exists;
        }

        try {
            //获取到对应任务的触发器
            TriggerKey triggerKey = TriggerKey.triggerKey(smartDataTask.getTkName());
            //设置定时任务的执行方式
            CronScheduleBuilder cronScheduleBuilder = CronScheduleBuilder.cronSchedule(smartDataTask.getTkCron());
            //重新构建任务的触发器
            CronTrigger trigger = (CronTrigger) scheduler.getTrigger(triggerKey);
            trigger = trigger.getTriggerBuilder()
                    .withIdentity(triggerKey)
                    .withSchedule(cronScheduleBuilder)
                    .build();
            //重置对应的Job
            scheduler.rescheduleJob(triggerKey, trigger);
            return CommonUtil.getReturnMap("0", "定时任务【" + smartDataTask.getTkName() + "】刷新成功！");
        } catch (SchedulerException e) {
            return CommonUtil.getReturnMap("1", "定时任务【" + smartDataTask.getTkName() + "】刷新出错：" + e.getMessage());
        }
    }

    /**
     * 根据定时任务名称从调度器中除定时任务
     *
     * @param scheduler 调度器
     * @param jobName   任务名称
     */
    public static Map<String, Object> deleteScheduleJob(Scheduler scheduler, String jobName) {
        Map<String, Object> exists = exists(scheduler, jobName);
        if ("1".equals(exists.get("code"))) {
            return exists;
        }

        JobKey jobKey = JobKey.jobKey(jobName);
        try {
            scheduler.deleteJob(jobKey);
            return CommonUtil.getReturnMap(String.valueOf(0), "定时任务【" + jobName + "】删除成功！");
        } catch (SchedulerException e) {
            return CommonUtil.getReturnMap(String.valueOf(1), "定时任务【" + jobName + "】删除出错：" + e.getMessage());
        }
    }
}