springboot整合quartz

一、使用步骤

1.引入依赖

代码如下(示例):

<!-- https://mvnrepository.com/artifact/org.springframework.boot/spring-boot-starter-quartz -->
<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-quartz</artifactId>
    <version>2.3.12.RELEASE</version>
</dependency>

2.定义一个jobInfo类

用来封装定时任务参数

import lombok.Data;
import org.quartz.JobDataMap;
import org.springframework.util.ObjectUtils;

/**
 * @Description 定时任务参数类
 */
@Data
public class JobInfo {

    /*
    任务名
    */
    private String jobName;

    /*
    cron 表达式
    */
    private String cron;

    /*
    任务组
    */
    private String jobGroup;

    /*
    定时器名(创建任务时自动组织, 格式为 Trigger_用户名_任务名)
    */
    private String triggerName;

    /*
    定时器组
    */
    private String triggerGroup;

    /*
    需要定时执行的 job(如果需要自定义自己的特殊方法, 可以自己添加一个 *Job 实现 Job 接口即可)
    */
    private String jobClassName;

    /*
    参数(可为空, 有的定时任务逻辑中可能需要调用其他方法或者传递参数, 可以设置在此处)
    */
    private JobDataMap jobDataMap;
}

3.QuartzService

用来封装定时任务的新增、编辑、删除、立即触发一次等方法

import com.google.common.base.Preconditions;
import lombok.extern.slf4j.Slf4j;
import newdt.quartz.bean.JobInfo;
import newdt.quartz.util.QuartzUtil;
import org.quartz.*;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.util.ObjectUtils;
import org.springframework.util.StringUtils;

import java.util.UUID;

/**
 * @Description 定时任务 service 包括定时任务的增删改方法等
 */
@Service
@Slf4j
public class QuartzService {

    @Autowired
    private Scheduler scheduler;

    private static final String DEFAULT_JOB_GROUP = "default_job_group";

    private static final String DEFAULT_TRIGGER_GROUP = "default_trigger_group";

    private static final String TRIGGER_PRE = "Trigger_";

    /**
     * 添加定时任务
     *
     * @param jobInfo  定时任务信息
     * @param userName 当前登录用户名
     * @return 是否添加成功 success 代表成功, 其他都是失败
     */
    public String createJob(JobInfo jobInfo, String userName) {
        /*
         逻辑:
         1. 参数校验
         2. 参数处理, 如果未指定任务组名和触发器组名就是用默认的
         3. 当前任务不存在才进行添加
         4. 构建 Job
         5. cron表达式定时构造器
         6. 构建 Trigger
         7. 启动调度器
         */

        // [1] 参数校验
        Preconditions.checkArgument(jobInfo.getJobName() != null, "定时任务名不能为空");
        Preconditions.checkArgument(jobInfo.getCron() != null, "cron 表达式不能为空");
        Preconditions.checkArgument(jobInfo.getJobClassName() != null, "需要定时执行的 job 不能为空");
        QuartzUtil.checkCronExpression(jobInfo.getCron());

        // [2] 参数处理, 如果未指定任务组名和触发器组名就是用默认的
        String jobGroup = ObjectUtils.isEmpty(jobInfo.getJobGroup()) ? DEFAULT_JOB_GROUP : jobInfo.getJobGroup();
        String triggerGroup = ObjectUtils.isEmpty(jobInfo.getTriggerGroup()) ? DEFAULT_TRIGGER_GROUP : jobInfo.getTriggerGroup();

        try {
            // [3] 当前任务不存在才进行添加
            JobKey jobKey = JobKey.jobKey(jobInfo.getJobName(), jobGroup);
            if (scheduler.checkExists(jobKey)) {
                log.warn("[添加定时任务]已存在该作业, jobkey为: {}", jobKey);
                throw new RuntimeException("[添加定时任务]已存在该作业, jobkey为: " + jobKey);
            }

            // [4] 构建 Job
            JobDetail job = JobBuilder.newJob(getClass(jobInfo.getJobClassName()).getClass()).withIdentity(jobKey)
                    .setJobData(ObjectUtils.isEmpty(jobInfo.getJobDataMap()) ? new JobDataMap() : jobInfo.getJobDataMap())
                    .build();

            // [5] cron表达式定时构造器
            CronScheduleBuilder cronScheduleBuilder = CronScheduleBuilder.cronSchedule(jobInfo.getCron());

            // [6] 构建 Trigger
            String triggerName = TRIGGER_PRE + userName + "_" + jobInfo.getJobName();
            Trigger trigger = TriggerBuilder.newTrigger()
                    .withIdentity(TriggerKey.triggerKey(triggerName, triggerGroup))
//                .startAt(DateUtil.parseDate(start))
//                .endAt(DateUtil.parseDate(end))
                    .withSchedule(cronScheduleBuilder).build();

            // [7] 启动调度器
            scheduler.scheduleJob(job, trigger);
            scheduler.start();
            log.info("[添加定时任务]jobName: {}, jobGroup: {}, triggerName: {}, triggerGroup: {}, jobClassName: {}, 添加成功--------------",
                    jobInfo.getJobName(), jobInfo.getJobGroup(), triggerName, jobInfo.getTriggerGroup(), jobInfo.getJobClassName());

            return "success";
        } catch (Exception e) {
            log.error("[添加定时任务]失败, 报错: ", e);
            throw new RuntimeException("[添加定时任务]失败, 报错: " + e.getMessage());
        }

    }

    /**
     * 修改定时任务
     *
     * @param jobInfo  定时任务信息
     * @param userName 当前登录用户名
     * @return 是否更新成功 success 代表成功, 其他都是失败
     */
    public String editJob(JobInfo jobInfo, String userName) {
        /*
        逻辑:
        1. 参数校验
        2. 判断任务是否存在
        3. 更新任务
        */

        // [1] 参数校验
        Preconditions.checkArgument(jobInfo.getJobName() != null, "定时任务名不能为空");
        Preconditions.checkArgument(jobInfo.getJobGroup() != null, "定时任务组名不能为空");
        Preconditions.checkArgument(jobInfo.getCron() != null, "cron 表达式不能为空");
        Preconditions.checkArgument(jobInfo.getTriggerGroup() != null, "触发器组名不能为空");
        Preconditions.checkArgument(jobInfo.getJobClassName() != null, "需要定时执行的 job 不能为空");
        QuartzUtil.checkCronExpression(jobInfo.getCron());

        try {
            // [2] 判断任务是否存在
            JobKey jobKey = JobKey.jobKey(jobInfo.getJobName(), jobInfo.getJobGroup());
            if (!scheduler.checkExists(jobKey)) {
                log.error("[更新定时任务]该作业不存在, jobkey为: {}", jobKey);
                throw new RuntimeException("[更新定时任务]该作业不存在, jobkey为: " + jobKey);
            }

            // [3] 更新任务
            CronScheduleBuilder cronScheduleBuilder = CronScheduleBuilder.cronSchedule(jobInfo.getCron());
            String newTriggerName = TRIGGER_PRE + userName + "_" + jobInfo.getJobName();
            Trigger newTrigger = TriggerBuilder.newTrigger()
                    .withIdentity(TriggerKey.triggerKey(newTriggerName, jobInfo.getTriggerGroup()))
                    .withSchedule(cronScheduleBuilder).build();

            scheduler.rescheduleJob(TriggerKey.triggerKey(newTriggerName, jobInfo.getTriggerGroup()), newTrigger);

            log.info("[更新定时任务]jobName: {}, jobGroup: {}, triggerName: {}, triggerGroup: {}, jobClassName: {}, 更新成功--------------",
                    jobInfo.getJobName(), jobInfo.getJobGroup(), newTriggerName, jobInfo.getTriggerGroup(), jobInfo.getJobClassName());

            return "success";
        } catch (Exception e) {
            log.error("[更新定时任务]失败, 报错: ", e);
            throw new RuntimeException("[更新定时任务]失败, 报错: " + e.getMessage());
        }
    }

    /**
     * 删除定时定时任务
     *
     * @param jobInfo 定时任务信息
     * @return 是否删除成功 success 代表成功, 其他都是失败
     */
    public String deleteJob(JobInfo jobInfo) {
        /*
         逻辑:
         1. 参数校验
         2. 参数处理, 根据触发器名和触发器组名查询校验是否存在该触发器
         3. 暂停触发器
         4. 移除触发器
         5. 删除任务
         */

        // [1] 参数校验
        Preconditions.checkArgument(jobInfo.getJobName() != null, "定时任务名不能为空");
        Preconditions.checkArgument(jobInfo.getJobGroup() != null, "定时任务组名不能为空");
        Preconditions.checkArgument(jobInfo.getTriggerName() != null, "触发器名不能为空");
        Preconditions.checkArgument(jobInfo.getTriggerGroup() != null, "触发器组名不能为空");

        try {
            // [2] 参数处理, 根据触发器名和触发器组名查询校验是否存在该触发器
            TriggerKey triggerKey = TriggerKey.triggerKey(jobInfo.getTriggerName(), jobInfo.getTriggerGroup());
            Trigger trigger = scheduler.getTrigger(triggerKey);
            if (null == trigger) {
                log.info("[删除定时任务]根据 triggerName: {}和 triggerGroup: {}未查询到相应的 trigger", jobInfo.getTriggerName(), jobInfo.getTriggerGroup());
                throw new RuntimeException("[删除定时任务]根据 triggerName: " + jobInfo.getTriggerName() + "和 triggerGroup: " + jobInfo.getTriggerGroup() + "未查询到相应的 trigger");
            }

            // [3] 暂停触发器
            scheduler.pauseTrigger(triggerKey);

            // [4] 移除触发器
            scheduler.unscheduleJob(triggerKey);

            // [5] 删除任务
            JobKey jobKey = JobKey.jobKey(jobInfo.getJobName(), jobInfo.getJobGroup());
            scheduler.deleteJob(jobKey);

            log.info("[停止定时任务]jobName: {}, jobGroup: {}, triggerName: {}, triggerGroup: {}, 停止--------------",
                    jobInfo.getJobName(), jobInfo.getJobGroup(), jobInfo.getTriggerName(), jobInfo.getTriggerGroup());

            return "success";
        } catch (SchedulerException e) {
            log.error("[停止定时任务]失败, 报错: ", e);
            throw new RuntimeException("[停止定时任务]失败, 报错: " + e.getMessage());
        }
    }

    /**
     * 立即执行一次指定的定时任务
     *
     * @param jobInfo  定时任务信息
     * @return 是否执行成功 success 代表成功, 其他都是失败
     */
    public String executeImmediately(JobInfo jobInfo) {
        /*
         逻辑:
         1. 参数校验
         2. 如果本次执行的 job 需要使用 jobData, 则必须传 jobName 和 jobGroup, 组装 jobKey 查询该 job 原始的 jobData
         3. 组装临时的 job 和 trigger
         4. 启动调度器
         */

        // [1] 参数校验
        Preconditions.checkArgument(jobInfo.getJobClassName() != null, "需要定时执行的 job 不能为空");

        try {
            // [2] 如果本次执行的 job 需要使用 jobData, 则必须传 jobName 和 jobGroup, 组装 jobKey 查询该 job 原始的 jobData
            JobDataMap jobDataMap = new JobDataMap();
            if(StringUtils.hasText(jobInfo.getJobName()) && StringUtils.hasText(jobInfo.getJobGroup())){
                JobKey jobKey = JobKey.jobKey(jobInfo.getJobName(), jobInfo.getJobGroup());
                jobDataMap = scheduler.getJobDetail(jobKey).getJobDataMap();
            }

            // [3] 组装临时的 job 和 trigger
            JobDetail job = JobBuilder.newJob(getClass(jobInfo.getJobClassName()).getClass())
                    .setJobData(jobDataMap)
                    .withIdentity(JobKey.jobKey(UUID.randomUUID().toString(), DEFAULT_JOB_GROUP)).build();

            Trigger trigger = TriggerBuilder.newTrigger()
                    .withIdentity(TriggerKey.triggerKey(UUID.randomUUID().toString(), DEFAULT_TRIGGER_GROUP))
                    .build();

            // [4] 启动调度器
            scheduler.scheduleJob(job, trigger);
            scheduler.start();

            log.info("[立即执行一次任务, 不定时]成功, jobName: {}, jobGroup: {}, triggerName: {}, triggerGroup: {},  jobClassName: {}, 执行成功--------------",
                    jobInfo.getJobName(), jobInfo.getJobGroup(), jobInfo.getTriggerName(), jobInfo.getTriggerGroup(), jobInfo.getJobClassName());

            return "success";
        } catch (Exception e) {
            log.error("[立即执行一次任务, 不定时]失败, 报错: ", e);
            throw new RuntimeException("[立即执行一次任务, 不定时]失败, 报错: " + e.getMessage());
        }
    }

    public static Job getClass(String className) throws Exception {
        Class<?> classTemp = Class.forName(className);
        return (Job) classTemp.newInstance();
    }

    /**
     * 暂停定时任务
     *
     * @param jobInfo 定时任务信息
     * @return 是否暂停成功 success 代表成功, 其他都是失败
     */
    public String pauseJob(JobInfo jobInfo) {
        /*
         逻辑:
         1. 参数校验
         2. 根据定时任务名和定时任务组名暂停该任务
         */

        // [1] 参数校验
        Preconditions.checkArgument(jobInfo.getJobName() != null, "定时任务名不能为空");
        Preconditions.checkArgument(jobInfo.getJobGroup() != null, "定时任务组名不能为空");

        try {
            // [2] 根据定时任务名和定时任务组名暂停该任务
            JobKey jobKey = JobKey.jobKey(jobInfo.getJobName(), jobInfo.getJobGroup());
            scheduler.pauseJob(jobKey);
            log.info("[暂停定时任务]成功, jobName: {}, jobGroup: {}", jobInfo.getJobName(), jobInfo.getJobGroup());

            return "success";
        } catch (SchedulerException e) {
            log.error("[暂停定时任务]失败, 报错: ", e);
            throw new RuntimeException("[暂停定时任务]失败, 报错: " + e.getMessage());
        }
    }

    /**
     * 恢复定时任务
     *
     * @param jobInfo 定时任务信息
     * @return 是否恢复成功 success 代表成功, 其他都是失败
     */
    public String resumeJob(JobInfo jobInfo) {
         /*
         逻辑:
         1. 参数校验
         2. 根据定时任务名和定时任务组名恢复该任务
         */

        // [1] 参数校验
        Preconditions.checkArgument(jobInfo.getJobName() != null, "定时任务名不能为空");
        Preconditions.checkArgument(jobInfo.getJobGroup() != null, "定时任务组名不能为空");

        try {
            // [2] 根据定时任务名和定时任务组名恢复该任务
            JobKey jobKey = JobKey.jobKey(jobInfo.getJobName(), jobInfo.getJobGroup());
            scheduler.resumeJob(jobKey);
            log.info("[恢复定时任务]成功, jobName: {}, jobGroup: {}", jobInfo.getJobName(), jobInfo.getJobGroup());

            return "success";
        } catch (SchedulerException e) {
            log.error("[恢复定时任务]失败, 报错: ", e);
            throw new RuntimeException("[恢复定时任务]失败, 报错: " + e.getMessage());
        }
    }
}

4.Job接口实现类

Job接口实现类里面写定时任务要执行的业务代码



import lombok.extern.slf4j.Slf4j;
import newdt.dsc.service.BaseService;
import newdt.review.bean.ReviewType;
import newdt.review.bean.review.DatabaseReviewTask;
import newdt.review.service.review.DatabaseReviewServiceImpl;
import org.quartz.Job;
import org.quartz.JobDataMap;
import org.quartz.JobExecutionContext;
import org.quartz.JobExecutionException;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;


/**
 * 定时任务 Job 实现类
 */
@Component
@Slf4j
public class DataBaseReviewTaskJob extends BaseService implements Job {
    @Autowired
    DatabaseReviewServiceImpl databaseReviewService;


    @Override
    public void execute(JobExecutionContext context) throws JobExecutionException {
        // [1] 获取 jobId
        JobDataMap jobDataMap = context.getMergedJobDataMap();

        // [2] 封装一个 DatabaseReviewTask 对象
        long jobId = jobDataMap.getLong("jobId");
        DatabaseReviewTask databaseReviewTask = new DatabaseReviewTask();
        databaseReviewTask.setJobId(jobId);
        databaseReviewTask.setTriggerType(1);

        // [3] 创建任务
        databaseReviewService.submitReviewTask(databaseReviewTask, ReviewType.DATABASE);
    }
}

5.封装一个JobInfo对象,调用quartzService的方法即可

    public JobInfo getJobInfo(ReviewJob reviewJob) {
        // [1] 获取审核类型
        log.info("[开始] 构建 jobInfo 对象);
        ReviewType reviewType = reviewJob.getReviewType();

        // [2] 根据审核类型组建不同的 JobInfo
        JobInfo jobInfo = new JobInfo();
        jobInfo.setJobName(LOG_JOB_NAME_PRE + reviewJob.getJobId());
        jobInfo.setJobGroup(LOG_JOB_GROUP);
        jobInfo.setTriggerGroup(LOG_TRIGGER_GROUP);
        jobInfo.setJobClassName(LOG_JOB_CLASS_NAME);
        jobInfo.setTriggerName(TRIGGER_PRE + reviewJob.getCreator() + "_" + jobInfo.getJobName());
        jobInfo.setCron(reviewJob.getScheduleCron());
        jobInfo.setTriggerName(TRIGGER_PRE + reviewJob.getCreator() + "_" + jobInfo.getJobName());

        // [3] 返回 jobInfo
        log.info("[结束] 构建 jobInfo 对象成功, 对象信息 [{}]", jobInfo);
        return jobInfo;
    }
  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

李小枫

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值