ThreadPoolTaskScheduler轻量级多线程定时任务框架

ThreadPoolTaskScheduler轻量级多线程定时任务框架

前言

面对一些小的功能需求点，如果需要非常灵活的进行处理定时任务处理，但是又因为你的需求需要使用到页面进行定时任务配置，显然使用Spring注解定时任务，无法满足你，这时你想到了xxl-job 或者 quezy等定时任务框架，但是过于繁琐，可能成本较大。那么本文将要解决你的问题。

一、ThreadPoolTaskScheduler是什么？

• springboot中有一个bean，ThreadPoolTaskScheduler，可以很方便的对重复执行的任务进行调度管理；相比于通过java自带的周期性任务线程池
• ScheduleThreadPoolExecutor，此bean对象支持根据cron表达式创建周期性任务。
• 当然，ThreadPoolTaskScheduler其实底层使用也是java自带的线程池。

二、上干货

1.ThreadPoolTaskScheduler常用的api介绍

ThreadPoolTaskScheduler 内部方法非常丰富，本文实现的是一种corn表达式，周期执行

• schedule(Runnable task, Trigger trigger) corn表达式，周期执行
• schedule(Runnable task, Date startTime) 定时执行
• scheduleAtFixedRate(Runnable task, Date startTime, long period) 定时周期间隔时间执行。间隔时间单位 TimeUnit.MILLISECONDS
• scheduleAtFixedRate(Runnable task, long period) 间隔时间执行。单位毫秒

2.springboot中创建配置类

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

@Configuration
public class TaskConfig {//配置定时任务线程池-自定义名称避免冲突
    @Bean(name ="myThreadPoolTaskScheduler")
    public ThreadPoolTaskScheduler threadPoolTaskScheduler(){
        ThreadPoolTaskScheduler executor = new ThreadPoolTaskScheduler();
        executor.setPoolSize(2);
        executor.setThreadNamePrefix("task-");
        executor.setWaitForTasksToCompleteOnShutdown(true);
        executor.setAwaitTerminationSeconds(60);return executor;}}

3.实现Runnable的业务处理类

代码如下（示例）：

@Slf4j
public class RunSameThing implements Runnable{

    @Override
    public voidrun(){
        log.info("===我执行了===");}}

4.业务处理服务service

import java.util.concurrent.ScheduledFuture;
import javax.annotation.Resource;
import lombok.extern.slf4j.Slf4j;
import org.springframework.scheduling.concurrent.ThreadPoolTaskScheduler;
import org.springframework.scheduling.support.CronTrigger;
import org.springframework.stereotype.Service;

@Service
@Slf4j
public class CommonTask {

    @Resource(name ="myThreadPoolTaskScheduler")
    private ThreadPoolTaskScheduler taskScheduler;

    private ScheduledFuture future;//开启任务
    public voidstartTask(){//每次调用前,可执行一次关闭之前的stop();//每秒执行-corn 表达式可传参形式
        String cron ="0/1 * * * * ?";//RunSameThing 为执行的业务逻辑
        future = taskScheduler.schedule(new RunSameThing(), new CronTrigger(cron));}//关闭任务
    public voidstop(){if(future != null){
            log.info("我关闭了");
            future.cancel(true);}}}

5.调用controller

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

@RestController
@RequestMapping("/task")
public class TestTaskController {

    @Autowired
    private CommonTask commonTask;

    @RequestMapping("start")
    public voidstartTask(){
        commonTask.startTask();}

    @RequestMapping("end")
    public voidendTask(){
        commonTask.stop();}}

6.让我们看看运行结果

task-1] com.easy.wheel.scheduler.RunSameThing    :===我执行了===
task-2] com.easy.wheel.scheduler.RunSameThing    :===我执行了===
task-2] com.easy.wheel.scheduler.RunSameThing    :===我执行了===
task-1] com.easy.wheel.scheduler.RunSameThing    :===我执行了===
task-1] com.easy.wheel.scheduler.RunSameThing    :===我执行了===
task-2] com.easy.wheel.scheduler.RunSameThing    :===我执行了===
task-1] com.easy.wheel.scheduler.RunSameThing    :===我执行了===
task-1] com.easy.wheel.scheduler.RunSameThing    :===我执行了===
task-1] com.easy.wheel.scheduler.RunSameThing    :===我执行了===
task-2] com.easy.wheel.scheduler.RunSameThing    :===我执行了===[nio-8080-exec-2] com.easy.wheel.scheduler.CommonTask      : 我关闭了

线程池配置的两个线程，可以看到task-1 和 task-2 交替支持，达到了我们想要的效果，关闭也能正常进行关闭，nice。这个轮子搞定了，下次写代码直接复制。巴适

总结

生活不能止步不前，为了更多的时间陪陪家人，轮子用起来啊，亲