如何使用ThreadPoolTaskScheduler处理离线并发任务？

傷城~ 2023-10-02 23:36 42阅读 0赞

你有没有遇到过离线并发任务场景？

你是怎么处理的呢？

今天跟着博住来学习下吧（高手请自动飘过哈），下面进入主题：

目标：确保同一时刻有10个并发离线任务。

分析：要实现这个功能其实实现方式很多，比如，

1)在一个主线成里手动创建10个子线程，并自己维护这些线程状态；

2)使用ThreadPoolTaskExecutor线程池，并在主线程中实时将任务丢到线程池执行队列；

3)使用ThreadPoolTaskScheduler，本文也主要使用该类事列，请看下文代码：

package com.test
    
    import org.springframework.context.annotation.Configuration;
    import org.springframework.boot.task.ThreadSchedulerBuilder;
    import org.springframework.scheduling.concurrent.ThreadPoolTaskScheduler;
    /** 
    * 本示例是基于springboot框架实现，使用时请注意防止与默认的定时任务管理器冲突。
    */
    @Configuration
    public class Test{
    
        
        @Bean(name="task")
        public ThreadPoolTaskScheduler taskScheduler(){
            //并发任务数量
            int taskNum= 10;
            ThreadPoolTaskScheduler scheduler = new ThreadSchedulerBuilder().poolsize(taskNum)
                .threadNamePrefix("task-").build();
            scheduler.setDaemon(true);
            // 提前初始化后才能添加任务
            scheduler.initialize();
            long now = System.currentTimeMillis();
            // 添加10个并行任务（runnable）
            for(int i=0;i<taskNum;i++){
                // 使用固定延迟执行，即上一个任务执行完成后等待多长时间执行，这里设置了1秒钟
                scheduler.scheduleWithFixedelay(this::task,1000);
                // 固定延迟时间可以尽量短点，以便能更加实时获取任务，这里也可以设置一个100毫秒的开始时间梯度,错开调度，避免统一时间消耗大量系统资源。
                // scheduler.scheduleWithFixedelay(this::task,new Date(now+i*100),1000);
            }
            return scheduler;
        }
    
        public void task(){
            // 使用场景，针对同一个业务存在大量的离线分析任务，每个任务之间的数据处理不冲突。
            System.out.println("执行任务逻辑");        
        }
    
    }

你学会了吗？赶紧动手试试吧。

《完》

PS：文中代码为纯手敲，如有错误之处，敬请指正。