文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

springboot为异步任务规划自定义线程池如何实现

2023-07-02 08:41

关注

本篇内容主要讲解“springboot为异步任务规划自定义线程池如何实现”,感兴趣的朋友不妨来看看。本文介绍的方法操作简单快捷,实用性强。下面就让小编来带大家学习“springboot为异步任务规划自定义线程池如何实现”吧!

一、Spring Boot任务线程池

线程池的作用

在高并发环境下,不断的分配新资源,可能导致系统资源耗尽。所以为了避免这个问题,我们为异步任务规划一个线程池。当然,如果没有配置线程池的话,springboot会自动配置一个ThreadPoolTaskExecutor 线程池到bean当中。

# 核心线程数spring.task.execution.pool.core-size=8  # 最大线程数spring.task.execution.pool.max-size=16# 空闲线程存活时间spring.task.execution.pool.keep-alive=60s# 是否允许核心线程超时spring.task.execution.pool.allow-core-thread-timeout=true# 线程队列数量spring.task.execution.pool.queue-capacity=100# 线程关闭等待spring.task.execution.shutdown.await-termination=falsespring.task.execution.shutdown.await-termination-period=# 线程名称前缀spring.task.execution.thread-name-prefix=task-

在springboot配置文件中加入上面的配置,即可实现ThreadPoolTaskExecutor 线程池。

二、自定义线程池

有的时候,我们希望将系统内的一类任务放到一个线程池,另一类任务放到另外一个线程池,所以使用Spring Boot自带的任务线程池就捉襟见肘了。下面介绍自定义线程池的方法。

创建一个 线程池配置类 TaskConfiguration ,并配置一个 任务线程池对象 taskExecutor。

@Configurationpublic class TaskConfiguration {    @Bean("taskExecutor")    public Executor taskExecutor() {        ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();        executor.setCorePoolSize(10);        executor.setMaxPoolSize(20);        executor.setQueueCapacity(200);        executor.setKeepAliveSeconds(60);        executor.setThreadNamePrefix("taskExecutor-");        executor.setRejectedExecutionHandler(new CallerRunsPolicy());        return executor;    }}

springboot为异步任务规划自定义线程池如何实现

上面我们通过使用 ThreadPoolTaskExecutor 创建了一个 线程池,同时设置了以下这些参数:

线程池属性属性的作用上文代码设置初始值
核心线程数CorePoolSize线程池创建时候初始化的线程数,最小线程数10
最大线程数MaxPoolSize线程池最大的线程数,只有在缓冲队列满了之后,才会申请超过核心线程数的线程20
缓冲任务队列QueueCapacity用来缓冲执行任务的队列200
允许线程的空闲时间KeepAliveSeconds超过了核心线程之外的线程,在空闲时间到达之后,没活干的线程会被销毁60秒
线程池名的前缀 ThreadNamePrefix可以用于定位处理任务所在的线程池taskExecutor-
线程池对任务的Reject策略RejectedExecutionHandler当线程池运行饱和,或者线程池处于shutdown临界状态时,用来拒绝一个任务的执行CallerRunsPolicy

Reject策略预定义有四种:

创建 AsyncExecutorTask类,三个任务的配置和 AsyncTask 一样,不同的是 @Async 注解需要指定前面配置的 线程池的名称 taskExecutor。

@Componentpublic class AsyncExecutorTask extends AbstractTask {    @Async("taskExecutor")    public Future<String> doTaskOneCallback() throws Exception {        super.doTaskOne();        System.out.println("任务一,当前线程:" + Thread.currentThread().getName());        return new AsyncResult<>("任务一完成");    }    @Async("taskExecutor")    public Future<String> doTaskTwoCallback() throws Exception {        super.doTaskTwo();        System.out.println("任务二,当前线程:" + Thread.currentThread().getName());        return new AsyncResult<>("任务二完成");    }    @Async("taskExecutor")    public Future<String> doTaskThreeCallback() throws Exception {        super.doTaskThree();        System.out.println("任务三,当前线程:" + Thread.currentThread().getName());        return new AsyncResult<>("任务三完成");    }}

在 单元测试 用例中,注入 AsyncExecutorTask 对象,并在测试用例中执行 doTaskOne(),doTaskTwo(),doTaskThree() 三个方法。

@SpringBootTestpublic class AsyncExecutorTaskTest {    @Autowired    private AsyncExecutorTask task;    @Test    public void testAsyncExecutorTask() throws Exception {        task.doTaskOneCallback();        task.doTaskTwoCallback();        task.doTaskThreeCallback();        sleep(30 * 1000L);    }}

执行一下上述的 单元测试,可以看到如下结果:

开始做任务一
开始做任务三
开始做任务二
完成任务二,耗时:3905毫秒
任务二,当前线程:taskExecutor-2
完成任务一,耗时:6184毫秒
任务一,当前线程:taskExecutor-1
完成任务三,耗时:9737毫秒
任务三,当前线程:taskExecutor-3

执行上面的单元测试,观察到 任务线程池 的 线程池名的前缀 被打印,说明 线程池 成功执行 异步任务!

三、优雅地关闭线程池

由于在应用关闭的时候异步任务还在执行,导致类似 数据库连接池 这样的对象一并被 销毁了,当 异步任务 中对 数据库 进行操作就会出错。

解决方案如下,重新设置线程池配置对象,新增线程池 setWaitForTasksToCompleteOnShutdown() 和 setAwaitTerminationSeconds() 配置:

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

异步任务** 的 销毁 就会先于 数据库连接池对象 的销毁。

到此,相信大家对“springboot为异步任务规划自定义线程池如何实现”有了更深的了解,不妨来实际操作一番吧!这里是编程网网站,更多相关内容可以进入相关频道进行查询,关注我们,继续学习!

阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯