39 lines
902 B
Java
39 lines
902 B
Java
package org.warp.commonutils.concurrency.executor;
|
|
|
|
import java.util.concurrent.Semaphore;
|
|
import javax.annotation.Nonnull;
|
|
|
|
public final class PermitReleasingRunnableDecorator extends RunnableDecorator {
|
|
|
|
@Nonnull
|
|
private final Runnable queueSizeUpdater;
|
|
|
|
@Nonnull
|
|
private final Semaphore semaphore;
|
|
|
|
PermitReleasingRunnableDecorator(@Nonnull final Runnable task,
|
|
@Nonnull final Runnable queueSizeUpdater,
|
|
@Nonnull final Semaphore semaphoreToRelease) {
|
|
super(task);
|
|
this.queueSizeUpdater = queueSizeUpdater;
|
|
this.semaphore = semaphoreToRelease;
|
|
}
|
|
|
|
@Override
|
|
public void run() {
|
|
try {
|
|
queueSizeUpdater.run();
|
|
} finally {
|
|
// however execution goes, release permit for next task
|
|
this.semaphore.release();
|
|
|
|
super.run();
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public final String toString() {
|
|
return String.format("%s[delegate='%s']", getClass().getSimpleName(), super.toString());
|
|
}
|
|
}
|