Skip to content

Instantly share code, notes, and snippets.

@mjunaidi
Forked from Glamdring/SimpleRateLimiter
Created August 27, 2018 06:06
Show Gist options
  • Save mjunaidi/7b466f81a79a3c79db70a87d43f00a17 to your computer and use it in GitHub Desktop.
Save mjunaidi/7b466f81a79a3c79db70a87d43f00a17 to your computer and use it in GitHub Desktop.
public class SimpleRateLimiter {
private Semaphore semaphore;
private int maxPermits;
private TimeUnit timePeriod;
private ScheduledExecutorService scheduler;
public static SimpleRateLimiter create(int permits, TimeUnit timePeriod) {
SimpleRateLimiter limiter = new SimpleRateLimiter(permits, timePeriod);
limiter.schedulePermitReplenishment();
return limiter;
}
private SimpleRateLimiter(int permits, TimeUnit timePeriod) {
this.semaphore = new Semaphore(permits);
this.maxPermits = permits;
this.timePeriod = timePeriod;
}
public boolean tryAcquire() {
return semaphore.tryAcquire();
}
public void stop() {
scheduler.shutdownNow();
}
public void schedulePermitReplenishment() {
scheduler = Executors.newScheduledThreadPool(1);
scheduler.schedule(() -> {
semaphore.release(maxPermits - semaphore.availablePermits());
}, 1, timePeriod);
}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment