2021与蓝度共同重构项目,服务端
chenjiantian
2021-12-16 0f98ec728d0403b1726273d94dc00f06993888b6
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
package com.sandu.common.quartz.service;
 
import com.sandu.common.execption.BusinessException;
import lombok.extern.slf4j.Slf4j;
import org.quartz.*;
import org.quartz.impl.triggers.CronTriggerImpl;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.scheduling.quartz.QuartzJobBean;
import org.springframework.stereotype.Service;
 
import javax.annotation.Resource;
import java.util.Date;
import java.util.Map;
 
/**
 * @author chenjiantian
 * @date 2021/12/15 10:39
 * quartz job管理类
 */
@Slf4j
@Service
@ConditionalOnProperty(prefix = "sandu.quartz", name = "enable", havingValue = "true")
public class QuartzServiceImpl implements QuartzService {
 
    @Resource(name = "scheduler")
    private Scheduler scheduler;
 
    @Override
    public void addJob(Class<? extends QuartzJobBean> jobClass, Long jobId, String cron, Map<String, Object> jobData) {
        try {
            String simpleName = jobClass.getSimpleName();
            JobDetail jobDetail = JobBuilder.newJob(jobClass)
                    .withIdentity(simpleName + jobId)
                    .build();
            CronTrigger cronTrigger = TriggerBuilder.newTrigger()
                    .withIdentity(simpleName + jobId)
                    .startNow()
                    .withSchedule(CronScheduleBuilder.cronSchedule(cron))
                    .build();
            cronTrigger.getJobDataMap().putAll(jobData);
            ((CronTriggerImpl) cronTrigger).setStartTime(new Date());
            scheduler.scheduleJob(jobDetail, cronTrigger);
 
            log.info("添加定时器{}", simpleName + jobId);
        } catch (SchedulerException e) {
            log.error("创建定时任务失败", e);
            throw new BusinessException("创建定时任务失败");
        }
    }
 
    @Override
    public void updateJobCron(Class<? extends QuartzJobBean> jobClass, Long jobId, String cron, Map<String, Object> jobData) {
        try {
            TriggerKey triggerKey = TriggerKey.triggerKey(jobClass.getSimpleName() + jobId);
            CronTrigger trigger = (CronTrigger) scheduler.getTrigger(triggerKey);
            if (trigger == null) {
                addJob(jobClass, jobId, cron, jobData);
                trigger = (CronTrigger) scheduler.getTrigger(triggerKey);
            }
            CronScheduleBuilder scheduleBuilder = CronScheduleBuilder.cronSchedule(cron);
            trigger = trigger.getTriggerBuilder().withIdentity(triggerKey).withSchedule(scheduleBuilder).build();
            trigger.getJobDataMap().putAll(jobData);
 
            scheduler.rescheduleJob(triggerKey, trigger);
        } catch (SchedulerException e) {
            log.error("更新定时任务失败", e);
            throw new BusinessException("更新定时任务失败");
        }
    }
 
    @Override
    public void deleteJob(Class<? extends QuartzJobBean> jobClass, Long jobId) {
        try {
            JobKey jobKey = JobKey.jobKey(jobClass.getSimpleName() + jobId);
            scheduler.pauseJob(jobKey);
            scheduler.deleteJob(jobKey);
            log.info("删除定时器{}", jobClass.getSimpleName() + jobId);
        } catch (SchedulerException e) {
            log.error("删除定时任务失败", e);
            throw new BusinessException("删除定时任务失败");
        }
    }
 
    @Override
    public void pauseJob(Class<? extends QuartzJobBean> jobClass, Long jobId) {
        try {
            JobKey jobKey = JobKey.jobKey(jobClass.getSimpleName() + jobId);
            scheduler.pauseJob(jobKey);
        } catch (Exception e) {
            log.error("定时任务暂停失败", e);
            throw new BusinessException("定时任务暂停失败");
        }
    }
 
    @Override
    public void resumeJob(Class<? extends QuartzJobBean> jobClass, Long jobId, String cron, Map<String, Object> jobData) {
        try {
            TriggerKey triggerKey = TriggerKey.triggerKey(jobClass.getSimpleName() + jobId);
            CronTrigger trigger = (CronTrigger) scheduler.getTrigger(triggerKey);
            // 如果不存在则创建一个定时任务
            if (trigger == null) {
                addJob(jobClass, jobId, cron, jobData);
            }
            JobKey jobKey = JobKey.jobKey(jobClass.getSimpleName() + jobId);
            scheduler.resumeJob(jobKey);
        } catch (Exception e) {
            log.error("恢复定时任务失败", e);
            throw new BusinessException("恢复定时任务失败");
        }
    }
 
    @Override
    public void runJobNow(Class<? extends QuartzJobBean> jobClass, Long jobId, String cron, Map<String, Object> jobData) {
        try {
            TriggerKey triggerKey = TriggerKey.triggerKey(jobClass.getSimpleName() + jobId);
            CronTrigger trigger = (CronTrigger) scheduler.getTrigger(triggerKey);
            // 如果不存在则创建一个定时任务
            if (trigger == null) {
                addJob(jobClass, jobId, cron, jobData);
            }
            JobDataMap dataMap = new JobDataMap();
            dataMap.putAll(jobData);
            JobKey jobKey = JobKey.jobKey(jobClass.getSimpleName() + jobId);
            scheduler.triggerJob(jobKey, dataMap);
        } catch (Exception e) {
            log.error("定时任务执行失败", e);
            throw new BusinessException("定时任务执行失败");
        }
    }
}