欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Spring动态配置定时任务(启用、停用)

程序员文章站 2022-05-30 23:41:48
...
package com.csdn.taskk;


import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.scheduling.concurrent.ThreadPoolTaskScheduler;
import org.springframework.scheduling.support.CronTrigger;

import java.util.HashMap;
import java.util.concurrent.ScheduledFuture;

/**
 * 定时计费工具类
 * @version 1.0.0
 */
@Configuration
public class ScheduleConfig {
    private static final Logger logger = LoggerFactory.getLogger(ScheduleConfig.class);

    @Autowired
    private ThreadPoolTaskScheduler threadPoolTaskScheduler;

    /**
     * 存放所有启动定时任务对象
     */
    private static HashMap<String, ScheduledFuture<?>> scheduleMap = new HashMap<>();

    @Bean
    public ThreadPoolTaskScheduler threadPoolTaskScheduler() {
        ThreadPoolTaskScheduler executor = new ThreadPoolTaskScheduler();
        executor.initialize();
        executor.setPoolSize(20);
        executor.setThreadNamePrefix("taskExecutor-");
        executor.setWaitForTasksToCompleteOnShutdown(true);
        executor.setAwaitTerminationSeconds(60);
        return executor;
    }

    /**
     * 手动启动定时器
     * <p><b>修改人:</b><br>&nbsp;&nbsp; 2019年11月28日 19:52:56<br>
     * <p><b>修改说明:</b><br>&nbsp;&nbsp;<br>
     * @param crons 定时表达式
     * @return
     */
    public  void startCron(String crons) {
        try {
            //每次都会把之前所有的定时任务都关闭,根据新的状态重新启用一次,达到最新配置
            ScheduledFuture<?> scheduledFuture =     
            scheduleMap.get(this.getClass().getName());
           //此处只需要判断对象是否为空即可
            if (scheduledFuture != null) {
                scheduledFuture.cancel(true);
            }

            
            //重新开启定时任务
            String name = Thread.currentThread().getName();
            threadPoolTaskScheduler =  threadPoolTaskScheduler();
            ScheduledFuture<?> future =this.threadPoolTaskScheduler.schedule(new myTask(), new CronTrigger(crons));
            //将配置存入到map中 以便下次停用时可以停用
            scheduleMap.put(this.getClass().getName(), future);

        } catch (Exception e) {
            e.printStackTrace();
        }
    }



    /**
     * 执行计费
     * <p><b>修改人:</b><br>&nbsp;&nbsp; 2019年11月28日 19:02:10<br>
     * <p><b>修改说明:</b><br>&nbsp;&nbsp;<br>
     * @return
     */
    private class myTask implements Runnable {
        @Override
        public void run() {
            System.out.println("写业务处理的内容");
        }
    }



}

备注:在注入ThreadPoolTaskScheduler 的时候可能会报错,需要在spring配置文件进行配置 

Spring动态配置定时任务(启用、停用)