📝 add take for RcuWeak

Xudong-Huang · Dec 27, 2024 · 72bcc95 · 72bcc95
1 parent 70cee0d
commit 72bcc95
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/rcu_weak.rs b/src/rcu_weak.rs
@@ -63,6 +63,12 @@ impl<T> RcuWeak<T> {
         ret
     }
 
+    /// take the value from the rcu weak, leave the rcu weak with default value
+    #[inline]
+    pub fn take(&self) -> Weak<T> {
+        ptr_to_weak(self.link.update(ptr::null()))
+    }
+
     /// write a new weak value to the rcu weak cell and return the old value
     #[inline]
     pub fn write(&self, data: Weak<T>) -> Weak<T> {