1. import java.util.concurrent.ConcurrentLinkedQueue;
  2. import java.util.concurrent.atomic.AtomicInteger;
  4. public class TimestampThrottler {
  6. private final int maxRequestsPerSecond;
  7. private final ConcurrentLinkedQueue<Long> requestQueue = new ConcurrentLinkedQueue<>();
  8. private final AtomicInteger requestCount = new AtomicInteger(0);
  10. /**
  11. * Constructor with maximum requests per second.
  12. * @param maxRequestsPerSecond The maximum number of requests allowed per second.
  13. */
  14. public TimestampThrottler(int maxRequestsPerSecond) {
  15. this.maxRequestsPerSecond = maxRequestsPerSecond;
  16. }
  18. /**
  19. * Adds a timestamp to the request queue, throttling requests.
  20. * @param timestamp The timestamp to add.
  21. */
  22. public void addRequest(long timestamp) {
  23. // Wait if the request queue is full
  24. while (requestQueue.size() >= maxRequestsPerSecond) {
  25. try {
  26. // Calculate how long to wait
  27. long timeToWait = 1000 / maxRequestsPerSecond; // Milliseconds
  28. Thread.sleep(timeToWait);
  29. } catch (InterruptedException e) {
  30. Thread.currentThread().interrupt();
  31. return; // Exit if interrupted
  32. }
  33. }
  34. requestQueue.offer(timestamp);
  35. }
  38. /**
  39. * Retrieves the next timestamp from the queue.
  40. * @return The next timestamp, or -1 if the queue is empty.
  41. */
  42. public long getNextTimestamp() {
  43. if (!requestQueue.isEmpty()) {
  44. return requestQueue.poll();
  45. }
  46. return -1; // Indicate queue is empty
  47. }
  49. public static void main(String[] args) throws InterruptedException {
  50. int maxRequests = 5;
  51. TimestampThrottler throttler = new TimestampThrottler(maxRequests);
  53. for (int i = 0; i < 20; i++) {
  54. long timestamp = System.currentTimeMillis() + i * 100;
  55. throttler.addRequest(timestamp);
  56. System.out.println("Added request: " + timestamp);
  57. Thread.sleep(50); // Simulate some processing time
  58. }
  60. System.out.println("Processing requests...");
  61. long nextTimestamp = throttler.getNextTimestamp();
  62. while (nextTimestamp != -1) {
  63. System.out.println("Processing timestamp: " + nextTimestamp);
  64. nextTimestamp = throttler.getNextTimestamp();
  65. }
  66. System.out.println("All requests processed.");
  67. }
  68. }

Add your comment