package cn.lihu.jh.module.infra.service.job;
|
|
import cn.hutool.extra.spring.SpringUtil;
|
import cn.lihu.jh.framework.common.pojo.PageResult;
|
import cn.lihu.jh.framework.common.util.object.BeanUtils;
|
import cn.lihu.jh.framework.quartz.core.handler.JobHandler;
|
import cn.lihu.jh.framework.quartz.core.scheduler.SchedulerManager;
|
import cn.lihu.jh.framework.quartz.core.util.CronUtils;
|
import cn.lihu.jh.module.infra.controller.admin.job.vo.job.JobPageReqVO;
|
import cn.lihu.jh.module.infra.controller.admin.job.vo.job.JobSaveReqVO;
|
import cn.lihu.jh.module.infra.dal.dataobject.job.JobDO;
|
import cn.lihu.jh.module.infra.dal.mysql.job.JobMapper;
|
import cn.lihu.jh.module.infra.enums.job.JobStatusEnum;
|
import lombok.extern.slf4j.Slf4j;
|
import org.quartz.SchedulerException;
|
import org.springframework.stereotype.Service;
|
import org.springframework.transaction.annotation.Transactional;
|
import org.springframework.validation.annotation.Validated;
|
|
import javax.annotation.Resource;
|
import java.util.List;
|
import java.util.Objects;
|
|
import static cn.lihu.jh.framework.common.exception.util.ServiceExceptionUtil.exception;
|
import static cn.lihu.jh.framework.common.util.collection.CollectionUtils.containsAny;
|
import static cn.lihu.jh.module.infra.enums.ErrorCodeConstants.*;
|
|
/**
|
* 定时任务 Service 实现类
|
*
|
* @author 芋道源码
|
*/
|
@Service
|
@Validated
|
@Slf4j
|
public class JobServiceImpl implements JobService {
|
|
@Resource
|
private JobMapper jobMapper;
|
|
@Resource
|
private SchedulerManager schedulerManager;
|
|
@Override
|
@Transactional(rollbackFor = Exception.class)
|
public Long createJob(JobSaveReqVO createReqVO) throws SchedulerException {
|
validateCronExpression(createReqVO.getCronExpression());
|
// 1.1 校验唯一性
|
if (jobMapper.selectByHandlerName(createReqVO.getHandlerName()) != null) {
|
throw exception(JOB_HANDLER_EXISTS);
|
}
|
// 1.2 校验 JobHandler 是否存在
|
validateJobHandlerExists(createReqVO.getHandlerName());
|
|
// 2. 插入 JobDO
|
JobDO job = BeanUtils.toBean(createReqVO, JobDO.class);
|
job.setStatus(JobStatusEnum.INIT.getStatus());
|
fillJobMonitorTimeoutEmpty(job);
|
jobMapper.insert(job);
|
|
// 3.1 添加 Job 到 Quartz 中
|
schedulerManager.addJob(job.getId(), job.getHandlerName(), job.getHandlerParam(), job.getCronExpression(),
|
createReqVO.getRetryCount(), createReqVO.getRetryInterval());
|
// 3.2 更新 JobDO
|
JobDO updateObj = JobDO.builder().id(job.getId()).status(JobStatusEnum.NORMAL.getStatus()).build();
|
jobMapper.updateById(updateObj);
|
return job.getId();
|
}
|
|
@Override
|
@Transactional(rollbackFor = Exception.class)
|
public void updateJob(JobSaveReqVO updateReqVO) throws SchedulerException {
|
validateCronExpression(updateReqVO.getCronExpression());
|
// 1.1 校验存在
|
JobDO job = validateJobExists(updateReqVO.getId());
|
// 1.2 只有开启状态，才可以修改.原因是，如果出暂停状态，修改 Quartz Job 时，会导致任务又开始执行
|
if (!job.getStatus().equals(JobStatusEnum.NORMAL.getStatus())) {
|
throw exception(JOB_UPDATE_ONLY_NORMAL_STATUS);
|
}
|
// 1.3 校验 JobHandler 是否存在
|
validateJobHandlerExists(updateReqVO.getHandlerName());
|
|
// 2. 更新 JobDO
|
JobDO updateObj = BeanUtils.toBean(updateReqVO, JobDO.class);
|
fillJobMonitorTimeoutEmpty(updateObj);
|
jobMapper.updateById(updateObj);
|
|
// 3. 更新 Job 到 Quartz 中
|
schedulerManager.updateJob(job.getHandlerName(), updateReqVO.getHandlerParam(), updateReqVO.getCronExpression(),
|
updateReqVO.getRetryCount(), updateReqVO.getRetryInterval());
|
}
|
|
private void validateJobHandlerExists(String handlerName) {
|
Object handler = SpringUtil.getBean(handlerName);
|
if (handler == null) {
|
throw exception(JOB_HANDLER_BEAN_NOT_EXISTS);
|
}
|
if (!(handler instanceof JobHandler)) {
|
throw exception(JOB_HANDLER_BEAN_TYPE_ERROR);
|
}
|
}
|
|
@Override
|
@Transactional(rollbackFor = Exception.class)
|
public void updateJobStatus(Long id, Integer status) throws SchedulerException {
|
// 校验 status
|
if (!containsAny(status, JobStatusEnum.NORMAL.getStatus(), JobStatusEnum.STOP.getStatus())) {
|
throw exception(JOB_CHANGE_STATUS_INVALID);
|
}
|
// 校验存在
|
JobDO job = validateJobExists(id);
|
// 校验是否已经为当前状态
|
if (job.getStatus().equals(status)) {
|
throw exception(JOB_CHANGE_STATUS_EQUALS);
|
}
|
// 更新 Job 状态
|
JobDO updateObj = JobDO.builder().id(id).status(status).build();
|
jobMapper.updateById(updateObj);
|
|
// 更新状态 Job 到 Quartz 中
|
if (JobStatusEnum.NORMAL.getStatus().equals(status)) { // 开启
|
schedulerManager.resumeJob(job.getHandlerName());
|
} else { // 暂停
|
schedulerManager.pauseJob(job.getHandlerName());
|
}
|
}
|
|
@Override
|
public void triggerJob(Long id) throws SchedulerException {
|
// 校验存在
|
JobDO job = validateJobExists(id);
|
|
// 触发 Quartz 中的 Job
|
schedulerManager.triggerJob(job.getId(), job.getHandlerName(), job.getHandlerParam());
|
}
|
|
@Override
|
@Transactional(rollbackFor = Exception.class)
|
public void syncJob() throws SchedulerException {
|
// 1. 查询 Job 配置
|
List<JobDO> jobList = jobMapper.selectList();
|
|
// 2. 遍历处理
|
for (JobDO job : jobList) {
|
// 2.1 先删除，再创建
|
schedulerManager.deleteJob(job.getHandlerName());
|
schedulerManager.addJob(job.getId(), job.getHandlerName(), job.getHandlerParam(), job.getCronExpression(),
|
job.getRetryCount(), job.getRetryInterval());
|
// 2.2 如果 status 为暂停，则需要暂停
|
if (Objects.equals(job.getStatus(), JobStatusEnum.STOP.getStatus())) {
|
schedulerManager.pauseJob(job.getHandlerName());
|
}
|
log.info("[syncJob][id({}) handlerName({}) 同步完成]", job.getId(), job.getHandlerName());
|
}
|
}
|
|
@Override
|
@Transactional(rollbackFor = Exception.class)
|
public void deleteJob(Long id) throws SchedulerException {
|
// 校验存在
|
JobDO job = validateJobExists(id);
|
// 更新
|
jobMapper.deleteById(id);
|
|
// 删除 Job 到 Quartz 中
|
schedulerManager.deleteJob(job.getHandlerName());
|
}
|
|
private JobDO validateJobExists(Long id) {
|
JobDO job = jobMapper.selectById(id);
|
if (job == null) {
|
throw exception(JOB_NOT_EXISTS);
|
}
|
return job;
|
}
|
|
private void validateCronExpression(String cronExpression) {
|
if (!CronUtils.isValid(cronExpression)) {
|
throw exception(JOB_CRON_EXPRESSION_VALID);
|
}
|
}
|
|
@Override
|
public JobDO getJob(Long id) {
|
return jobMapper.selectById(id);
|
}
|
|
@Override
|
public PageResult<JobDO> getJobPage(JobPageReqVO pageReqVO) {
|
return jobMapper.selectPage(pageReqVO);
|
}
|
|
private static void fillJobMonitorTimeoutEmpty(JobDO job) {
|
if (job.getMonitorTimeout() == null) {
|
job.setMonitorTimeout(0);
|
}
|
}
|
|
}
|
