build: 定时任务公共方法
This commit is contained in:
fit2-zhao
fit2-zhao
parent
b18e2e85c4
commit
95c1f764f6
|
|
@ -0,0 +1,28 @@
|
|||
package io.metersphere.sdk.sechedule;
|
||||
|
||||
import io.metersphere.sdk.util.LogUtils;
|
||||
import org.quartz.*;
|
||||
|
||||
public abstract class BaseScheduleJob implements Job {
|
||||
|
||||
protected String resourceId;
|
||||
|
||||
protected String userId;
|
||||
|
||||
protected String expression;
|
||||
|
||||
@Override
|
||||
public void execute(JobExecutionContext context) throws JobExecutionException {
|
||||
JobKey jobKey = context.getTrigger().getJobKey();
|
||||
JobDataMap jobDataMap = context.getJobDetail().getJobDataMap();
|
||||
this.resourceId = jobDataMap.getString("resourceId");
|
||||
this.userId = jobDataMap.getString("userId");
|
||||
this.expression = jobDataMap.getString("expression");
|
||||
|
||||
LogUtils.info(jobKey.getGroup() + " Running: " + resourceId);
|
||||
LogUtils.info("CronExpression: " + expression);
|
||||
businessExecute(context);
|
||||
}
|
||||
|
||||
protected abstract void businessExecute(JobExecutionContext context);
|
||||
}
|
||||
|
|
@ -0,0 +1,80 @@
|
|||
package io.metersphere.sdk.sechedule;
|
||||
|
||||
import io.metersphere.system.domain.Schedule;
|
||||
import io.metersphere.system.domain.ScheduleExample;
|
||||
import io.metersphere.system.mapper.ScheduleMapper;
|
||||
import jakarta.annotation.Resource;
|
||||
import org.apache.commons.collections.CollectionUtils;
|
||||
import org.quartz.JobKey;
|
||||
import org.quartz.TriggerKey;
|
||||
import org.springframework.transaction.annotation.Transactional;
|
||||
|
||||
import java.util.List;
|
||||
import java.util.UUID;
|
||||
|
||||
@Transactional(rollbackFor = Exception.class)
|
||||
public class BaseScheduleService {
|
||||
|
||||
@Resource
|
||||
private ScheduleMapper scheduleMapper;
|
||||
@Resource
|
||||
private ScheduleManager scheduleManager;
|
||||
|
||||
public void addSchedule(Schedule schedule) {
|
||||
schedule.setId(UUID.randomUUID().toString());
|
||||
schedule.setCreateTime(System.currentTimeMillis());
|
||||
schedule.setUpdateTime(System.currentTimeMillis());
|
||||
scheduleMapper.insert(schedule);
|
||||
}
|
||||
|
||||
|
||||
public Schedule getSchedule(String ScheduleId) {
|
||||
return scheduleMapper.selectByPrimaryKey(ScheduleId);
|
||||
}
|
||||
|
||||
public int editSchedule(Schedule schedule) {
|
||||
schedule.setUpdateTime(System.currentTimeMillis());
|
||||
return scheduleMapper.updateByPrimaryKeySelective(schedule);
|
||||
}
|
||||
|
||||
public Schedule getScheduleByResource(String resourceId, String job) {
|
||||
ScheduleExample example = new ScheduleExample();
|
||||
example.createCriteria().andResourceIdEqualTo(resourceId).andJobEqualTo(job);
|
||||
List<Schedule> schedules = scheduleMapper.selectByExample(example);
|
||||
if (CollectionUtils.isNotEmpty(schedules)) {
|
||||
return schedules.get(0);
|
||||
}
|
||||
return null;
|
||||
}
|
||||
|
||||
public int deleteByResourceId(String resourceId, String group) {
|
||||
ScheduleExample scheduleExample = new ScheduleExample();
|
||||
scheduleExample.createCriteria().andResourceIdEqualTo(resourceId);
|
||||
removeJob(resourceId, group);
|
||||
return scheduleMapper.deleteByExample(scheduleExample);
|
||||
}
|
||||
|
||||
public int deleteByResourceIds(List<String> resourceIds, String group) {
|
||||
ScheduleExample scheduleExample = new ScheduleExample();
|
||||
scheduleExample.createCriteria().andResourceIdIn(resourceIds);
|
||||
for (String resourceId : resourceIds) {
|
||||
removeJob(resourceId, group);
|
||||
}
|
||||
return scheduleMapper.deleteByExample(scheduleExample);
|
||||
}
|
||||
|
||||
|
||||
public int deleteByProjectId(String projectId) {
|
||||
ScheduleExample scheduleExample = new ScheduleExample();
|
||||
scheduleExample.createCriteria().andProjectIdEqualTo(projectId);
|
||||
List<Schedule> schedules = scheduleMapper.selectByExample(scheduleExample);
|
||||
schedules.forEach(item -> {
|
||||
removeJob(item.getKey(), item.getJob());
|
||||
});
|
||||
return scheduleMapper.deleteByExample(scheduleExample);
|
||||
}
|
||||
|
||||
private void removeJob(String key, String job) {
|
||||
scheduleManager.removeJob(new JobKey(key, job), new TriggerKey(key, job));
|
||||
}
|
||||
}
|
||||
|
|
@ -0,0 +1,155 @@
|
|||
package io.metersphere.sdk.sechedule;
|
||||
|
||||
import io.metersphere.sdk.exception.MSException;
|
||||
import io.metersphere.sdk.util.LogUtils;
|
||||
import io.metersphere.system.domain.Schedule;
|
||||
import jakarta.annotation.Resource;
|
||||
import org.quartz.*;
|
||||
|
||||
/**
|
||||
* @Description: 定时任务管理类
|
||||
*/
|
||||
public class ScheduleManager {
|
||||
|
||||
@Resource
|
||||
private Scheduler scheduler;
|
||||
|
||||
/**
|
||||
* 添加 simpleJob
|
||||
*/
|
||||
public void addSimpleJob(JobKey jobKey, TriggerKey triggerKey, Class<? extends Job> cls, int repeatIntervalTime, JobDataMap jobDataMap)
|
||||
throws SchedulerException {
|
||||
|
||||
JobBuilder jobBuilder = JobBuilder.newJob(cls).withIdentity(jobKey);
|
||||
if (jobDataMap != null) {
|
||||
jobBuilder.usingJobData(jobDataMap);
|
||||
}
|
||||
|
||||
SimpleTrigger trigger = TriggerBuilder.newTrigger().withIdentity(triggerKey)
|
||||
.withSchedule(SimpleScheduleBuilder.simpleSchedule().withIntervalInHours(repeatIntervalTime).repeatForever())
|
||||
.startNow().build();
|
||||
|
||||
scheduler.scheduleJob(jobBuilder.build(), trigger);
|
||||
}
|
||||
|
||||
/**
|
||||
* 添加 cronJob
|
||||
*/
|
||||
public void addCronJob(JobKey jobKey, TriggerKey triggerKey, Class<? extends Job> jobClass, String cron, JobDataMap jobDataMap) {
|
||||
try {
|
||||
LogUtils.info("addCronJob: " + triggerKey.getName() + "," + triggerKey.getGroup());
|
||||
JobBuilder jobBuilder = JobBuilder.newJob(jobClass).withIdentity(jobKey);
|
||||
if (jobDataMap != null) {
|
||||
jobBuilder.usingJobData(jobDataMap);
|
||||
}
|
||||
|
||||
TriggerBuilder<Trigger> triggerBuilder = TriggerBuilder.newTrigger();
|
||||
triggerBuilder.withIdentity(triggerKey);
|
||||
triggerBuilder.startNow();
|
||||
triggerBuilder.withSchedule(CronScheduleBuilder.cronSchedule(cron));
|
||||
CronTrigger trigger = (CronTrigger) triggerBuilder.build();
|
||||
scheduler.scheduleJob(jobBuilder.build(), trigger);
|
||||
|
||||
} catch (Exception e) {
|
||||
LogUtils.error(e);
|
||||
throw new MSException("定时任务配置异常: " + e.getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
public void addCronJob(JobKey jobKey, TriggerKey triggerKey, Class<? extends Job> jobClass, String cron) {
|
||||
addCronJob(jobKey, triggerKey, jobClass, cron, null);
|
||||
}
|
||||
|
||||
/**
|
||||
* 修改 cronTrigger
|
||||
*/
|
||||
public void modifyCronJobTime(TriggerKey triggerKey, String cron) throws SchedulerException {
|
||||
|
||||
LogUtils.info("modifyCronJobTime: " + triggerKey.getName() + "," + triggerKey.getGroup());
|
||||
try {
|
||||
CronTrigger trigger = (CronTrigger) scheduler.getTrigger(triggerKey);
|
||||
if (trigger == null) {
|
||||
return;
|
||||
}
|
||||
|
||||
String oldTime = trigger.getCronExpression();
|
||||
if (!oldTime.equalsIgnoreCase(cron)) {
|
||||
/* 方式一 ：调用 rescheduleJob 开始 */
|
||||
TriggerBuilder<Trigger> triggerBuilder = TriggerBuilder.newTrigger();// 触发器
|
||||
triggerBuilder.withIdentity(triggerKey);// 触发器名,触发器组
|
||||
triggerBuilder.startNow(); // 立即执行
|
||||
triggerBuilder.withSchedule(CronScheduleBuilder.cronSchedule(cron)); // 触发器时间设定
|
||||
trigger = (CronTrigger) triggerBuilder.build(); // 创建Trigger对象
|
||||
scheduler.rescheduleJob(triggerKey, trigger); // 修改一个任务的触发时间
|
||||
}
|
||||
} catch (Exception e) {
|
||||
throw new RuntimeException(e);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* @Description: 根据job和trigger删除任务
|
||||
*/
|
||||
public void removeJob(JobKey jobKey, TriggerKey triggerKey) {
|
||||
try {
|
||||
LogUtils.info("RemoveJob: " + jobKey.getName() + "," + jobKey.getGroup());
|
||||
scheduler.pauseTrigger(triggerKey);
|
||||
scheduler.unscheduleJob(triggerKey);
|
||||
scheduler.deleteJob(jobKey);
|
||||
} catch (Exception e) {
|
||||
LogUtils.error(e.getMessage(), e);
|
||||
throw new RuntimeException(e);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
public static void startJobs(Scheduler schedule) {
|
||||
try {
|
||||
schedule.start();
|
||||
} catch (Exception e) {
|
||||
LogUtils.error(e);
|
||||
throw new RuntimeException(e);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
public void shutdownJobs(Scheduler schedule) {
|
||||
try {
|
||||
if (!schedule.isShutdown()) {
|
||||
schedule.shutdown();
|
||||
}
|
||||
} catch (Exception e) {
|
||||
LogUtils.error(e.getMessage(), e);
|
||||
throw new RuntimeException(e);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 添加或修改 cronJob
|
||||
*/
|
||||
public void addOrUpdateCronJob(JobKey jobKey, TriggerKey triggerKey, Class<? extends Job> jobClass, String cron, JobDataMap jobDataMap)
|
||||
throws SchedulerException {
|
||||
|
||||
LogUtils.info("AddOrUpdateCronJob: " + jobKey.getName() + "," + triggerKey.getGroup());
|
||||
|
||||
if (scheduler.checkExists(triggerKey)) {
|
||||
modifyCronJobTime(triggerKey, cron);
|
||||
} else {
|
||||
addCronJob(jobKey, triggerKey, jobClass, cron, jobDataMap);
|
||||
}
|
||||
}
|
||||
|
||||
public void addOrUpdateCronJob(JobKey jobKey, TriggerKey triggerKey, Class<? extends Job> jobClass, String cron) throws SchedulerException {
|
||||
addOrUpdateCronJob(jobKey, triggerKey, jobClass, cron, null);
|
||||
}
|
||||
|
||||
public JobDataMap getDefaultJobDataMap(Schedule schedule, String expression, String userId) {
|
||||
JobDataMap jobDataMap = new JobDataMap();
|
||||
jobDataMap.put("resourceId", schedule.getResourceId());
|
||||
jobDataMap.put("expression", expression);
|
||||
jobDataMap.put("userId", userId);
|
||||
jobDataMap.put("config", schedule.getConfig());
|
||||
jobDataMap.put("projectId", schedule.getProjectId());
|
||||
return jobDataMap;
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue