1. <?php
  3. class RequestThrottler {
  5. private $request_headers = []; // Array to store request header counts
  6. private $rate_limit; // Request rate limit (e.g., 10 requests per minute)
  7. private $window; // Time window for rate limiting (in seconds)
  9. public function __construct(array $allowed_headers, int $rate_limit, int $window = 60) {
  10. $this->request_headers = $allowed_headers;
  11. $this->rate_limit = $rate_limit;
  12. $this->window = $window;
  13. }
  15. public function isAllowed(array $headers): bool {
  16. // Reset counts for headers not in the allowed list.
  17. foreach ($this->request_headers as $header) {
  18. if (!isset($this->request_headers[$header])) {
  19. $this->request_headers[$header] = 0;
  20. }
  21. }
  23. $now = time();
  25. // Remove expired entries from the window.
  26. $this->cleanupWindow($now);
  28. // Check if the header count exceeds the rate limit.
  29. foreach ($headers as $header => $value) {
  30. if (isset($this->request_headers[$header])) {
  31. $this->request_headers[$header]++;
  32. if ($this->request_headers[$header] > $this->rate_limit) {
  33. return false; // Rate limit exceeded
  34. }
  35. }
  36. }
  38. return true; // Request allowed
  39. }
  41. private function cleanupWindow(int $now): void {
  42. foreach ($this->request_headers as $header => $count) {
  43. if ($now - $this->window > 0) {
  44. $this->request_headers[$header] = 0;
  45. }
  46. }
  47. }
  48. }

Add your comment