reserve to ret &mut self

tickbh · Jun 17, 2024 · ec5168f · ec5168f
1 parent 281410b
commit ec5168f
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/src/cache/arc.rs b/src/cache/arc.rs
@@ -122,12 +122,13 @@ impl<K, V, S> ArcCache<K, V, S> {
     }
 
     /// 扩展当前容量
-    pub fn reserve(&mut self, additional: usize) {
+    pub fn reserve(&mut self, additional: usize) -> &mut Self {
         self.cap += additional;
         self.main_lfu.reserve(additional);
         self.main_lru.reserve(additional);
         self.ghost_lfu.reserve(additional);
         self.ghost_lru.reserve(additional);
+        self
     }
 
     /// 遍历当前的所有值

diff --git a/src/cache/lfu.rs b/src/cache/lfu.rs
@@ -243,8 +243,9 @@ impl<K, V, S> LfuCache<K, V, S> {
     }
 
     /// 扩展当前容量
-    pub fn reserve(&mut self, additional: usize) {
+    pub fn reserve(&mut self, additional: usize) -> &mut Self {
         self.cap += additional;
+        self
     }
 
     /// 遍历当前的所有值

diff --git a/src/cache/lruk.rs b/src/cache/lruk.rs
@@ -208,8 +208,9 @@ impl<K, V, S> LruKCache<K, V, S> {
     }
 
     /// 扩展当前容量
-    pub fn reserve(&mut self, additional: usize) {
+    pub fn reserve(&mut self, additional: usize) -> &mut Self {
         self.cap += additional;
+        self
     }
 
     /// 遍历当前的所有值