diff --git a/queues/logger_rate_limiter.py b/queues/stream_with_rate_limiter.py similarity index 84% rename from queues/logger_rate_limiter.py rename to queues/stream_with_rate_limiter.py index 4b35009..da58743 100644 --- a/queues/logger_rate_limiter.py +++ b/queues/stream_with_rate_limiter.py @@ -5,11 +5,10 @@ class Logger: def __init__(self): - self.msg_set = set() self.msg_queue = deque() - def print_message(self, timestamp: int, message: str) -> bool: + def print_message(self, timestamp, message) -> bool: while self.msg_queue: msg, msg_timestamp = self.msg_queue[0] @@ -23,6 +22,6 @@ class Logger: self.msg_set.add(message) self.msg_queue.append((message, timestamp)) return True - else: - return False + + return False