Java阻塞线程太多_Java中如何让线程池阻塞过多任务的提交

Java中如何让线程池阻塞过载任务的提交

Java的线程池一般是基于concurrent包下的ThreadPoolExecutor类实现的，不过当我们基于spring框架开发程序时，通常会使用其包装类ThreadPoolTaskExecutor。

不管使用JDK还是Spring的线程池，都有一个小问题，就是当使用线程池执行任务的时候，任务的消费速度小于生产速度时，任务通常会被暂存到阻塞队列。阻塞队列大小通常是固定的，然而阻塞队列满的时候，execute方法并不会被阻塞，客户端线程仍然可以不断的提交任务。

只是pool发现实在无力处理时，启用拒绝机制来应对不断到来的任务，默认是使用RejectedExecutionHandler去处理拒绝的任务，默认策略是AbortPolicy，直接抛出RejectedExecutionException异常。

而很多情况下，这不符合我们的业务需求，我们希望被拒绝的任务能够阻塞执行，从而阻止任务的生产速度；

一个比较巧妙的解决方案如下，仅供参考，具体还需要根据实际场景来应用：

import java.util.concurrent.Executor;

import java.util.concurrent.ThreadPoolExecutor;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.springframework.context.annotation.Bean;

import org.springframework.context.annotation.Configuration;

import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;

@Slf4j

@Configuration

public class BatchProcessPoolConfig {

@Bean("batchProcessPool")

public Executor threadPoolTaskExecutor() {

ThreadPoolTaskExecutor taskExecutor = new ThreadPoolTaskExecutor();

taskExecutor.setCorePoolSize(6);

taskExecutor.setMaxPoolSize(50);

taskExecutor.setQueueCapacity(1000);

taskExecutor.setRejectedExecutionHandler((r, executor) -> {

if (!executor.isShutdown()) {

try {

executor.getQueue().put(r);

} catch (InterruptedException e) {

log.error(e.toString(), e);

Thread.currentThread().interrupt();

}

}

}

);

taskExecutor.initialize();

return taskExecutor;

}

}

这里之所以能实现阻塞，是基于BlockingQueue的put方法来实现的，当阻塞队列满时，put方法会一直等待。

注意：鉴于线程池是先填满队列再考虑补充到maxSize个线程，此时的maxSize最好与coreSize相等。