Expose share counts in Buffer (#1072)

jorgecarleitao · Jun 14, 2022 · 72df09f · 72df09f
1 parent 83082fa
commit 72df09f
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/src/buffer/immutable.rs b/src/buffer/immutable.rs
@@ -168,6 +168,16 @@ impl<T: NativeType> Buffer<T> {
             Either::Right(data) => data,
         }
     }
+
+    /// Get the strong count of underlying `Arc` data buffer.
+    pub fn shared_count_strong(&self) -> usize {
+        Arc::strong_count(&self.data)
+    }
+
+    /// Get the weak count of underlying `Arc` data buffer.
+    pub fn shared_count_weak(&self) -> usize {
+        Arc::weak_count(&self.data)
+    }
 }
 
 impl<T: NativeType> From<Vec<T>> for Buffer<T> {