1、自定义任务调度
首先覆盖 TaskSchedulingAutoConfiguration 自动配置类里面的 ThreadPoolTaskScheduler Bean：
@Data
@Component
class CustomTaskScheduler extends ThreadPoolTaskScheduler {
private Map> 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* 方法...
}
2、按条件自动停止任务
新建一个每 3 秒执行一次的任务：
@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);
取的的根据当前的方法名取就行了，这里是方法名，也可是类名＋方法名＋参数，防止重复。
来源：又被问了：Spring Boot 定时任务开启后，怎么符合条件自动停止？ - Java技术栈 - 博客园
“做程序员，圈子和学习最重要”因为有有了圈子可以让你少走弯路，扩宽人脉，扩展思路，学习他人的一些经验及学习方法！同时在这分享一下是一直以来整理的Java后端进阶笔记文档和学习资料免费分享给大家！需要资料的朋友私信我扣【06】
