| 12345678910111213141516171819202122232425262728293031323334353637383940414243 |
- from typing import TYPE_CHECKING
- if TYPE_CHECKING:
- from typing import Any
- _SENTINEL = object()
- class LRUCache:
- def __init__(self, max_size: int) -> None:
- if max_size <= 0:
- raise AssertionError(f"invalid max_size: {max_size}")
- self.max_size = max_size
- self._data: "dict[Any, Any]" = {}
- self.hits = self.misses = 0
- self.full = False
- def set(self, key: "Any", value: "Any") -> None:
- current = self._data.pop(key, _SENTINEL)
- if current is not _SENTINEL:
- self._data[key] = value
- elif self.full:
- self._data.pop(next(iter(self._data)))
- self._data[key] = value
- else:
- self._data[key] = value
- self.full = len(self._data) >= self.max_size
- def get(self, key: "Any", default: "Any" = None) -> "Any":
- try:
- ret = self._data.pop(key)
- except KeyError:
- self.misses += 1
- ret = default
- else:
- self.hits += 1
- self._data[key] = ret
- return ret
- def get_all(self) -> "list[tuple[Any, Any]]":
- return list(self._data.items())
|
