一、業務場景

先說業務場景，根據用戶輸入的cron表達式進行定時調度，舉個例子：如圖

根據用戶輸入的參數，生成表達式，然后定時去執行相應的業務邏輯。

二、Controller層（不是重點）

domain類

@TableName("sys_schedule")
@Data
public class Schedule {

    private static final long serialVersionUID = 1L;

    /**
     * 主鍵ID
     */
    @TableId(value = "id", type = IdType.AUTO)
    private Long id;

    /**
     * 類型: 時間， 周， 日
     */
    @TableField("type")
    private Integer type;

    /**
     * time 時間
     */
    @TableField("day_time")
    private String dayTime;

    /**
     * time 周
     */
    @TableField("weekday")
    private String weekday;

    /**
     * time 日
     */
    @TableField("monthday")
    private String monthday;

    /**
     * 選中狀態
     */
    @TableField("status")
    private Integer status;

    @TableField("cron")
    private String cron;

    @Override
    public String toString() {
        return "Schedule{" +
                "id=" + id +
                ", type=" + type +
                ", dayTime='" + dayTime + '\\'' +
                ", weekday='" + weekday + '\\'' +
                ", monthday='" + monthday + '\\'' +
                ", status=" + status +
                ", cron='" + cron + '\\'' +
                '}';
    }
}

controller方法

@Autowired
    private ScheduleService scheduleService;
 
  @PutMapping("/editItem")
    @ResponseBody
    public ResponseData editItem(Schedule schedule) {
        //ResponseData自己封裝的響應結果
        //param是你前端傳的，包含你的數據，如cron表達式
        this.scheduleService.update(schedule);
        return ResponseData.success();
    }

三、Service層

接口層

public interface ScheduleService {

    /**
     * 更新
     * @Date 2020-12-25
     */
    void update(Schedule param);

    /**
     * 業務處理
     * @Date 2020-12-25
     */
    void work();

}

實現層

@Service
public class ScheduleServiceImpl implements ScheduleService {
    
 @Override
    public void update(Schedule param) {
  //這里簡化了我自己的業務邏輯代碼，根據公司自己需要來寫即可
        //業務邏輯代碼
        
        //思路就是：更新數據庫的cron，類型等值，然后交給下面的定時工具類走定時方法
        this.scheduleMapper.updateById(param);
        String cron = param.getCron();
        //this就是你的業務執行service, 這里就是scheduleService
        ScheduleUtil.reset(new ScheduleTask(param.getId().toString(), this, null), cron);
    }
    
    @Override
    public void work() {
        //抓取數據的代碼
        System.out.println("業務代碼執行。。。");
    }

    
}

四、定時工具類

import org.springframework.scheduling.concurrent.ThreadPoolTaskScheduler;
import org.springframework.scheduling.support.CronTrigger;

import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.ScheduledFuture;

public class ScheduleUtil {

    private static ThreadPoolTaskScheduler threadPoolTaskScheduler = new    ThreadPoolTaskScheduler();
    private static Map<String, ScheduledFuture?> scheduledFutureMap = new HashMap<>();

    static {
        threadPoolTaskScheduler.initialize();
        System.out.println("定時任務線程池啟動");
    }

    /**
     * 啟動
     *
     * @param scheduleTask 定時任務
     * @param corn         執行時間表達式
     */
    public static boolean start(ScheduleTask scheduleTask, String corn) {
        System.out.println("啟動定時任務線程 taskId " + scheduleTask.getId());
        ScheduledFuture? scheduledFuture = threadPoolTaskScheduler
                .schedule(scheduleTask, new CronTrigger(corn));
        scheduledFutureMap.put(scheduleTask.getId(), scheduledFuture);
        return true;
    }

    /**
     * 取消
     *
     * @param scheduleTask 定時任務
     */
    public static boolean cancel(ScheduleTask scheduleTask) {
        System.out.println("關閉定時任務線程 taskId " + scheduleTask.getId());
        ScheduledFuture? scheduledFuture =       scheduledFutureMap.get(scheduleTask.getId());
        if (scheduledFuture != null && !scheduledFuture.isCancelled()) {
            scheduledFuture.cancel(false);
        }
        scheduledFutureMap.remove(scheduleTask.getId());
        return true;
    }

    /**
     * 修改
     *
     * @param scheduleTask 定時任務
     * @param corn         執行時間表達式
     */
    public static boolean reset(ScheduleTask scheduleTask, String corn) {
        //先取消定時任務
        cancel(scheduleTask);
        //然后啟動新的定時任務
        start(scheduleTask, corn);
        return true;
    }
}

五、工作線程

public class ScheduleTask implements Runnable {

    private static final int TIMEOUT = 30000;

    private String id;
    private ScheduleService service;

    public String getId() {
        return id;
    }

    /**
     * @param id      任務ID
     * @param service 業務類
     * @param keyword 關鍵字參數
     */
    public ScheduleTask(String id, ScheduleService service) {
        this.id = id;
        this.service = service;
    }

    @Override
    public void run() {
        //你自己的業務邏輯
        service.work();
    }
}

到這里結束了，這里主要是根據前臺傳的時間，進行生成cron，入庫，執行定時任務。但是有一個問題，就是如果項目重啟了的話，那么定時類那是無法加載到cron表達式的。那么接下來，加一個項目啟動時，把數據庫里的cron表達式重新灌入到定時類就可以。代碼如下：

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.CommandLineRunner;
import org.springframework.core.annotation.Order;
import org.springframework.stereotype.Component;

/**
 * 項目啟動時，加載
 * @Date: 2020/12/25:16:00
 **/
@Component
@Order(1)
public class initLineRunner implements CommandLineRunner {

    @Autowired
    private ScheduleService scheduleService;

    @Override
    public void run(String... args) throws Exception {
        System.out.println("init ........... ");
        //這里就是獲取你數據庫里的那個cron表達式
        Schedule param = scheduleService.get();
        String cron = param.getCron();
        ScheduleUtil.reset(new ScheduleTask(param.getId().toString(), scheduleService, null), cron);
    }
}

版權聲明：本文為博主原創文章，遵循 CC 4.0 BY-SA 版權協議，轉載請附上原文出處鏈接和本聲明。

本文鏈接：

https://blog.csdn.net/bbscz007/article/details/111695289