     1
     2
     3
     4
     5
     6
     7
     8
     9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    22
    23
    24
    25
    26
    27
    28
    29
    30
    31
    32
    33
    34
    35
    36
    37
    38
    39
    40
    41
    42
    43
    44
    45
    46
    47
    48
    49
    50
    51
    52
    53
    54
    55
    56
    57
    58
    59
    60
    61
    62
    63
    64
    65
    66
    67
    68
    69
    70
    71
    72
    73
    74
    75
    76
    77
    78
    79
    80
    81
    82
    83
    84
    85
    86
    87
    88
    89

    use std::marker::PhantomData;
    use std::mem;
    use std::sync::Arc;
    use std::sync::atomic::{AtomicUsize, Ordering};

    /// A type providing atomic storage and retrieval of an `Arc<T>`.
    pub struct ArcCell<T>(AtomicUsize, PhantomData<Arc<T>>);

    impl<T> Drop for ArcCell<T> {
        fn drop(&mut self) {
            self.take();
        }
    }

    impl<T> ArcCell<T> {
        /// Creates a new `ArcCell`.
        pub fn new(t: Arc<T>) -> ArcCell<T> {
            ArcCell(AtomicUsize::new(unsafe { mem::transmute(t) }), PhantomData)
        }

        fn take(&self) -> Arc<T> {
            loop {
                match self.0.swap(0, Ordering::Acquire) {
                    0 => {}
                    n => return unsafe { mem::transmute(n) }
                }
            }
        }

        fn put(&self, t: Arc<T>) {
            debug_assert_eq!(self.0.load(Ordering::SeqCst), 0);
            self.0.store(unsafe { mem::transmute(t) }, Ordering::Release);
        }

        /// Stores a new value in the `ArcCell`, returning the previous
        /// value.
        pub fn set(&self, t: Arc<T>) -> Arc<T> {
            let old = self.take();
            self.put(t);
            old
        }

        /// Returns a copy of the value stored by the `ArcCell`.
        pub fn get(&self) -> Arc<T> {
            let t = self.take();
            // NB: correctness here depends on Arc's clone impl not panicking
            let out = t.clone();
            self.put(t);
            out
        }
    }

    #[cfg(test)]
    mod test {
        use std::sync::Arc;
        use std::sync::atomic::{ATOMIC_USIZE_INIT, AtomicUsize, Ordering};

        use super::*;

        #[test]
        fn basic() {
            let r = ArcCell::new(Arc::new(0));
            assert_eq!(*r.get(), 0);
            assert_eq!(*r.set(Arc::new(1)), 0);
            assert_eq!(*r.get(), 1);
        }

        #[test]
        fn drop_runs() {
            static DROPS: AtomicUsize = ATOMIC_USIZE_INIT;

            struct Foo;

            impl Drop for Foo {
                fn drop(&mut self) {
                    DROPS.fetch_add(1, Ordering::SeqCst);
                }
            }

            let r = ArcCell::new(Arc::new(Foo));
            let _f = r.get();
            r.get();
            r.set(Arc::new(Foo));
            drop(_f);
            assert_eq!(DROPS.load(Ordering::SeqCst), 1);
            drop(r);
            assert_eq!(DROPS.load(Ordering::SeqCst), 2);
        }
    }

# Help

## Keyboard Shortcuts

?
  Show this help dialog
S
  Focus the search field
⇤
  Move up in search results
⇥
  Move down in search results
⏎
  Go to active search result

## Search Tricks

Prefix searches with a type followed by a colon (e.g. `fn:`) to restrict the search to a given type.

Accepted types are: `fn`, `mod`, `struct`, `enum`, `trait`, `type`, `macro`, and `const`.

Search functions by type signature (e.g. `vec -> usize` or `* -> vec`)