平常咱们用Java写定时任务一般都是写的死代码,比如用spring框架的时候,写一个定时任务就加上一个@Scheduled的注解。
有这么一个业务场景:定时任务是用户在页面上配置的,用户配置好以后就会即时生成这么一个定时任务出来执行。
这种情况下用上面的死代码就会有问题,所以需要写代码让他把定时任务动态生成出来,代码如下:
java
@Autowired
private TaskScheduler taskScheduler;
//用来保存所有的定时任务
private Map<String, ScheduledFuture<?>> futureMap = new ConcurrentHashMap<>();
/*
* 启动定时任务
*/
public void startTask(CustomerTask task) {
//停止现有任务
stopTask(task.getId());
int seconds = task.getIntervalTime();
if(seconds <= 0) {
seconds = 60;//默认间隔时常
}
//生成新的定时任务
ScheduledFuture<?> scheduledFuture = taskScheduler.scheduleAtFixedRate(
() -> executeTask(task),
Duration.ofSeconds(seconds)
);
futureMap.put(task.getId(), scheduledFuture);
}
// 停止任务
public void stopTask(String taskId) {
if (futureMap.containsKey(taskId)) {
//停止现有任务
futureMap.get(taskId).cancel(true);
//从缓存中清除
futureMap.remove(taskId);
}
}
这里展示的是固定时间间隔的定时任务。
每生成一个定时任务就放到map中缓存起来,以便于有定时任务修改的时候可以停止现有的任务重新生成。