Fix `Arc` incorrectly initializing itself as `Rc`

diff --git a/narcissus-core/src/ref_count.rs b/narcissus-core/src/ref_count.rs
index 5408b35e53250f9af2055e0dcc4db570064fa09f..35ea1d230fd3d0dc3066cc0b1fd05d84ce1259aa 100644 (file)
--- a/narcissus-core/src/ref_count.rs
+++ b/narcissus-core/src/ref_count.rs
@@ -21,6 +21,14 @@ impl<T> Inner<T> {
             value,
         }
     }
+
+    #[inline]
+    fn new_atomic(value: T) -> Self {
+        Self {
+            strong: AtomicI32::new(1),
+            value,
+        }
+    }
 }
 
 impl<T: ?Sized> Inner<T> {
@@ -188,7 +196,7 @@ unsafe impl<T: ?Sized + Sync + Send> Sync for Arc<T> {}
 
 impl<T> Arc<T> {
     pub fn new(value: T) -> Self {
-        Self::from_inner(Box::leak(Box::new(Inner::new(value))).into())
+        Self::from_inner(Box::leak(Box::new(Inner::new_atomic(value))).into())
     }
 }
 
@@ -301,6 +309,28 @@ impl<T: ?Sized> Deref for Arc<T> {
 mod tests {
     use crate::*;
 
+    #[test]
+    fn rc_drop() {
+        use std::sync::atomic::{AtomicU32, Ordering};
+
+        struct A<'a>(&'a AtomicU32);
+        impl<'a> Drop for A<'a> {
+            fn drop(&mut self) {
+                self.0.fetch_add(1, Ordering::SeqCst);
+            }
+        }
+
+        let counter = AtomicU32::new(0);
+        let a = Arc::new(A(&counter));
+        assert_eq!(counter.load(Ordering::Relaxed), 0);
+        let b = a.clone();
+        assert_eq!(counter.load(Ordering::Relaxed), 0);
+        drop(a);
+        assert_eq!(counter.load(Ordering::Relaxed), 0);
+        drop(b);
+        assert_eq!(counter.load(Ordering::Relaxed), 1);
+    }
+
     #[test]
     fn rc_double_upgrade() {
         let rc1 = Rc::new(());