|
|
@@ -0,0 +1,316 @@
|
|
|
+
|
|
|
+package com.lxq.daemon.quartz.controller;
|
|
|
+
|
|
|
+import cn.hutool.core.util.StrUtil;
|
|
|
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
|
|
+import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
|
|
|
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
|
|
+import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
|
|
+import com.lxq.common.core.util.R;
|
|
|
+import com.lxq.common.log.annotation.SysLog;
|
|
|
+import com.lxq.common.security.util.SecurityUtils;
|
|
|
+import com.lxq.daemon.quartz.constants.SkyQuartzEnum;
|
|
|
+import com.lxq.daemon.quartz.entity.SysJob;
|
|
|
+import com.lxq.daemon.quartz.entity.SysJobLog;
|
|
|
+import com.lxq.daemon.quartz.service.SysJobLogService;
|
|
|
+import com.lxq.daemon.quartz.service.SysJobService;
|
|
|
+import com.lxq.daemon.quartz.util.TaskUtil;
|
|
|
+import com.lxq.common.excel.annotation.ResponseExcel;
|
|
|
+import io.swagger.v3.oas.annotations.Operation;
|
|
|
+import io.swagger.v3.oas.annotations.security.SecurityRequirement;
|
|
|
+import io.swagger.v3.oas.annotations.tags.Tag;
|
|
|
+import lombok.AllArgsConstructor;
|
|
|
+import lombok.extern.slf4j.Slf4j;
|
|
|
+import org.quartz.Scheduler;
|
|
|
+import org.quartz.SchedulerException;
|
|
|
+import org.springframework.http.HttpHeaders;
|
|
|
+import org.springframework.security.access.prepost.PreAuthorize;
|
|
|
+import org.springframework.web.bind.annotation.*;
|
|
|
+
|
|
|
+import java.util.List;
|
|
|
+
|
|
|
+/**
|
|
|
+ * @author LXQ
|
|
|
+ * @date 2021-01-27 10:04:42
|
|
|
+ * <p>
|
|
|
+ * 定时任务管理
|
|
|
+ */
|
|
|
+@RestController
|
|
|
+@AllArgsConstructor
|
|
|
+@RequestMapping("/sysJob")
|
|
|
+@Tag(description = "sys-job", name = "定时任务")
|
|
|
+@SecurityRequirement(name = HttpHeaders.AUTHORIZATION)
|
|
|
+@Slf4j
|
|
|
+public class SysJobController {
|
|
|
+
|
|
|
+ private final SysJobService sysJobService;
|
|
|
+
|
|
|
+ private final SysJobLogService sysJobLogService;
|
|
|
+
|
|
|
+ private final TaskUtil taskUtil;
|
|
|
+
|
|
|
+ private final Scheduler scheduler;
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 定时任务分页查询
|
|
|
+ * @param page 分页对象
|
|
|
+ * @param sysJob 定时任务调度表
|
|
|
+ */
|
|
|
+ @GetMapping("/page")
|
|
|
+ @Operation(description = "分页定时业务查询")
|
|
|
+ public R getSysJobPage(Page page, SysJob sysJob) {
|
|
|
+ LambdaQueryWrapper<SysJob> wrapper = Wrappers.<SysJob>lambdaQuery()
|
|
|
+ .like(StrUtil.isNotBlank(sysJob.getJobName()), SysJob::getJobName, sysJob.getJobName())
|
|
|
+ .like(StrUtil.isNotBlank(sysJob.getJobGroup()), SysJob::getJobGroup, sysJob.getJobGroup())
|
|
|
+ .eq(StrUtil.isNotBlank(sysJob.getJobStatus()), SysJob::getJobStatus, sysJob.getJobGroup())
|
|
|
+ .eq(StrUtil.isNotBlank(sysJob.getJobExecuteStatus()), SysJob::getJobExecuteStatus,
|
|
|
+ sysJob.getJobExecuteStatus());
|
|
|
+ return R.ok(sysJobService.page(page, wrapper));
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 通过id查询定时任务
|
|
|
+ * @param id id
|
|
|
+ * @return R
|
|
|
+ */
|
|
|
+ @GetMapping("/{id}")
|
|
|
+ @Operation(description = "唯一标识查询定时任务")
|
|
|
+ public R getById(@PathVariable("id") Long id) {
|
|
|
+ return R.ok(sysJobService.getById(id));
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 新增定时任务
|
|
|
+ * @param sysJob 定时任务调度表
|
|
|
+ * @return R
|
|
|
+ */
|
|
|
+ @SysLog("新增定时任务")
|
|
|
+ @PostMapping("add")
|
|
|
+ @PreAuthorize("@pms.hasPermission('job_sys_job_add')")
|
|
|
+ @Operation(description = "新增定时任务")
|
|
|
+ public R save(@RequestBody SysJob sysJob) {
|
|
|
+ long count = sysJobService.count(
|
|
|
+ Wrappers.query(SysJob.builder().jobName(sysJob.getJobName()).jobGroup(sysJob.getJobGroup()).build()));
|
|
|
+
|
|
|
+ if (count > 0) {
|
|
|
+ return R.failed("任务重复,请检查此组内是否已包含同名任务");
|
|
|
+ }
|
|
|
+ sysJob.setJobStatus(SkyQuartzEnum.JOB_STATUS_RELEASE.getType());
|
|
|
+ sysJob.setCreateBy(SecurityUtils.getUser().getUsername());
|
|
|
+ return R.ok(sysJobService.save(sysJob));
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 修改定时任务
|
|
|
+ * @param sysJob 定时任务调度表
|
|
|
+ * @return R
|
|
|
+ */
|
|
|
+ @SysLog("修改定时任务")
|
|
|
+ @PutMapping("update")
|
|
|
+ @PreAuthorize("@pms.hasPermission('job_sys_job_edit')")
|
|
|
+ @Operation(description = "修改定时任务")
|
|
|
+ public R updateById(@RequestBody SysJob sysJob) {
|
|
|
+ sysJob.setUpdateBy(SecurityUtils.getUser().getUsername());
|
|
|
+ SysJob querySysJob = this.sysJobService.getById(sysJob.getJobId());
|
|
|
+ if (SkyQuartzEnum.JOB_STATUS_NOT_RUNNING.getType().equals(querySysJob.getJobStatus())) {
|
|
|
+ // 如修改暂停的需更新调度器
|
|
|
+ this.taskUtil.addOrUpateJob(sysJob, scheduler);
|
|
|
+ sysJobService.updateById(sysJob);
|
|
|
+ }
|
|
|
+ else if (SkyQuartzEnum.JOB_STATUS_RELEASE.getType().equals(querySysJob.getJobStatus())) {
|
|
|
+ sysJobService.updateById(sysJob);
|
|
|
+ }
|
|
|
+ return R.ok();
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 通过id删除定时任务
|
|
|
+ * @param id id
|
|
|
+ * @return R
|
|
|
+ */
|
|
|
+ @SysLog("删除定时任务")
|
|
|
+ @DeleteMapping("remove/{id}")
|
|
|
+ @PreAuthorize("@pms.hasPermission('job_sys_job_del')")
|
|
|
+ @Operation(description = "唯一标识查询定时任务,暂停任务才能删除")
|
|
|
+ public R removeById(@PathVariable Long id) {
|
|
|
+ SysJob querySysJob = this.sysJobService.getById(id);
|
|
|
+ if (SkyQuartzEnum.JOB_STATUS_NOT_RUNNING.getType().equals(querySysJob.getJobStatus())) {
|
|
|
+ this.taskUtil.removeJob(querySysJob, scheduler);
|
|
|
+ this.sysJobService.removeById(id);
|
|
|
+ }
|
|
|
+ else if (SkyQuartzEnum.JOB_STATUS_RELEASE.getType().equals(querySysJob.getJobStatus())) {
|
|
|
+ this.sysJobService.removeById(id);
|
|
|
+ }
|
|
|
+ else if (SkyQuartzEnum.JOB_STATUS_RUNNING.getType().equals(querySysJob.getJobStatus())) {
|
|
|
+ return R.failed("暂停任务才能删除");
|
|
|
+ }
|
|
|
+ return R.ok();
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 暂停全部定时任务
|
|
|
+ */
|
|
|
+ @SysLog("暂停全部定时任务")
|
|
|
+ @PostMapping("/shutdownJobs")
|
|
|
+ @PreAuthorize("@pms.hasPermission('job_sys_job_shutdown_job')")
|
|
|
+ @Operation(description = "暂停全部定时任务")
|
|
|
+ public R shutdownJobs() {
|
|
|
+ taskUtil.pauseJobs(scheduler);
|
|
|
+ long count = this.sysJobService.count(
|
|
|
+ new LambdaQueryWrapper<SysJob>().eq(SysJob::getJobStatus, SkyQuartzEnum.JOB_STATUS_RUNNING.getType()));
|
|
|
+ if (count <= 0) {
|
|
|
+ return R.ok("无正在运行定时任务");
|
|
|
+ }
|
|
|
+ else {
|
|
|
+ // 更新定时任务状态条件,运行状态2更新为暂停状态2
|
|
|
+ this.sysJobService.update(
|
|
|
+ SysJob.builder().jobStatus(SkyQuartzEnum.JOB_STATUS_NOT_RUNNING.getType()).build(),
|
|
|
+ new UpdateWrapper<SysJob>().lambda()
|
|
|
+ .eq(SysJob::getJobStatus, SkyQuartzEnum.JOB_STATUS_RUNNING.getType()));
|
|
|
+ return R.ok("暂停成功");
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 启动全部定时任务
|
|
|
+ */
|
|
|
+ @SysLog("启动全部定时任务")
|
|
|
+ @PostMapping("/startJobs")
|
|
|
+ @PreAuthorize("@pms.hasPermission('job_sys_job_start_job')")
|
|
|
+ @Operation(description = "启动全部定时任务")
|
|
|
+ public R startJobs() {
|
|
|
+ // 更新定时任务状态条件,暂停状态3更新为运行状态2
|
|
|
+ this.sysJobService.update(SysJob.builder().jobStatus(SkyQuartzEnum.JOB_STATUS_RUNNING.getType()).build(),
|
|
|
+ new UpdateWrapper<SysJob>().lambda()
|
|
|
+ .eq(SysJob::getJobStatus, SkyQuartzEnum.JOB_STATUS_NOT_RUNNING.getType()));
|
|
|
+ taskUtil.startJobs(scheduler);
|
|
|
+ return R.ok();
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 刷新全部定时任务
|
|
|
+ */
|
|
|
+ @SysLog("刷新全部定时任务")
|
|
|
+ @PostMapping("/refreshJobs")
|
|
|
+ @PreAuthorize("@pms.hasPermission('job_sys_job_refresh_job')")
|
|
|
+ @Operation(description = "刷新全部定时任务")
|
|
|
+ public R refreshJobs() {
|
|
|
+ sysJobService.list().forEach((sysjob) -> {
|
|
|
+ if (SkyQuartzEnum.JOB_STATUS_RELEASE.getType().equals(sysjob.getJobStatus())
|
|
|
+ || SkyQuartzEnum.JOB_STATUS_DEL.getType().equals(sysjob.getJobStatus())) {
|
|
|
+ taskUtil.removeJob(sysjob, scheduler);
|
|
|
+ }
|
|
|
+ else if (SkyQuartzEnum.JOB_STATUS_RUNNING.getType().equals(sysjob.getJobStatus())
|
|
|
+ || SkyQuartzEnum.JOB_STATUS_NOT_RUNNING.getType().equals(sysjob.getJobStatus())) {
|
|
|
+ taskUtil.addOrUpateJob(sysjob, scheduler);
|
|
|
+ }
|
|
|
+ else {
|
|
|
+ taskUtil.removeJob(sysjob, scheduler);
|
|
|
+ }
|
|
|
+ });
|
|
|
+ return R.ok();
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 启动定时任务
|
|
|
+ * @param job 定时任务
|
|
|
+ */
|
|
|
+ @SysLog("启动定时任务")
|
|
|
+ @PostMapping("/startJob")
|
|
|
+ @PreAuthorize("@pms.hasPermission('job_sys_job_start_job')")
|
|
|
+ @Operation(description = "启动定时任务")
|
|
|
+ public R startJob(@RequestBody SysJob job) throws SchedulerException {
|
|
|
+ SysJob querySysJob = this.sysJobService.getById(job.getJobId());
|
|
|
+ if (querySysJob == null) {
|
|
|
+ return R.failed("无此定时任务,请确认");
|
|
|
+ }
|
|
|
+
|
|
|
+ // 如果定时任务不存在,强制状态为1已发布
|
|
|
+ if (!scheduler.checkExists(TaskUtil.getJobKey(querySysJob))) {
|
|
|
+ querySysJob.setJobStatus(SkyQuartzEnum.JOB_STATUS_RELEASE.getType());
|
|
|
+ log.warn("定时任务不在quartz中,任务id:{},强制状态为已发布并加入调度器", job.getJobId());
|
|
|
+ }
|
|
|
+
|
|
|
+ if (SkyQuartzEnum.JOB_STATUS_RELEASE.getType().equals(querySysJob.getJobStatus())) {
|
|
|
+ taskUtil.addOrUpateJob(querySysJob, scheduler);
|
|
|
+ }
|
|
|
+ else {
|
|
|
+ taskUtil.resumeJob(querySysJob, scheduler);
|
|
|
+ }
|
|
|
+ // 更新定时任务状态为运行状态2
|
|
|
+ this.sysJobService
|
|
|
+ .updateById(SysJob.builder().jobId(job.getJobId()).jobStatus(SkyQuartzEnum.JOB_STATUS_RUNNING.getType()).build());
|
|
|
+ return R.ok();
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 立刻执行定时任务
|
|
|
+ * @param job 任务
|
|
|
+ */
|
|
|
+ @SysLog("立刻执行定时任务")
|
|
|
+ @PostMapping("/runJob")
|
|
|
+ @PreAuthorize("@pms.hasPermission('job_sys_job_run_job')")
|
|
|
+ @Operation(description = "立刻执行定时任务")
|
|
|
+ public R runJob(@RequestBody SysJob job) throws SchedulerException {
|
|
|
+ SysJob querySysJob = this.sysJobService.getById(job.getJobId());
|
|
|
+
|
|
|
+ // 执行定时任务前判定任务是否在quartz中
|
|
|
+ if (!scheduler.checkExists(TaskUtil.getJobKey(querySysJob))) {
|
|
|
+ querySysJob.setJobStatus(SkyQuartzEnum.JOB_STATUS_NOT_RUNNING.getType());
|
|
|
+ log.warn("立刻执行定时任务-定时任务不在quartz中,任务id:{},强制状态为暂停并加入调度器", job.getJobId());
|
|
|
+ taskUtil.addOrUpateJob(querySysJob, scheduler);
|
|
|
+ }
|
|
|
+
|
|
|
+ return TaskUtil.runOnce(scheduler, querySysJob) ? R.ok() : R.failed();
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 暂停定时任务
|
|
|
+ */
|
|
|
+ @SysLog("暂停定时任务")
|
|
|
+ @PostMapping("/shutdownJob")
|
|
|
+ @PreAuthorize("@pms.hasPermission('job_sys_job_shutdown_job')")
|
|
|
+ @Operation(description = "暂停定时任务")
|
|
|
+ public R shutdownJob(@RequestBody SysJob job) {
|
|
|
+ SysJob querySysJob = this.sysJobService.getById(job.getJobId());
|
|
|
+ // 更新定时任务状态条件,运行状态2更新为暂停状态3
|
|
|
+ this.sysJobService.updateById(SysJob.builder()
|
|
|
+ .jobId(querySysJob.getJobId())
|
|
|
+ .jobStatus(SkyQuartzEnum.JOB_STATUS_NOT_RUNNING.getType())
|
|
|
+ .build());
|
|
|
+ taskUtil.pauseJob(querySysJob, scheduler);
|
|
|
+ return R.ok();
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 唯一标识查询定时执行日志
|
|
|
+ */
|
|
|
+ @GetMapping("/jobLog")
|
|
|
+ @Operation(description = "唯一标识查询定时执行日志")
|
|
|
+ public R getJobLog(Page<SysJobLog> page, SysJobLog sysJobLog) {
|
|
|
+ return R.ok(sysJobLogService.page(page, Wrappers.lambdaQuery(sysJobLog).orderByDesc(SysJobLog::getCreateTime)));
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 检验任务名称和任务组联合是否唯一
|
|
|
+ */
|
|
|
+ @GetMapping("/isValidTaskName")
|
|
|
+ @Operation(description = "检验任务名称和任务组联合是否唯一")
|
|
|
+ public R isValidTaskName(@RequestParam String jobName, @RequestParam String jobGroup) {
|
|
|
+ return this.sysJobService
|
|
|
+ .count(Wrappers.query(SysJob.builder().jobName(jobName).jobGroup(jobGroup).build())) > 0
|
|
|
+ ? R.failed("任务重复,请检查此组内是否已包含同名任务") : R.ok();
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 导出任务
|
|
|
+ * @param sysJob
|
|
|
+ */
|
|
|
+ @ResponseExcel
|
|
|
+ @GetMapping("/export")
|
|
|
+ @Operation(description = "导出任务")
|
|
|
+ public List<SysJob> export(SysJob sysJob) {
|
|
|
+ return sysJobService.list(Wrappers.query(sysJob));
|
|
|
+ }
|
|
|
+
|
|
|
+}
|