mobeigi
/
filebot
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Refactor FloodLimit

This commit is contained in:
Reinhard Pointner 2016-04-20 06:41:07 +00:00
parent 03aa028815
commit 7321da5817
1 changed files with 6 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
source/net/filebot/web/FloodLimit.java
View File

@ -1,3 +1,4 @@
package net.filebot.web; package net.filebot.web;
import java.util.concurrent.ScheduledFuture; import java.util.concurrent.ScheduledFuture;
@ -5,10 +6,13 @@ import java.util.concurrent.ScheduledThreadPoolExecutor;
import java.util.concurrent.Semaphore; import java.util.concurrent.Semaphore;
import java.util.concurrent.TimeUnit; import java.util.concurrent.TimeUnit;
import net.filebot.util.DefaultThreadFactory;
public class FloodLimit { public class FloodLimit {
private static final ScheduledThreadPoolExecutor TIMER = new ScheduledThreadPoolExecutor(1, new DefaultThreadFactory("FloodLimitTimer", Thread.NORM_PRIORITY, true));
private final Semaphore permits; private final Semaphore permits;
private final ScheduledThreadPoolExecutor timer = new ScheduledThreadPoolExecutor(1);
private final long releaseDelay; private final long releaseDelay;
private final TimeUnit timeUnit; private final TimeUnit timeUnit;
@ -21,20 +25,7 @@ public class FloodLimit {
public ScheduledFuture<?> acquirePermit() throws InterruptedException { public ScheduledFuture<?> acquirePermit() throws InterruptedException {
permits.acquire(); permits.acquire();
return timer.schedule(new ReleasePermit(), releaseDelay, timeUnit); return TIMER.schedule(permits::release, releaseDelay, timeUnit);
} }
public void releaseNow(ScheduledFuture<?> future) {
if (future.cancel(false)) {
permits.release();
}
}
private class ReleasePermit implements Runnable {
@Override
public void run() {
permits.release();
}
}
} }