1. import java.util.HashMap;
  2. import java.util.Map;
  3. import java.util.concurrent.ConcurrentHashMap;
  4. import java.util.concurrent.atomic.AtomicLong;
  6. public class UrlParameterEnvironment {
  8. private final Map<String, Long> requestCounts = new ConcurrentHashMap<>(); // Track request counts per URL parameter
  9. private final Map<String, Long> lastRequestTimes = new ConcurrentHashMap<>(); // Track last request time per URL parameter
  10. private final long rateLimitWindow = 60; // Time window in seconds
  11. private final long maxRequestsPerWindow = 10; // Maximum requests allowed per window
  13. public boolean isAllowed(String parameter) {
  14. long now = System.currentTimeMillis();
  15. long lastRequestTime = lastRequestTimes.getOrDefault(parameter, 0L);
  17. // Check if within the rate limit window
  18. if (now - lastRequestTime < rateLimitWindow * 1000) {
  19. // Request is within the window, check if rate limit exceeded
  20. long requestCount = requestCounts.getOrDefault(parameter, 0L);
  21. if (requestCount >= maxRequestsPerWindow) {
  22. return false; // Rate limit exceeded
  23. }
  24. return true; // Request allowed
  25. } else {
  26. // Request is outside the window, reset the count
  27. requestCounts.remove(parameter);
  28. lastRequestTimes.remove(parameter);
  29. requestCounts.put(parameter, 1L);
  30. lastRequestTimes.put(parameter, now);
  31. return true; // Request allowed
  32. }
  33. }
  35. public void incrementRequestCount(String parameter) {
  36. requestCounts.compute(parameter, (key, count) -> (count == null) ? 1 : count + 1);
  37. }
  39. public long getRateLimitWindow() {
  40. return rateLimitWindow;
  41. }
  43. public long getMaxRequestsPerWindow() {
  44. return maxRequestsPerWindow;
  45. }
  46. }

Add your comment