1. import time
  2. import logging
  4. logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
  6. class ArrayRequestThrottler:
  7. def __init__(self, max_requests, interval):
  8. self.max_requests = max_requests
  9. self.interval = interval
  10. self.request_count = 0
  11. self.last_request_time = 0
  12. self.lock = threading.Lock() # Protects shared resources
  14. def __enter__(self):
  15. return self
  17. def __exit__(self, exc_type, exc_val, exc_tb):
  18. pass
  20. def throttle(self):
  21. with self.lock:
  22. elapsed_time = time.time() - self.last_request_time
  23. if elapsed_time < self.interval:
  24. time.sleep(self.interval - elapsed_time)
  26. self.last_request_time = time.time()
  27. self.request_count += 1
  29. if self.request_count >= self.max_requests:
  30. logging.warning(f"Throttling: Max requests ({self.max_requests}) reached. Pausing.")
  31. time.sleep(self.interval) #Pause if max requests reached
  32. self.request_count = 0 #Reset counter
  33. self.last_request_time = time.time()
  34. logging.info("Throttling reset")
  36. def request_array(self, array):
  37. self.throttle()
  38. logging.info(f"Processing array: {array}")
  39. # Simulate array processing
  40. time.sleep(0.1)
  41. logging.info(f"Finished processing array: {array}")
  43. import threading
  45. if __name__ == '__main__':
  46. max_requests = 5
  47. interval = 2 # seconds
  48. throttler = ArrayRequestThrottler(max_requests, interval)
  50. arrays = [list(range(10)), list(range(11)), list(range(12)), list(range(13)), list(range(14)), list(range(15)), list(range(16))]
  52. for arr in arrays:
  53. with throttler:
  54. throttler.request_array(arr)

Add your comment