springboot之配置动态定时任务

导入依赖:

        <dependency>
            <groupId>org.mybatis.spring.boot</groupId>
            <artifactId>mybatis-spring-boot-starter</artifactId>
        </dependency>

基于接口配置:

初始化bean实例是当cron配置存在时启动定时任务;
修改配置cron,先停止在重新启动。
具体myTaskScheduledConfig可参考下方

@Slf4j
@RestController
@RequestMapping("/task")
public class DynamicTaskController {

    @Autowired
    private ThreadPoolTaskScheduler threadPoolTaskScheduler;

    private ScheduledFuture<?> future;

    @Autowired
    private CdsServiceImpl cdsService;

    @Bean
    public ThreadPoolTaskScheduler threadPoolTaskScheduler() {
        return new ThreadPoolTaskScheduler();
    }

    @PostConstruct
    private void init(){
        if (Strings.isNotEmpty(myTaskScheduledConfig.getCron())) {
            log.info("动态定时任务 init->start。。。{}", DateUtils.getCurrentTime());
            startTask();
        }

    }

    @PostMapping("/start")
    public ResponseDTO startCron() {
        if (Strings.isEmpty(myTaskScheduledConfig.getCron())) {
            CronConfigEntity cron = cdsService.getCron();
            if (Objects.isNull(cron)) {
                return ResponseDTO.onError("定时任务配置信息为空");
            }
            myTaskScheduledConfig.setCron(cron.getCorn());
        }
        startTask();
        return ResponseDTO.onSuccess("start success");
    }

    @PostMapping("/stop")
    public ResponseDTO stopCron() {
        stopTask();
        return ResponseDTO.onSuccess("stop success");
    }

    @PostMapping("/updateCron")
    public ResponseDTO updateCron(@RequestBody TaskVO vo) {
    	//修改配置 可以是mysql中的 也可以使用堆中的
        if (cdsService.updateCron(vo.getCron())) {
            myTaskScheduledConfig.setCron(vo.getCron());
            stopTask();
            startTask();
            return ResponseDTO.onSuccess("success");
        }
        return ResponseDTO.onError(ErrorCode.OPERATE_ERROR);

    }


    private void startTask() {
        log.info("动态定时任务 start。。。{}", DateUtils.getCurrentTime());
        future = threadPoolTaskScheduler.schedule(() -> {
            log.info("动态定时任务执行。。。{}", DateUtils.getCurrentTime());
           
            //业务代码
            log.info("动态定时任务执行结束。。。{}", DateUtils.getCurrentTime());
        }, new Trigger() {
            @Override
            public Date nextExecutionTime(TriggerContext triggerContext) {
                return new CronTrigger(myTaskScheduledConfig.getCron()).nextExecutionTime(triggerContext);
            }
        });
    }

    private void stopTask() {
        if (future != null) {
            future.cancel(true);
        }
        log.info("动态定时任务 stop。。。{}", DateUtils.getCurrentTime());
    }


}

定时任务时间间隔配置MyTaskScheduledConfig

@Data
@Configuration
@EqualsAndHashCode(callSuper = false)
public class MyTaskScheduledConfig {

    private boolean enable;

    private String cron;

    @Autowired
    private CronConfigRepository cronConfigRepository;

    @PostConstruct
    private void init(){
        CronConfigEntity config = cronConfigRepository.findFirstByOrderByIdDesc();
        if (Objects.isNotNull(config)){
            this.enable = config.getIsEnable();
            this.cron = config.getCorn();
        }

    }
}

总结:

目前这个方式已经用在公司实际生产环境,虽然对比其他使用@Scheduled的静态定时任务没有那么优雅,也没有Quartz那么方便,当是我个人觉得能不添加其他依赖就尽量不添加,而且其本身也比较灵活。只是由于本人的技术水平有限暂时不知道怎么做到直接配置时间间隔的方式,可能需要自己做处理,等待后续研究出来再和大家分享。

上一篇:挖矿病毒排查


下一篇:定时任务不在硬编码,动态定时刷起来