自定义线程池实现

迈不过友情╰ 2022-12-24 00:35 239阅读 0赞

# 线程池 - 自定义线程池实现 #

用线程池来管理线程，减少系统消耗，线程数可控。

--------------------

**目录**

线程池 - 自定义线程池实现

一、自定义线程池代码

二、测试

--------------------

## 一、自定义线程池代码  ##

自定义使用 ThreadPoolTaskExecutor

package com.yuantiaokj.commonmodule.threadpool;
    
    import org.springframework.beans.factory.annotation.Autowired;
    import org.springframework.context.annotation.Bean;
    import org.springframework.context.annotation.Configuration;
    import org.springframework.scheduling.annotation.EnableAsync;
    import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;
    
    import java.util.concurrent.Executor;
    import java.util.concurrent.ThreadPoolExecutor;
    
    /**
     * ***********************************************************
     * Copyright © 2019  Inc.All rights reserved.  *    *
     * **********************************************************
     *
     * @program: pay_inspector
     * @name: TaskExecutePool
     * @author: Mr.Cnzz
     * @create: 2019-11-28 15:25
     * @description: 创建线程池
     **/
    
    
    @Configuration
    @EnableAsync
    public class TaskExecutePool {
    
        /**
         * 1.这种形式的线程池配置是需要在使用的方法上面@Async("taskExecutor"),
         * 2.如果在使用的方法上面不加该注解那么spring就会使用默认的线程池
         * 3.所以如果加@Async注解但是不指定使用的线程池，又想自己定义线程池那么就可以重写spring默认的线程池
         * 4.所以第二个方法就是重写默认线程池
         * 注意：完全可以把线程池的参数写到配置文件中
         */
        @Autowired
        private TaskThreadPoolConfig config;
    
        @Bean("taskExecutePoolCnzz")
        public Executor taskExecutePoolCnzz() {
            ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
            //核心线程池大小
            executor.setCorePoolSize(config.getCorePoolSize());
            //最大线程数
            executor.setMaxPoolSize(config.getMaxPoolSize());
            //队列容量
            executor.setQueueCapacity(config.getQueueCapacity());
            //活跃时间
            executor.setKeepAliveSeconds(config.getKeepAliveSeconds());
            //线程名字前缀
            executor.setThreadNamePrefix("taskExecutePoolCnzz");
    
            // setRejectedExecutionHandler：当pool已经达到max size的时候，如何处理新任务
            // CallerRunsPolicy：不在新线程中执行任务，而是由调用者所在的线程来执行
            executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
            // 等待所有任务结束后再关闭线程池
            executor.setWaitForTasksToCompleteOnShutdown(true);
            executor.initialize();
    
            return executor;
        }
    
    }

配置类

package com.yuantiaokj.commonmodule.threadpool;
    
    import lombok.Data;
    import org.springframework.boot.context.properties.ConfigurationProperties;
    
    /**
     * ***********************************************************
     * Copyright © 2019  Inc.All rights reserved.  *    *
     * **********************************************************
     *
     * @program: pay_inspector
     * @name: TaskThreadPoolConfig
     * @author: Mr.Cnzz
     * @create: 2019-11-28 15:21
     * @description: 线程池配置类
     **/
    @ConfigurationProperties(prefix = "task.pool")
    @Data
    @Component
    public class TaskThreadPoolConfig {
        
        //核心线程
        private int corePoolSize;
        //最大线程
        private int maxPoolSize;
        //空闲时间
        private int keepAliveSeconds;
        //队列容量
        private int queueCapacity;
    }

yml 配置线程池参数

#线程池配置参数
    task:
      pool:
        corePoolSize: 20 #设置核心线程数
        maxPoolSize: 100  #设置最大线程数
        keepAliveSeconds: 60 #设置线程活跃时间（秒）
        queueCapacity: 50 #设置队列容量

启动类Application 添加异步注解

//开启线程异步支持
    @EnableAsync
    // 开启配置属性支持
    @EnableConfigurationProperties({TaskThreadPoolConfig.class})

使用，springboot直接使用@Async异步执行

--------------------

## 二、测试 ##

@Async
        public void printNum(String str) {
            log.info("str={}", str);
        }

@Resource
        AsyncTest asyncTest;
    
        @Test
        public void testAsync(){
            for (int i = 0; i <20 ; i++) {
                asyncTest.printNum(String.valueOf(i));
            }
    
        }

这里注意调用方法要在不同的类才会生效，由于容器加载原因，在同一个类时需要容器重新加载下。

1、队列，最大线程数 都较小的情况下，会生成到10个线程，触发饱和策略，也会有主线程执行。

测试配置，实际按环境情况配置，这里为了看到测试效果

#线程池配置参数
    task:
      pool:
        corePoolSize: 2 #设置核心线程数
        maxPoolSize: 10  #设置最大线程数
        keepAliveSeconds: 10 #设置线程活跃时间（秒）
        queueCapacity: 5 #设置队列容量

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3hpbnB6_size_16_color_FFFFFF_t_70][]

2、增大队列容量，让在队列等待，只有2个主线程在跑

#线程池配置参数
    task:
      pool:
        corePoolSize: 2 #设置核心线程数
        maxPoolSize: 10  #设置最大线程数
        keepAliveSeconds: 10 #设置线程活跃时间（秒）
        queueCapacity: 20 #设置队列容量

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3hpbnB6_size_16_color_FFFFFF_t_70 1][]

3、稍微减少队列，看下当队列满了，及未达到线程总数，未饱和的情况

#线程池配置参数
    task:
      pool:
        corePoolSize: 2 #设置核心线程数
        maxPoolSize: 20  #设置最大线程数
        keepAliveSeconds: 10 #设置线程活跃时间（秒）
        queueCapacity: 15 #设置队列容量

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3hpbnB6_size_16_color_FFFFFF_t_70 2][]

具体参数配置及原理可参考   [线程池-线程池参数及配置][-]  [https://blog.csdn.net/xinpz/article/details/110132365][-]

--------------------

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3hpbnB6_size_16_color_FFFFFF_t_70]: /images/20221120/4401879151c94a0b89003583b4d4202e.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3hpbnB6_size_16_color_FFFFFF_t_70 1]: /images/20221120/5ac7e330e33042aa87255b8a2ea92c98.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3hpbnB6_size_16_color_FFFFFF_t_70 2]: /images/20221120/463f953f06ff491089f499cfca259d0c.png
[-]: https://blog.csdn.net/xinpz/article/details/110132365