ss

jiangping

2025-06-19 6f2abed09bb02b22f73477642c21a333fe741207

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
package com.doumee.config.quartz;
 
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import com.doumee.core.constants.Constants;
import com.doumee.core.job.SchedulerProxy;
import com.doumee.dao.system.SystemJobMapper;
import com.doumee.dao.system.SystemJobSnippetMapper;
import com.doumee.dao.system.model.SystemJob;
import com.doumee.dao.system.model.SystemJobSnippet;
import com.doumee.service.system.SystemJobService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
 
import javax.annotation.PostConstruct;
import java.util.List;
 
/**
 * 任务初始化
 * @author  dm
 * @since 2025/03/31 16:44
 */
@Slf4j
@Component
public class JobInitializer {
 
    @Autowired
    private SystemJobService systemJobService;
 
    @Autowired
    private SystemJobMapper systemJobMapper;
 
    @Autowired
    private SystemJobSnippetMapper systemJobSnippetMapper;
 
    @Autowired
    private SchedulerProxy schedulerProxy;
 
    /**
     * 初始化Jobs
     */
//    @PostConstruct
    public void initJobs () {
        log.debug("Initializing jobs");
        // 修复JOB
        this.repairJobs();
        // 初始化JOB
        long startTime = System.currentTimeMillis();
        SystemJob queryJobDto = new SystemJob();
        queryJobDto.setStatus(Constants.Job.JobStatus.READY.getCode());
        queryJobDto.setDisabled(Boolean.FALSE);
        queryJobDto.setDeleted(Boolean.FALSE);
        List<SystemJob> jobs = systemJobService.findList(queryJobDto);
        for (SystemJob job : jobs) {
            schedulerProxy.createJob(new SchedulerProxy.Job(job.getId(), job.getHandler(), job.getCronExpression(), job.getDisallowConcurrent()));
            log.debug("job '{}' initialized.", job.getJobName());
        }
        log.debug("Jobs initialization completed in {} ms", System.currentTimeMillis() - startTime);
    }
 
    /**
     * 修复Jobs和Snippets，防止因服务器宕机导致正在执行中的Job或Snippet状态未得到处理，导致任务或分片丢失
     */
    private void repairJobs () {
        log.debug("Repair jobs");
        // 将运行中的JOB调整为准备状态 & 将预计下一次执行时间调整为null
        systemJobMapper.update(null, new LambdaUpdateWrapper<SystemJob>()
                .set(SystemJob::getStatus, Constants.Job.JobStatus.READY.getCode())
                .set(SystemJob::getNextFireTime, null)
                .eq(SystemJob::getStatus, Constants.Job.JobStatus.RUNNING.getCode())
                .eq(SystemJob::getDeleted, Boolean.FALSE)
        );
        log.debug("Repair job snippets");
        // 将运行中的SNIPPET调整为准备状态
        SystemJobSnippet newSnippet = new SystemJobSnippet();
        newSnippet.setStatus(Constants.Job.SnippetStatus.READY.getCode());
        systemJobSnippetMapper.update(newSnippet, new LambdaUpdateWrapper<SystemJobSnippet>()
                .eq(SystemJobSnippet::getStatus, Constants.Job.SnippetStatus.READY.getCode())
        );
    }
}