Scheduler定时任务动态实现-记录

1.定义线程池

	@Bean("sp-syncScheduler")
    public ThreadPoolTaskScheduler syncScheduler() {
        ThreadPoolTaskScheduler syncScheduler = new ThreadPoolTaskScheduler();
        syncScheduler.setPoolSize(5);
        syncScheduler.setThreadGroupName("sp");
        syncScheduler.setThreadNamePrefix("spThread-");
        syncScheduler.initialize();
        return syncScheduler;
    }

2.定义任务服务类

@Service
@Slf4j
@RequiredArgsConstructor
public class SpDynamicTaskService {

    /**
     * 以下两个都是线程安全的集合类。
     */
    public Map<String, ScheduledFuture<?>> taskMap = new ConcurrentHashMap<>();

    public Map<String, SpTaskDo> taskDoMap = new HashMap<>();

    @Resource(name = "sp-syncScheduler")
    private ThreadPoolTaskScheduler syncScheduler;

    private final String IDENTIFY_TASK_TIME = "0/5 * * * * ?";

    private final String TASK_NAME = "任务名称";

    private final String CAPTURE_TASK_TIME = "0 30 8 * * ? ";


    /**
     * 查看已开启但还未执行的动态任务
     * @return
     */
    public SpTaskDto getTask() {
        Set<Map.Entry<String, SitePlotTaskDo>> set = taskDoMap.entrySet();
        SpTaskDto taskDto = new SpTaskDto();
        for (Map.Entry<String, SpTaskDto> stringSpTaskDoEntry : set) {
            taskDto.setTaskName(stringSitePlotTaskDoEntry.getValue().getTaskName());
            taskDto.setCron(stringSitePlotTaskDoEntry.getValue().getCron());
            taskDto.setLastExecutionTime(stringSitePlotTaskDoEntry.getValue().getLastExecutionTime());
            taskDto.setNextExecutionTime(stringSitePlotTaskDoEntry.getValue().getNextExecutionTime());
            return taskDto;
        }
        return taskDto;
    }

    @PostConstruct
    public void taskInit(){
        log.info("===================定时任务初始化===================");
        SpTaskDo spTaskDo = new SpTaskDo();
        spTaskDo.setTaskName(TASK_NAME);
        spTaskDo.setCron(CAPTURE_TASK_TIME);
        this.add(spTaskDo);
    }

    /**
     * 添加一个动态任务
     *
     * @param task
     * @return
     */
    public Response add(SpTaskDo task) {
        if (null != taskMap.get(task.getTaskName())) {
            stop(task.getTaskName());
        }
        ScheduledFuture<?> scheduleCapture = syncScheduler.schedule(getCaptureRunnable(task), new CronTrigger(task.getCron()));
        taskMap.put(task.getTaskName(), scheduleCapture);
        Date date = DateUtils.addSeconds(new Date(), Long.valueOf(scheduleCapture.getDelay(TimeUnit.SECONDS)).intValue());
        task.setLastExecutionTime(DateUtil.formatDateTime(date));
        task.setNextExecutionTime(task.getLastExecutionTime());
        taskDoMap.put(task.getTaskName(), task);
        return Response.buildSuccess();
    }


    /**
     * 定时任务执行
     * @param task
     * @return
     */
    public Runnable getCaptureRunnable(SpTaskDo task) {
        return () -> {
            log.info("---任务运行开始---");
            // TODO 做自己的业务
            task.setLastExecutionTime(task.getNextExecutionTime());
            LocalDateTime nextFirst = CronExpression.parse(task.getCron()).next(LocalDateTime.now());
            DateTimeFormatter dtf2 = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss");
            task.setNextExecutionTime(dtf2.format(nextFirst));
            taskDoMap.put(task.getTaskName(), task);
            log.info("---任务运行结束---");
        };
    }

    /**
     * 停止任务
     *
     * @param name
     * @return
     */
    public Response stop(String name) {
        if (null == taskMap.get(name)) {
            return Response.buildFailure("500", "任务不存在");
        }
        ScheduledFuture<?> scheduledFuture = taskMap.get(name);
        scheduledFuture.cancel(true);
        taskMap.remove(name);
        taskDoMap.remove(name);
        return Response.buildSuccess();
    }

3.任务实体

@Data
public class SpTaskDo {

    /**
     * 动态任务名
     */
    private String taskName;

    /**
     * 设定动态任务
     */
    private String cron;

    /**
     * 上次执行时间
     */
    private String lastExecutionTime;

    /**
     * 下次执行时间
     */
    private String nextExecutionTime;
}