feat(pkarr): add Cache::capacity() method

Nuhvi · Feb 17, 2025 · 85bd27e · 85bd27e
1 parent e712fd2
commit 85bd27e
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/pkarr/src/client/cache.rs b/pkarr/src/client/cache.rs
@@ -31,6 +31,11 @@ impl From<crate::PublicKey> for CacheKey {
 
 /// A trait for a [SignedPacket]s cache for Pkarr [Client][crate::Client].
 pub trait Cache: Debug + Send + Sync + DynClone {
+    /// Returns the maximum capacity of [SignedPacket]s allowed in this cache.
+    fn capacity(&self) -> usize {
+        // backward compatibility
+        0
+    }
     /// Returns the number of [SignedPacket]s in this cache.
     fn len(&self) -> usize;
     /// Returns true if this cache is empty.
@@ -73,6 +78,14 @@ impl InMemoryCache {
 }
 
 impl Cache for InMemoryCache {
+    fn capacity(&self) -> usize {
+        self.inner
+            .read()
+            .expect("InMemoryCache RwLock")
+            .cap()
+            .into()
+    }
+
     fn len(&self) -> usize {
         self.inner.read().expect("InMemoryCache RwLock").len()
     }

diff --git a/pkarr/src/extra/lmdb_cache.rs b/pkarr/src/extra/lmdb_cache.rs
@@ -251,6 +251,10 @@ fn update_lru(
 }
 
 impl Cache for LmdbCache {
+    fn capacity(&self) -> usize {
+        self.capacity
+    }
+
     fn len(&self) -> usize {
         match self.internal_len() {
             Ok(result) => result,