springboot中@scheduled的开关机制（针对全局定时任务）

最新推荐文章于 2024-08-09 07:15:00 发布

qq_37826571

最新推荐文章于 2024-08-09 07:15:00 发布

阅读量6.7k

点赞数 2

文章标签： java

本文链接：https://blog.csdn.net/qq_37826571/article/details/105117810

版权

在SpringBoot中，利用@scheduled实现定时任务时，如何实现在不同环境中控制所有定时任务的开启或关闭？本文介绍一种解决方案，通过了解ScheduledTaskRegistrar类，清空扫描的定时任务类别，达到统一控制的效果。

摘要由CSDN通过智能技术生成

目前遇到的问题：

我们可以在springboot中使用@scheduled注解来完成定时任务。我现在有两台机器部署同一个应用。

一台希望开启全部的定时器，另一台希望关闭全部定时器，考虑到上一篇博客

（https://blog.csdn.net/qq_37826571/article/details/105097249）只能对一个定时任务进行开关，我不能把之前的定时任务都分离

出来一个个的改代码，太麻烦了，怎么办？

解决方案：

那就要从源头去解决，这里大家要了解ScheduledTaskRegistrar 这个类，我这里为了大家方便，给大家展示出来这个类。

//
// Source code recreated from a .class file by IntelliJ IDEA
// (powered by Fernflower decompiler)
//

package org.springframework.scheduling.config;

import java.util.ArrayList;
import java.util.Collections;
import java.util.Date;
import java.util.HashMap;
import java.util.Iterator;
import java.util.LinkedHashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.Map.Entry;
import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledExecutorService;
import org.springframework.beans.factory.DisposableBean;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.scheduling.TaskScheduler;
import org.springframework.scheduling.Trigger;
import org.springframework.scheduling.concurrent.ConcurrentTaskScheduler;
import org.springframework.util.Assert;
import org.springframework.util.CollectionUtils;

public class ScheduledTaskRegistrar implements InitializingBean, DisposableBean {
    private TaskScheduler taskScheduler;
    private ScheduledExecutorService localExecutor;
    private List<TriggerTask> triggerTasks;
    private List<CronTask> cronTasks;
    private List<IntervalTask> fixedRateTasks;
    private List<IntervalTask> fixedDelayTasks;
    private final Map<Task, ScheduledTask> unresolvedTasks = new HashMap(16);
    private final Set<ScheduledTask> scheduledTasks = new LinkedHashSet(16);

    public ScheduledTaskRegistrar() {
    }

    public void setTaskScheduler(TaskScheduler taskScheduler) {
        Assert.notNull(taskScheduler, "TaskScheduler must not be null");
        this.taskScheduler = taskScheduler;
    }

    public void setScheduler(Object scheduler) {
        Assert.notNull(scheduler, "Scheduler object must not be null");
        if (scheduler instanceof TaskScheduler) {
            this.taskScheduler = (TaskScheduler)scheduler;
        } else {
            if (!(scheduler instanceof ScheduledExecutorService)) {
                throw new IllegalArgumentException("Unsupported scheduler type: " + scheduler.getClass());
            }

            this.taskScheduler = new ConcurrentTaskScheduler((ScheduledExecutorService)scheduler);
        }

    }

    public TaskScheduler getScheduler() {
        return this.taskScheduler;
    }

    public void setTriggerTasks(Map<Runnable, Tri