1. import java.util.concurrent.ConcurrentLinkedQueue;
  2. import java.util.concurrent.atomic.AtomicInteger;
  4. public class RequestThrottler {
  6. private final int maxRequestsPerSecond; // Maximum requests allowed per second
  7. private final ConcurrentLinkedQueue<Long> requestTimestamps; //Queue to store request timestamps
  8. private final AtomicInteger requestCount; //Atomic counter for request count
  10. public RequestThrottler(int maxRequestsPerSecond) {
  11. this.maxRequestsPerSecond = maxRequestsPerSecond;
  12. this.requestTimestamps = new ConcurrentLinkedQueue<>();
  13. this.requestCount = new AtomicInteger(0);
  14. }
  16. /**
  17. * Processes a request, throttling it if necessary.
  18. *
  19. * @param timestamp The timestamp of the request.
  20. * @return True if the request is allowed, false otherwise.
  21. */
  22. public boolean allowRequest(long timestamp) {
  23. // Remove outdated timestamps from the queue
  24. while (!requestTimestamps.isEmpty() && requestTimestamps.peek() <= timestamp - 1000) { //remove timestamps older than 1 second.
  25. requestTimestamps.poll();
  26. }
  28. // Check if the request count exceeds the limit
  29. if (requestCount.get() >= maxRequestsPerSecond) {
  30. return false; // Request throttled
  31. }
  33. // Allow the request and update the count and timestamp
  34. requestCount.incrementAndGet();
  35. requestTimestamps.offer(timestamp);
  36. return true; // Request allowed
  37. }
  39. public static void main(String[] args) {
  40. //Example Usage
  41. RequestThrottler throttler = new RequestThrottler(5); // Allow 5 requests per second
  43. for (int i = 0; i < 10; i++) {
  44. long timestamp = System.currentTimeMillis();
  45. if (throttler.allowRequest(timestamp)) {
  46. System.out.println("Request allowed at: " + timestamp);
  47. } else {
  48. System.out.println("Request throttled at: " + timestamp);
  49. }
  50. try {
  51. Thread.sleep(100); // Simulate requests arriving at different times
  52. } catch (InterruptedException e) {
  53. e.printStackTrace();
  54. }
  55. }
  56. }
  57. }

Add your comment