spring手动设置定时任务

spring手动设置定时任务

设置任务池配置

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.scheduling.annotation.SchedulingConfigurer;
import org.springframework.scheduling.concurrent.ThreadPoolTaskScheduler;
import org.springframework.scheduling.config.ScheduledTaskRegistrar;


/**
 * @desc: 程序启动后手动维护定时任务
 * <p>
 * by  zhang.Z
 */
@Configuration
public class MultiTimerConfig implements SchedulingConfigurer {


    /**
     * 设置定时任务使用的线程池
     *
     * @return
     */
    @Bean(destroyMethod = "shutdown", name = "taskScheduler")
    public ThreadPoolTaskScheduler taskScheduler() {
        ThreadPoolTaskScheduler scheduler = new ThreadPoolTaskScheduler();
        scheduler.setPoolSize(10); //设置任务池的大小
        scheduler.setThreadNamePrefix("task-"); //设置线程的前缀名
        scheduler.setAwaitTerminationSeconds(60);//设置超时时间
        scheduler.setWaitForTasksToCompleteOnShutdown(true);
        return scheduler;
    }


    /**
     * 将配置的线程池注册配置进来
     * @param scheduledTaskRegistrar
     */
    @Override
    public void configureTasks(ScheduledTaskRegistrar scheduledTaskRegistrar) {
        scheduledTaskRegistrar.setTaskScheduler(taskScheduler());
    }

}

cron触发设置器

import org.springframework.scheduling.Trigger;
import org.springframework.scheduling.TriggerContext;
import org.springframework.scheduling.support.CronTrigger;

import java.util.Date;

/**
 * @desc: cron触发设置器
 * <p>
 * by  zhang.Z
 */
public class TriggerImpl implements Trigger {

    private String cronExpress;

    public TriggerImpl(String cronExpress) {
        this.cronExpress = cronExpress;
    }

    @Override
    public Date nextExecutionTime(TriggerContext triggerContext) {
        CronTrigger cronTrigger = new CronTrigger(cronExpress);
        return cronTrigger.nextExecutionTime(triggerContext);
    }
}

测试接口

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.scheduling.concurrent.ThreadPoolTaskScheduler;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;

import java.time.LocalDateTime;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.ScheduledFuture;

/**
 * @desc: 模拟测试手动任务
 * <p>
 * by  zhang.Z
 */
@RestController
public class TPTSController {


    @Autowired
    private ThreadPoolTaskScheduler taskScheduler;

    /**
     * 定义执行任务的缓存池
     */
    private static Map<String, ScheduledFuture<?>> sm = new HashMap<>();

    /**
     * 增加任务
     */
    @RequestMapping(value = "/task/add")
    public void addTask(@RequestParam(value = "taskName")String taskName){
        ScheduledFuture<?> schedule = taskScheduler.schedule(new Task(), new TriggerImpl("0/5 * * * * ?"));
        sm.put(taskName,schedule);
    }


    /**
     * 删除任务
     * @param taskName
     */
    @RequestMapping(value = "/task/remove")
    public void removeTask(@RequestParam(value = "taskName")String taskName){
        ScheduledFuture<?> future = sm.get(taskName);
        //判断任务是否为空或者已经结束
        if (null != future && !future.isCancelled()){
            future.cancel(true);
            while (!future.isDone()) {
                future.cancel(true);
            }
        }
        sm.remove(taskName);
    }

    /**
     * 更新任务
     * @param taskName
     */
    @RequestMapping(value = "/task/update")
    public void updateTask(@RequestParam(value = "taskName")String taskName){
        removeTask(taskName);
        addTask(taskName);
    }


    /**
     * 测试任务
     */
    class Task  implements Runnable{
        @Override
        public void run() {
            String name = Thread.currentThread().getName();
            System.out.println(name+"----"+ LocalDateTime.now());
        }
    }

}