lsm/hutool
1
0
Fork 0
mirror of https://gitee.com/dromara/hutool.git synced 2025-04-05 17:37:59 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

!660 线程池阻塞策略完善

Browse Source 
Merge pull request !660 from witt/v5-dev
This commit is contained in:
Looly 2022-06-20 04:08:07 +00:00 committed by Gitee
commit 9c43ee5f7d
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
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);
}
}
}