mirror of
https://github.com/borgbackup/borg.git
synced 2026-05-28 04:03:21 -04:00
32cfddc34c
fix: resolve KeyError and memory leaks in LRUCache - __setitem__: assign value before popping from _lru to avoid KeyError when exceeding capacity. - clear(): clear the _lru list also to prevent stale keys causing KeyErrors during future evictions.
58 lines
1.7 KiB
Python
58 lines
1.7 KiB
Python
sentinel = object()
|
|
|
|
|
|
class LRUCache:
|
|
def __init__(self, capacity, dispose):
|
|
self._cache = {}
|
|
self._lru = []
|
|
self._capacity = capacity
|
|
self._dispose = dispose
|
|
|
|
def __setitem__(self, key, value):
|
|
assert key not in self._cache, (
|
|
"Unexpected attempt to replace a cached item,"
|
|
" without first deleting the old item.")
|
|
self._cache[key] = value
|
|
self._lru.append(key)
|
|
while len(self._lru) > self._capacity:
|
|
del self[self._lru[0]]
|
|
|
|
def __getitem__(self, key):
|
|
value = self._cache[key] # raise KeyError if not found
|
|
self._lru.remove(key)
|
|
self._lru.append(key)
|
|
return value
|
|
|
|
def __delitem__(self, key):
|
|
value = self._cache.pop(key) # raise KeyError if not found
|
|
self._dispose(value)
|
|
self._lru.remove(key)
|
|
|
|
def __contains__(self, key):
|
|
return key in self._cache
|
|
|
|
def get(self, key, default=None):
|
|
value = self._cache.get(key, sentinel)
|
|
if value is sentinel:
|
|
return default
|
|
self._lru.remove(key)
|
|
self._lru.append(key)
|
|
return value
|
|
|
|
def upd(self, key, value):
|
|
# Special use only: update the value for an existing key without having to dispose it first.
|
|
# This method complements __setitem__, which should be used for the normal use case.
|
|
assert key in self._cache, "Unexpected attempt to update a non-existing item."
|
|
self._cache[key] = value
|
|
|
|
def clear(self):
|
|
for value in self._cache.values():
|
|
self._dispose(value)
|
|
self._cache.clear()
|
|
self._lru.clear()
|
|
|
|
def items(self):
|
|
return self._cache.items()
|
|
|
|
def __len__(self):
|
|
return len(self._cache)
|