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

// Copyright 2016 Amanieu d'Antras
//
// Licensed under the Apache License, Version 2.0, <LICENSE-APACHE or
// http://apache.org/licenses/LICENSE-2.0> or the MIT license <LICENSE-MIT or
// http://opensource.org/licenses/MIT>, at your option. This file may not be
// copied, modified, or distributed except according to those terms.

#![allow(dead_code)]

use std::sync::atomic;

// Re-export this for convenience
pub use std::sync::atomic::{Ordering, fence};

// Wrapper around AtomicUsize for non-nightly which has usable compare_exchange
// and compare_exchange_weak methods.
pub struct AtomicUsize(atomic::AtomicUsize);
pub use self::AtomicUsize as AtomicU8;

// Constants for static initialization
pub const ATOMIC_USIZE_INIT: AtomicUsize = AtomicUsize(atomic::ATOMIC_USIZE_INIT);
pub use self::ATOMIC_USIZE_INIT as ATOMIC_U8_INIT;

impl AtomicUsize {
    #[inline]
    pub fn new(val: usize) -> AtomicUsize {
        AtomicUsize(atomic::AtomicUsize::new(val))
    }
    #[inline]
    pub fn load(&self, order: Ordering) -> usize {
        self.0.load(order)
    }
    #[inline]
    pub fn store(&self, val: usize, order: Ordering) {
        self.0.store(val, order);
    }
    #[inline]
    pub fn swap(&self, val: usize, order: Ordering) -> usize {
        self.0.swap(val, order)
    }
    #[inline]
    pub fn fetch_add(&self, val: usize, order: Ordering) -> usize {
        self.0.fetch_add(val, order)
    }
    #[inline]
    pub fn fetch_sub(&self, val: usize, order: Ordering) -> usize {
        self.0.fetch_sub(val, order)
    }
    #[inline]
    pub fn fetch_and(&self, val: usize, order: Ordering) -> usize {
        self.0.fetch_and(val, order)
    }
    #[inline]
    pub fn fetch_or(&self, val: usize, order: Ordering) -> usize {
        self.0.fetch_or(val, order)
    }
    #[inline]
    pub fn compare_exchange(&self,
                            old: usize,
                            new: usize,
                            order: Ordering,
                            _: Ordering)
                            -> Result<usize, usize> {
        let res = self.0.compare_and_swap(old, new, order);
        if res == old { Ok(res) } else { Err(res) }
    }
    #[inline]
    pub fn compare_exchange_weak(&self,
                                 old: usize,
                                 new: usize,
                                 order: Ordering,
                                 _: Ordering)
                                 -> Result<usize, usize> {
        let res = self.0.compare_and_swap(old, new, order);
        if res == old { Ok(res) } else { Err(res) }
    }
}