QMS/src/main/java/com/example/demo/controller/QuartzController.java

package com.example.demo.controller;

import com.example.demo.base.Constants;
import com.example.demo.domain.QrtzTask;
import com.example.demo.model.dto.QuartzTaskDTO;
import com.example.demo.service.JobService;
import com.example.demo.service.QrTaskService;
import com.example.demo.util.Response;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag;
import lombok.RequiredArgsConstructor;
import org.quartz.SchedulerException;
import org.springframework.context.annotation.Lazy;
import org.springframework.web.bind.annotation.*;

import java.util.List;


@RestController
@Tag(name = "任务管理")
@RequestMapping(Constants.QUARTZ_API + "/task")
@RequiredArgsConstructor(onConstructor_ = {@Lazy})
public class QuartzController {
    private final JobService jobService;
    private final QrTaskService qrTaskService;

    @Operation(summary = "添加任务")
    @PostMapping("/addTask")
    public String addTask(@RequestBody QuartzTaskDTO quartzTaskDTO) throws SchedulerException {
        //添加任务
        QrtzTask qrtzTask = qrTaskService.addQuartzTask(quartzTaskDTO);
        //任务添加完成后，判断是定时任务则再由定时任务定时添加该任务
        if (quartzTaskDTO.getTaskType().equals(1)) {
            jobService.addJob(qrtzTask);
        }
        return Response.success("添加任务完成");
    }

    @Operation(summary = "删除任务")
    @PostMapping("/deleteTasks")
    public String deleteTasks(@RequestBody List<Integer> taskIds) throws SchedulerException {
        return Response.success(qrTaskService.deleteQuartzTask(taskIds));
    }

    @Operation(summary = "查询任务")
    @PostMapping("/getTasks")
    public Response getTasks(@RequestBody QuartzTaskDTO quartzTaskDTO) {
        return Response.success2(qrTaskService.getTasks(quartzTaskDTO));
    }

    @Operation(summary = "更新任务")
    @PostMapping("/update")
    public void update(@RequestBody QuartzTaskDTO quartzTaskDTO) throws InterruptedException, SchedulerException {
        QrtzTask qrtzTask = qrTaskService.updateQuartzTask(quartzTaskDTO);
        jobService.updateJob(qrtzTask);

    }

    @Operation(summary = "暂停任务")
    @GetMapping("/pauseJob")
    public void pauseJob(Integer taskId) throws SchedulerException {
        jobService.pauseJob(taskId);

    }

    @Operation(summary = "继续任务")
    @GetMapping("/resumeJob")
    public void resumeJob(Integer taskId) throws SchedulerException {
        jobService.resumeJob(taskId);

    }

}