ZhouXY108/hutool
1
0
Fork 0
mirror of https://gitee.com/chinabugotech/hutool.git synced 2025-05-09 23:51:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

修复线程池关闭时，阻塞策略仍然可以将任务添加到线程池

Browse Source
This commit is contained in:
witt 2022-06-17 11:55:32 +08:00
parent 231e0cf1d7
commit 4670dccb67
1 changed files with 25 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
hutool-core/src/main/java/cn/hutool/core/thread/BlockPolicy.java
View File

@ -3,6 +3,7 @@ package cn.hutool.core.thread;
import java.util.concurrent.RejectedExecutionException;
import java.util.concurrent.RejectedExecutionHandler;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.function.Consumer;
/**
* 当任务队列过长时处于阻塞状态，直到添加到队列中
@ -14,15 +15,35 @@ import java.util.concurrent.ThreadPoolExecutor;
*/
public class BlockPolicy implements RejectedExecutionHandler {
/**
* 线程池关闭时，为避免任务丢失，留下处理方法
* 如果需要由调用方来运行，可以{@code new BlockPolicy(Runnable::run)}
*/
private final Consumer<Runnable> handlerwhenshutdown;
public BlockPolicy(final Consumer<Runnable> handlerwhenshutdown) {
this.handlerwhenshutdown = handlerwhenshutdown;
}
public BlockPolicy() {
this(null);
}
@Override
public void rejectedExecution(Runnable r, ThreadPoolExecutor e) {
try {
e.getQueue().put(r);
} catch (InterruptedException ex) {
throw new RejectedExecutionException("Task " + r + " rejected from " + e);
// 线程池未关闭时，阻塞等待
if(!e.isShutdown()){
try {
e.getQueue().put(r);
} catch (InterruptedException ex) {
throw new RejectedExecutionException("Task " + r + " rejected from " + e);
}
return;
}
// 当设置了关闭时候的处理
if(null != handlerwhenshutdown){
handlerwhenshutdown.accept(r);
}
}
}