1. import java.util.concurrent.ConcurrentLinkedQueue;
  2. import java.util.concurrent.atomic.AtomicInteger;
  4. public class UserRecordThrottler {
  6. private final int maxRequestsPerSecond;
  7. private final ConcurrentLinkedQueue<Long> requestQueue = new ConcurrentLinkedQueue<>();
  8. private final AtomicInteger requestCount = new AtomicInteger(0);
  10. public UserRecordThrottler(int maxRequestsPerSecond) {
  11. this.maxRequestsPerSecond = maxRequestsPerSecond;
  12. }
  14. /**
  15. * Attempts to process a user record request.
  16. *
  17. * @param userId The ID of the user.
  18. * @return True if the request was allowed, false if throttled.
  19. */
  20. public synchronized boolean allowRequest(long userId) {
  21. // Check if we're exceeding the rate limit
  22. if (requestCount.getAndIncrement() >= maxRequestsPerSecond) {
  23. return false; // Throttle
  24. }
  26. // Add the request to the queue
  27. requestQueue.offer(System.currentTimeMillis());
  29. // Process the request (simulated here)
  30. processRequest(userId);
  31. return true;
  32. }
  34. private void processRequest(long userId) {
  35. // Simulate processing the request. Replace with actual logic.
  36. try {
  37. // Simulate some work
  38. Thread.sleep(10);
  39. System.out.println("Processed request for user: " + userId);
  40. } catch (InterruptedException e) {
  41. Thread.currentThread().interrupt(); // Restore interrupted state
  42. System.err.println("Request processing interrupted: " + e.getMessage());
  43. }
  44. }
  48. public static void main(String[] args) throws InterruptedException {
  49. int maxRequests = 10; // Example: Allow 10 requests per second
  50. UserRecordThrottler throttler = new UserRecordThrottler(maxRequests);
  52. for (int i = 0; i < 20; i++) { // Simulate 20 requests
  53. long userId = i;
  54. if (throttler.allowRequest(userId)) {
  55. System.out.println("Request " + userId + " allowed.");
  56. } else {
  57. System.out.println("Request " + userId + " throttled.");
  58. }
  59. Thread.sleep(50); // Simulate request intervals
  60. }
  61. }
  62. }

Add your comment