SpringBoot定时任务开启后，怎么符合条件自动停止？

转公众号&＃xff1a;java技术栈

1、自定义任务调度

首先覆盖 TaskSchedulingAutoConfiguration 自动配置类里面的 ThreadPoolTaskScheduler Bean&＃xff1a;

/*** 自定义任务调度* */
&＃64;Data
&＃64;Component
class CustomTaskScheduler extends ThreadPoolTaskScheduler {private Map> scheduledTasks &＃61; new IdentityHashMap<>();&＃64;Overridepublic ScheduledFuture schedule(Runnable task, Trigger trigger) {ScheduledFuture future &＃61; super.schedule(task, trigger);this.putScheduledTasks(task, future);return future;}&＃64;Overridepublic ScheduledFuture scheduleAtFixedRate(Runnable task, long period) {ScheduledFuture future &＃61; super.scheduleAtFixedRate(task, period);this.putScheduledTasks(task, future);return future;}&＃64;Overridepublic ScheduledFuture scheduleAtFixedRate(Runnable task, Date startTime, long period) {ScheduledFuture future &＃61; super.scheduleAtFixedRate(task, startTime, period);this.putScheduledTasks(task, future);return future;}private void putScheduledTasks(Runnable task, ScheduledFuture future) {ScheduledMethodRunnable runnable &＃61; (ScheduledMethodRunnable) task;scheduledTasks.put(runnable.getTarget(), future);}// 重写所有 schedule* 方法...}


Spring Boot 基础就不介绍了&＃xff0c;推荐下这个实战教程&＃xff1a;

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

重写所有 schedule* 方法...

因为要停止一个任务&＃xff0c;就必须调用 ScheduledFuture -> Future 接口中的 cancel 方法。

所以&＃xff0c;思路就是在任务执行的时候&＃xff0c;把任务所在的实例 Bean 和任务启动后的 ScheduledFuture 维护到一个 Map 里面&＃xff0c;然后需要停止的时候&＃xff0c;从 Map 里面取出来&＃xff0c;再进行 cancel 停止即可。

2、按条件自动停止任务

新建一个每 3 秒执行一次的任务&＃xff1a;

/*** 按条件自动停止任务* 公众号&＃xff1a;Java技术栈*/
&＃64;Slf4j
&＃64;Component
public class AutoStopTask {&＃64;Autowiredprivate CustomTaskScheduler customTaskScheduler;private int count;&＃64;Scheduled(cron &＃61; "*/3 * * * * *")public void printTask() {log.info("公众号Java技术栈&＃xff0c;任务执行次数&＃xff1a;{}", count &＃43; 1);count&＃43;&＃43;;// 执行3次后自动停止if (count >&＃61; 3) {log.info("任务已执行指定次数&＃xff0c;现在自动停止");boolean cancelled &＃61; customTaskScheduler.getScheduledTasks().get(this).cancel(true);// 停止后再次启动if (cancelled) {count &＃61; 0;ScheduledMethodRunnable runnable &＃61; new ScheduledMethodRunnable(this, ReflectionUtils.findMethod(this.getClass(), "printTask"));customTaskScheduler.schedule(runnable, new CronTrigger("*/3 * * * * *"));}}}}


这里是统计执行&＃xff0c;当执行次数超过 3 次时就自动停止。如果需要再次启动&＃xff0c;上面也提供了参数代码。

需要注意的是&＃xff0c;自定义调度里面绑定的是实例 Bean 和 Future 的关系&＃xff0c;所以仅限 Bean 中的单个任务&＃xff0c;如果一个 Bean 维护了多个任务&＃xff0c;最后一个任务的启动就会覆盖之前的。

如果要维护 Bean 中的多个任务&＃xff0c;自动停止该怎么做呢&＃xff1f;

答案就是把任务的方法名和 Future 关联起来&＃xff1a;

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


取的的根据当前的方法名取就行了&＃xff0c;这里是方法名&＃xff0c;也可是类名&＃xff0b;方法名&＃xff0b;参数&＃xff0c;防止重复。