1. import java.util.concurrent.ConcurrentLinkedQueue;
  3. public class RequestThrottle {
  5. private final int maxRequestsPerSecond;
  6. private final ConcurrentLinkedQueue<Long> requestTimestamps;
  8. public RequestThrottle(int maxRequestsPerSecond) {
  9. this.maxRequestsPerSecond = maxRequestsPerSecond;
  10. this.requestTimestamps = new ConcurrentLinkedQueue<>();
  11. }
  13. public boolean allowRequest() {
  14. long now = System.currentTimeMillis();
  16. // Remove timestamps older than 1 second
  17. while (!requestTimestamps.isEmpty() && requestTimestamps.peek() <= now - 1000) {
  18. requestTimestamps.poll();
  19. }
  21. // Check if the request limit has been exceeded
  22. if (requestTimestamps.size() >= maxRequestsPerSecond) {
  23. return false; // Throttled
  24. }
  26. requestTimestamps.offer(now); // Add the current timestamp
  27. return true; // Allowed
  28. }
  30. public static void main(String[] args) throws InterruptedException {
  31. // Example usage:
  32. RequestThrottle throttle = new RequestThrottle(5); // Allow 5 requests per second
  34. for (int i = 0; i < 10; i++) {
  35. if (throttle.allowRequest()) {
  36. System.out.println("Request " + (i + 1) + " allowed.");
  37. } else {
  38. System.out.println("Request " + (i + 1) + " throttled.");
  39. }
  40. Thread.sleep(100); // Simulate requests coming in
  41. }
  42. }
  43. }

Add your comment