tdlight-team/common-utils
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Update BoundedExecutorServiceImpl.java

Browse Source
This commit is contained in:
Andrea Cavalli 2020-06-24 12:12:31 +02:00
parent fd426938f5
commit 46dcf99277
1 changed files with 16 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
src/main/java/org/warp/commonutils/concurrency/executor/BoundedExecutorServiceImpl.java
View File

@ -1,9 +1,14 @@
package org.warp.commonutils.concurrency.executor;
import org.jetbrains.annotations.Nullable;
import java.util.concurrent.*;
import java.util.concurrent.Callable;
import java.util.concurrent.Future;
import java.util.concurrent.LinkedBlockingQueue;
import java.util.concurrent.Semaphore;
import java.util.concurrent.ThreadFactory;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.TimeUnit;
import java.util.function.BiConsumer;
import org.jetbrains.annotations.Nullable;
class BoundedExecutorServiceImpl extends ThreadPoolExecutor implements BoundedExecutorService {
@ -43,8 +48,7 @@ class BoundedExecutorServiceImpl extends ThreadPoolExecutor implements BoundedEx
*/
@Override
public <T> Future<T> submitButBlockIfFull(final Callable<T> task) throws InterruptedException {
blockIfFull();
return submit(task);
return super.submit(task);
}
/**
@ -53,8 +57,7 @@ class BoundedExecutorServiceImpl extends ThreadPoolExecutor implements BoundedEx
*/
@Override
public void executeButBlockIfFull(final Runnable task) throws InterruptedException {
blockIfFull();
execute(task);
super.execute(task);
}
private void blockIfFull() throws InterruptedException {
@ -68,6 +71,12 @@ class BoundedExecutorServiceImpl extends ThreadPoolExecutor implements BoundedEx
@Override
public void beforeExecute(Thread t, Runnable r) {
try {
blockIfFull();
} catch (InterruptedException e) {
throw new RuntimeException(e);
}
var queueSize = getQueue().size();
synchronized (queueSizeStatusLock) {
if (queueSizeStatus != null) queueSizeStatus.accept(queueSize >= maxQueueSize, queueSize);