1. import java.util.concurrent.ConcurrentHashMap;
  2. import java.util.concurrent.atomic.AtomicInteger;
  4. public class RequestThrottler {
  6. private final ConcurrentHashMap<String, AtomicInteger> requestCounts = new ConcurrentHashMap<>(); // Entry ID to request count
  7. private final long throttleWindowMillis; // Time window for throttling
  8. private final long maxRequestsPerWindow; // Maximum requests allowed in the window
  11. public RequestThrottler(long throttleWindowMillis, long maxRequestsPerWindow) {
  12. this.throttleWindowMillis = throttleWindowMillis;
  13. this.maxRequestsPerWindow = maxRequestsPerWindow;
  14. }
  16. public boolean allowRequest(String entryId) {
  17. // Get the current count for the entry ID
  18. AtomicInteger count = requestCounts.computeIfAbsent(entryId, k -> new AtomicInteger(0));
  20. long now = System.currentTimeMillis();
  22. // Reset count if outside the throttle window
  23. if (now - count.getAndSet(0, 0) > throttleWindowMillis) {
  24. count.set(0, 0); // Reset count
  25. }
  27. // Check if the request is within the limit
  28. if (count.get() < maxRequestsPerWindow) {
  29. count.incrementAndGet(); // Increment the request count
  30. return true;
  31. } else {
  32. return false; // Throttle - request denied
  33. }
  34. }
  36. public void removeEntry(String entryId) {
  37. requestCounts.remove(entryId);
  38. }
  40. public static void main(String[] args) throws InterruptedException {
  41. RequestThrottler throttler = new RequestThrottler(1000, 5); // Throttle to 5 requests per second
  43. for (int i = 0; i < 10; i++) {
  44. if (throttler.allowRequest("entry1")) {
  45. System.out.println("Request allowed for entry1");
  46. } else {
  47. System.out.println("Request throttled for entry1");
  48. }
  49. Thread.sleep(100); // Simulate request delay
  50. }
  52. throttler.removeEntry("entry1");
  55. for (int i = 0; i < 5; i++) {
  56. if (throttler.allowRequest("entry1")) {
  57. System.out.println("Request allowed for entry1");
  58. } else {
  59. System.out.println("Request throttled for entry1");
  60. }
  61. Thread.sleep(200); // Simulate request delay
  62. }
  63. }
  64. }

Add your comment