@Async 配合线程池的使用

小鱼儿 2023-07-01 11:54 76阅读 0赞

## 一、编写线程池配置类 ##

@Configuration
    @EnableAsync
    public class NlpThreadPoolConfig { 
    
        private static final int CORE_POOL_SIZE = 10;
    
        private static final int MAX_POOL_SIZE = 20;
    
        private static final int QUEUE_CAPACITY = 200;
    
        public static final String SENTIMENT_EVENT_EXECUTOR = "test_executor";
    
        /** * 事件和情感接口线程池执行器配置 * @return 事件和情感接口线程池执行器bean * */
        @Bean(SENTIMENT_EVENT_EXECUTOR)
        public Executor executor() { 
            ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
            executor.setCorePoolSize(CORE_POOL_SIZE);
            executor.setMaxPoolSize(MAX_POOL_SIZE);
            // 设置队列容量
            executor.setQueueCapacity(QUEUE_CAPACITY);
            // 设置线程活跃时间(秒)
            executor.setKeepAliveSeconds(60);
            executor.setThreadNamePrefix("SE-Pool#Task");
            // 设置拒绝策略
            executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
            executor.initialize();
            return executor;
        }
    }

说一下拒绝策略，生效的情况：`当Executor已经关闭（即执行了executorService.shutdown()方法后），并且Executor将有限边界用于最大线程和工作队列容量，且已经饱和时，在方法execute()中提交的新任务将被拒绝.`  
在以上述情况下，execute 方法将调用其 RejectedExecutionHandler 的RejectedExecutionHandler.rejectedExecution(java.lang.Runnable, java.util.concurrent.ThreadPoolExecutor) 方法。一共有4个取值：

*  ThreadPoolExecutor.AbortPolicy:丢弃任务并抛出RejectedExecutionException异常。
 *  ThreadPoolExecutor.DiscardPolicy：也是丢弃任务，但是不抛出异常。
 *  ThreadPoolExecutor.DiscardOldestPolicy：丢弃队列最前面的任务，然后重新尝试执行任务（重复此过程）
 *  ThreadPoolExecutor.CallerRunsPolicy：由调用线程处理该任务

同时注意需要在配置类上添加`@EnableAsync`，当然也可以在启动类上添加，表示开启spring的@`@Async`

> 记录一下线程池参数的生效流程：  
> 1、当提交一个新任务到线程池时首先线程池判断基本线程池(corePoolSize)是否已满？没满，创建一个工作线程来执行任务。满了，则进入下个流程；  
> 2、其次线程池判断工作队列(workQueue)是否已满？没满，则将新提交的任务存储在工作队列里。满了，则进入下个流程；  
> 3、最后线程池判断整个线程池(maximumPoolSize)是否已满？没满，则创建一个新的工作线程来执行任务，满了，则交给饱和策略来处理这个任务；如果线程池中的线程数量大于 corePoolSize 时，如果某线程空闲时间超过keepAliveTime，线程将被终止，直至线程池中的线程数目不大于corePoolSize；如果允许为核心池中的线程设置存活时间，那么核心池中的线程空闲时间超过 keepAliveTime，线程也会被终止。

三、ExecutorService与ThreadPoolTaskExecutor对比  
自己开发的过程中用了ThreadPoolTaskExecutor，ThreadPoolTaskExecutor是spring core包中的，而ThreadPoolExecutor是JDK中的JUC。ThreadPoolTaskExecutor是对ThreadPoolExecutor进行了封装处理。  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzIwOTg5MTA1_size_16_color_FFFFFF_t_70]

## 二、添加方法 ##

需要在多线程处理的方法上添加@Async注解  
例如：

@Async(NlpThreadPoolConfig.SENTIMENT_EVENT_EXECUTOR)
    public CompletableFuture<List<EmotionClassity>> analysisByBert(ArticleModel articleModel)

注解中可以指定自己创建的线程池，`NlpThreadPoolConfig.SENTIMENT_EVENT_EXECUTOR`

注意：当这个方法被当前类的另一个方法执行的时候，会导致只是执行单线程，需要在另外一个类调用这个类的这个异步方法的时候才会是多线程的。  
具体原因：  
因为这个方法循环调用的，当在其他类中调用这个方法，每一次的循环都会创建这个对象，所以，每一次的创建和调用都是在一个独立的线程中进行的；如果是在当前对象中调用这个方法，那么在调用这个方法之前这个对象已经被创建了，所以结果就这样了。  
当然，这个是一个很低级的错误，多线程的基础，但是有的时候就容易踩到这个坑。

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzIwOTg5MTA1_size_16_color_FFFFFF_t_70]: https://img-blog.csdnimg.cn/2020021310485423.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzIwOTg5MTA1,size_16,color_FFFFFF,t_70