1. import java.util.concurrent.ConcurrentLinkedQueue;
  3. public class ApiThrottler {
  5. private final int maxRequestsPerSecond; // Maximum requests allowed per second
  6. private final ConcurrentLinkedQueue<Long> requestQueue = new ConcurrentLinkedQueue<>(); // Queue to store request timestamps
  7. private final boolean dryRun; // Enable dry-run mode
  9. public ApiThrottler(int maxRequestsPerSecond, boolean dryRun) {
  10. this.maxRequestsPerSecond = maxRequestsPerSecond;
  11. this.dryRun = dryRun;
  12. }
  14. public synchronized boolean allowRequest() {
  15. long now = System.currentTimeMillis();
  17. // Remove requests older than 1 second
  18. while (!requestQueue.isEmpty() && requestQueue.peek() <= now - 1000) {
  19. requestQueue.poll();
  20. }
  22. // Check if the rate limit is exceeded
  23. if (requestQueue.size() >= maxRequestsPerSecond) {
  24. if (dryRun) {
  25. System.out.println("Dry Run: Request throttled. Rate limit exceeded.");
  26. return false; // Simulate throttling in dry-run mode
  27. } else {
  28. try {
  29. Thread.sleep(1000); // Wait for 1 second
  30. } catch (InterruptedException e) {
  31. Thread.currentThread().interrupt();
  32. }
  33. return false; // Actually throttle the request
  34. }
  35. }
  37. requestQueue.offer(now); // Add the current request timestamp to the queue
  38. return true; // Allow the request
  39. }
  41. public static void main(String[] args) throws InterruptedException {
  42. // Example usage
  43. int maxRequests = 5; // Allow 5 requests per second
  44. boolean isDryRun = true; // Set to false for actual throttling
  46. ApiThrottler throttler = new ApiThrottler(maxRequests, isDryRun);
  48. for (int i = 0; i < 10; i++) {
  49. if (throttler.allowRequest()) {
  50. System.out.println("Request " + (i + 1) + " allowed.");
  51. // Simulate API call here
  52. } else {
  53. System.out.println("Request " + (i + 1) + " throttled.");
  54. }
  55. Thread.sleep(200); // Simulate request interval
  56. }
  57. }
  58. }

Add your comment