又被問了：Spring Boot 定時任務開啟后，怎么符合條件自動停止？

背景

昨天，咱們的《知識星球：Java技術堆疊》里面有粉絲向我提問：

問題大概就是：

Spring Boot 定時任務開啟后，怎么符合條件自動停止？

當時我有空，雖然已經給出了參考答案，但可能還有一些細節地方要注意的，另外，我也覺得這個問題特別有意思，現在特別拿出來整理下，分享下給大家，

1、自定義任務調度

首先覆寫 TaskSchedulingAutoConfiguration 自動配置類里面的 ThreadPoolTaskScheduler Bean：

/**
 * 自定義任務調度
 * 公眾號：Java技術堆疊
 */
@Data
@Component
class CustomTaskScheduler extends ThreadPoolTaskScheduler {

    private Map<Object, ScheduledFuture<?>> scheduledTasks = new IdentityHashMap<>();

    @Override
    public ScheduledFuture<?> schedule(Runnable task, Trigger trigger) {
        ScheduledFuture<?> future = super.schedule(task, trigger);
        this.putScheduledTasks(task, future);
        return future;
    }

    @Override
    public ScheduledFuture<?> scheduleAtFixedRate(Runnable task, long period) {
        ScheduledFuture<?> future = super.scheduleAtFixedRate(task, period);
        this.putScheduledTasks(task, future);
        return future;
    }

    @Override
    public ScheduledFuture<?> scheduleAtFixedRate(Runnable task, Date startTime, long period) {
        ScheduledFuture<?> future = super.scheduleAtFixedRate(task, startTime, period);
        this.putScheduledTasks(task, future);
        return future;
    }

    private void putScheduledTasks(Runnable task, ScheduledFuture<?> future) {
        ScheduledMethodRunnable runnable = (ScheduledMethodRunnable) task;
        scheduledTasks.put(runnable.getTarget(), future);
    }
    
    // 重寫所有 schedule* 方法...

}

Spring Boot 基礎就不介紹了，推薦下這個實戰教程：

https://github.com/javastacks/spring-boot-best-practice

重寫所有 schedule* 方法...

因為要停止一個任務，就必須呼叫 ScheduledFuture -> Future 介面中的 cancel 方法，

所以，思路就是在任務執行的時候，把任務所在的實體 Bean 和任務啟動后的 ScheduledFuture 維護到一個 Map 里面，然后需要停止的時候，從 Map 里面取出來，再進行 cancel 停止即可，

2、按條件自動停止任務

新建一個每 3 秒執行一次的任務：

/**
 * 按條件自動停止任務
 * 公眾號：Java技術堆疊
 */
@Slf4j
@Component
public class AutoStopTask {

    @Autowired
    private CustomTaskScheduler customTaskScheduler;

    private int count;

    @Scheduled(cron = "*/3 * * * * *")
    public void printTask() {
        log.info("公眾號Java技術堆疊，任務執行次數：{}", count + 1);

        count++;

        // 執行3次后自動停止
        if (count >= 3) {
            log.info("任務已執行指定次數，現在自動停止");
            boolean cancelled = customTaskScheduler.getScheduledTasks().get(this).cancel(true);
            
            // 停止后再次啟動
            if (cancelled) {
                count = 0;
                ScheduledMethodRunnable runnable = new ScheduledMethodRunnable(this, ReflectionUtils.findMethod(this.getClass(), "printTask"));
                customTaskScheduler.schedule(runnable, new CronTrigger("*/3 * * * * *"));
            }
        }
    }

}

這里是統計執行，當執行次數超過 3 次時就自動停止，如果需要再次啟動，上面也提供了引數代碼，

需要注意的是，自定義調度里面系結的是實體 Bean 和 Future 的關系，所以僅限 Bean 中的單個任務，如果一個 Bean 維護了多個任務，最后一個任務的啟動就會覆寫之前的，

如果要維護 Bean 中的多個任務，自動停止該怎么做呢？

答案就是把任務的方法名和 Future 關聯起來：

/scheduledMethodTasks.put(runnable.getMethod(), future);

取的的根據當前的方法名取就行了，這里是方法名，也可是類名＋方法名＋引數，防止重復，

關于這個問題，你還有哪些實作方案呢？歡迎留言分享！

總結

本文完整示例代碼已經上傳到 Github：

https://github.com/javastacks/spring-boot-best-practice

如果你感興趣的話，可以 Star 學習，后續會持續更新，

最后打一波咱們星球的廣告：

• 如果你也有許多疑難問題、疑惑無從解答，可以加入我們的《知識星球：Java技術堆疊》，只要堆疊長我知道的我都會解答，還有 2500+ 球友也在；
• 如果你學習沒有方向、作業沒有提升，《知識星球：Java技術堆疊》里面也沉淀了大量技術知識、學習資料、面試題、簡歷模板等，加入就值回門票；

星球原價是 199 的，現在是活動優惠價 159 元，正是加入的好時候，過段時間肯定會恢復原價的，

公眾號、微信上面好友太多，不可能一一回答每個人的問題，時間精力不允許啊（回答是情分，不回答是義務），所以我就創建了《知識星球：Java技術堆疊》，知識付費，現在就變成了一種責任了，堆疊長是實打實的純技術人，不玩套路，也從不玩虛的，咱們的星球絕對物超所值，歡迎加入一起學習吧，

最后，如果你想關注和學習最新、最主流的 Java 技術，可以持續關注公眾號Java技術堆疊，公眾號第一時間推送，

著作權宣告： 本文系公眾號 "Java技術堆疊" 原創，轉載、參考本文內容請注明出處，抄襲、洗稿一律投訴侵權，后果自負，并保留追究其法律責任的權利，

近期熱文推薦：

1.1,000+ 道 Java面試題及答案整理(2022最新版)

2.勁爆！Java 協程要來了，，，

3.Spring Boot 2.x 教程，太全了！

4.別再寫滿屏的爆爆爆炸類了，試試裝飾器模式，這才是優雅的方式！！

5.《Java開發手冊（嵩山版）》最新發布，速速下載！

覺得不錯，別忘了隨手點贊+轉發哦！

標籤：Java

上一篇：【面試普通人VS高手系列】說一說Mybatis里面的快取機制

下一篇：批量上傳檔案或者上傳大檔案時 gateWay報錯DataBufferLimitException: Exceeded limit on max bytes to buffer : 262144