又被問了:Spring Boot 定時任務開啟後,怎麼符合條件自動停止?

来源:https://www.cnblogs.com/javastack/archive/2022/05/09/16249481.html
-Advertisement-
Play Games

背景 昨天,咱們的《知識星球:Java技術棧》裡面有粉絲向我提問: 問題大概就是: Spring Boot 定時任務開啟後,怎麼符合條件自動停止? 當時我有空,雖然已經給出了參考答案,但可能還有一些細節地方要註意的,另外,我也覺得這個問題特別有意思,現在特別拿出來整理下,分享下給大家。 1、自定義任 ...


背景

昨天,咱們的《知識星球: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開發手冊(嵩山版)》最新發佈,速速下載!

覺得不錯,別忘了隨手點贊+轉發哦!


您的分享是我們最大的動力!

-Advertisement-
Play Games
更多相關文章
一周排行
    -Advertisement-
    Play Games
  • 一:背景 準備開個系列來聊一下 PerfView 這款工具,熟悉我的朋友都知道我喜歡用 WinDbg,這東西雖然很牛,但也不是萬能的,也有一些場景他解決不了或者很難解決,這時候藉助一些其他的工具來輔助,是一個很不錯的主意。 很多朋友喜歡在項目中以記錄日誌的方式來監控項目的流轉情況,其實 CoreCL ...
  • 本來閑來無事,準備看看Dapper擴展的源碼學習學習其中的編程思想,同時整理一下自己代碼的單元測試,為以後的進一步改進打下基礎。 突然就發現問題了,源碼也不看了,開始改代碼,改了好久。 測試Dapper.LiteSql數據批量插入的時候,耗時20秒,感覺不正常,於是我測試了非Dapper版的Lite ...
  • 需求如下,在DEV框架項目中,需要在表格中增加一列顯示圖片,並且能編輯該列圖片,然後進行保存等操作,最終效果如下 這裡使用的是PictureEdit控制項來實現,打開DEV GridControl設計器,在ColumnEdit選擇PictureEdit: 綁定圖片代碼如下: DataTable dtO ...
  • 前兩天微軟偷偷更新了Visual Studio 2022 正式版版本 17.3 發佈,發佈摘要: MAUI 工作負荷 GA 生成 MAUI/Blazor CSS 熱重載支持 現在,你將能夠使用我們的新增功能在 Visual Studio 中使用每個更新試用一系列新功能。 選擇每個功能以瞭解有關特定功 ...
  • 航天和軍工領域的數字化轉型和建設正在積極推進,在與航天二院、航天三院、航天六院、航天九院、無線電廠、兵工廠等單位交流的過程中,用戶更聚焦試驗和生產過程中的痛點,迫切需要解決軟體平臺統一監測和控制設備及軟體與設備協同的問題。 ...
  • .NET 項目預設情況下 日誌是使用的 ILogger 介面,預設提供一下四種日誌記錄程式: 控制台 調試 EventSource EventLog 這四種記錄程式都是預設包含在 .NET 運行時庫中。關於這四種記錄程式的詳細介紹可以直接查看微軟的官方文檔 https://docs.microsof ...
  • 一:背景 上一篇我們聊到瞭如何去找 熱點函數,這一篇我們來看下當你的程式出現了 非托管記憶體泄漏 時如何去尋找可疑的代碼源頭,其實思路很簡單,就是在 HeapAlloc 或者 VirtualAlloc 時做 Hook 攔截,記錄它的調用棧以及分配的記憶體量, PerfView 會將這個 分配量 做成一個 ...
  • 背景 在 CI/CD 流程當中,測試是 CI 中很重要的部分。跟開發人員關係最大的就是單元測試,單元測試編寫完成之後,我們可以使用 IDE 或者 dot cover 等工具獲得單元測試對於業務代碼的覆蓋率。不過我們需要一個獨立的 CLI 工具,這樣我們才能夠在 Jenkins 的 CI 流程集成。 ...
  • 一、應用場景 大家在使用Mybatis進行開發的時候,經常會遇到一種情況:按照月份month將數據放在不同的表裡面,查詢數據的時候需要跟不同的月份month去查詢不同的表。 但是我們都知道,Mybatis是ORM持久層框架,即:實體關係映射,實體Object與資料庫表之間是存在一一對應的映射關係。比 ...
  • 我國目前並未出台專門針對網路爬蟲技術的法律規範,但在司法實踐中,相關判決已屢見不鮮,K 哥特設了“K哥爬蟲普法”專欄,本欄目通過對真實案例的分析,旨在提高廣大爬蟲工程師的法律意識,知曉如何合法合規利用爬蟲技術,警鐘長鳴,做一個守法、護法、有原則的技術人員。 案情介紹 深圳市快鴿互聯網科技有限公司 2 ...