logos-storage/logos-storage-nim-dht
1
0
Fork 0
mirror of https://github.com/logos-storage/logos-storage-nim-dht.git synced 2026-01-05 23:13:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

prevent npe when table is 0

Browse Source
This commit is contained in:
Dmitriy Ryajov 2022-09-12 21:08:55 -06:00
parent f84bc647ce
commit f5afe784c5
No known key found for this signature in database
GPG Key ID: DA8C680CE7C657A4
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
libp2pdht/private/eth/p2p/discoveryv5/lru.nim
View File

@ -11,6 +11,7 @@ type
capacity: int capacity: int
func init*[K, V](T: type LRUCache[K, V], capacity: int): LRUCache[K, V] = func init*[K, V](T: type LRUCache[K, V], capacity: int): LRUCache[K, V] =
doAssert capacity > 0, "Capacity should be greater than 0!"
LRUCache[K, V](capacity: capacity) # Table and list init is done default LRUCache[K, V](capacity: capacity) # Table and list init is done default
func get*[K, V](lru: var LRUCache[K, V], key: K): Option[V] = func get*[K, V](lru: var LRUCache[K, V], key: K): Option[V] =
@ -27,7 +28,7 @@ func put*[K, V](lru: var LRUCache[K, V], key: K, value: V) =
if not node.isNil: if not node.isNil:
lru.list.remove(node) lru.list.remove(node)
else: else:
if lru.table.len >= lru.capacity: if lru.len > 0 and lru.table.len >= lru.capacity:
lru.table.del(lru.list.tail.value[0]) lru.table.del(lru.list.tail.value[0])
lru.list.remove(lru.list.tail) lru.list.remove(lru.list.tail)
@ -43,10 +44,13 @@ func len*[K, V](lru: LRUCache[K, V]): int =
lru.table.len lru.table.len
proc contains*[K, V](lru: LRUCache[K, V], k: K): bool = proc contains*[K, V](lru: LRUCache[K, V], k: K): bool =
## Check for cached item - this doesn't touch the cache
##
k in lru.table k in lru.table
iterator items*[K, V](lru: LRUCache[K, V]): V = iterator items*[K, V](lru: LRUCache[K, V]): V =
## This will not increment LRU/MRU access stats ## Get cached items - this doesn't touch the cache
## ##
for item in lru.list: for item in lru.list: