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

冷不防 2024-03-17 15:32 52阅读 0赞

## 前言 ##

面对一些小的功能需求点，如果需要非常灵活的进行处理定时任务处理，但是又因为你的需求需要使用到页面进行定时任务配置，显然使用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 void run() {
            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 void startTask(){
            //每次调用前,可执行一次关闭之前的
            stop();
            //每秒执行-corn 表达式可传参形式
            String cron = "0/1 * * * * ?";
            //RunSameThing 为执行的业务逻辑
            future = taskScheduler.schedule(new RunSameThing(), new CronTrigger(cron));
        }
        //关闭任务
        public void stop() {
            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 void startTask(){
            commonTask.startTask();
        }
    
        @RequestMapping("end")
        public void endTask(){
            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      : 我关闭了

————————————————

版权声明：本文为CSDN博主「ice&wind」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。  
原文链接：https://blog.csdn.net/weixin\_43480441/article/details/127869051