rds_05.py

from redis_data_structures import RingBuffer
from datetime import datetime, timedelta

class RateLimiter:
    def __init__(self, window_size: int, max_requests: int):
        self.buffer = RingBuffer("rate-limits", capacity=max_requests)
        self.window_size = window_size
    
    def is_allowed(self, client_id: str) -> bool:
        key = f"rate_limit:{client_id}"
        now = datetime.now()
        cutoff = now - timedelta(seconds=self.window_size)
        
        # Cleanup old requests first
        requests = self.buffer.get_all(key)
        valid_requests = [
            req for req in requests
            if datetime.fromisoformat(req["timestamp"]) > cutoff
        ]
        
        if len(valid_requests) >= self.buffer.capacity:
            return False
            
        self.buffer.push({"timestamp": now.isoformat()})
        return True