46 lines
1.4 KiB
Python
46 lines
1.4 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import time
|
||
|
|
from typing import Any
|
||
|
|
|
||
|
|
|
||
|
|
class SentMessageCache:
|
||
|
|
MAX_ENTRIES = 1024
|
||
|
|
TTL_SECONDS = 600
|
||
|
|
|
||
|
|
def __init__(self) -> None:
|
||
|
|
self._cache: dict[str, dict[str, Any]] = {}
|
||
|
|
|
||
|
|
def track(self, message_id: str, chat_guid: str, metadata: dict[str, Any] | None = None) -> None:
|
||
|
|
self._cache[message_id] = {
|
||
|
|
"chat_guid": chat_guid,
|
||
|
|
"metadata": metadata or {},
|
||
|
|
"sent_at": time.monotonic(),
|
||
|
|
}
|
||
|
|
self._trim()
|
||
|
|
|
||
|
|
def lookup(self, message_id: str) -> dict[str, Any] | None:
|
||
|
|
entry = self._cache.get(message_id)
|
||
|
|
if entry is None:
|
||
|
|
return None
|
||
|
|
if time.monotonic() - entry["sent_at"] > self.TTL_SECONDS:
|
||
|
|
del self._cache[message_id]
|
||
|
|
return None
|
||
|
|
return entry
|
||
|
|
|
||
|
|
def remove(self, message_id: str) -> None:
|
||
|
|
self._cache.pop(message_id, None)
|
||
|
|
|
||
|
|
def _trim(self) -> None:
|
||
|
|
if len(self._cache) > self.MAX_ENTRIES:
|
||
|
|
now = time.monotonic()
|
||
|
|
expired = [k for k, v in self._cache.items() if now - v["sent_at"] > self.TTL_SECONDS]
|
||
|
|
for k in expired:
|
||
|
|
del self._cache[k]
|
||
|
|
if len(self._cache) > self.MAX_ENTRIES:
|
||
|
|
oldest = sorted(self._cache.items(), key=lambda x: x[1]["sent_at"])[
|
||
|
|
: len(self._cache) - self.MAX_ENTRIES
|
||
|
|
]
|
||
|
|
for k, _ in oldest:
|
||
|
|
del self._cache[k]
|