froodi 1.0.0-beta.15

An ergonomic Rust IoC container
Documentation 
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

#![allow(dead_code)]

use ahash::AHasher;
use core::{
    array,
    hash::{Hash, Hasher as _},
};
use parking_lot::Mutex;

use crate::utils::thread_safety::RcThreadSafety;

#[allow(clippy::cast_possible_truncation)]
fn stripe_index<const N: usize>(val: impl Hash) -> usize {
    let mut hasher = AHasher::default();
    val.hash(&mut hasher);

    hasher.finish() as usize % N
}

#[derive(Clone)]
pub(crate) struct StripedLocks<const N: usize> {
    stripes: RcThreadSafety<[Mutex<()>; N]>,
}

impl<const N: usize> StripedLocks<N> {
    #[inline]
    #[must_use]
    fn new() -> Self {
        Self {
            stripes: RcThreadSafety::new(array::from_fn(|_| Mutex::new(()))),
        }
    }
}

impl<const N: usize> StripedLocks<N> {
    #[inline]
    #[must_use]
    pub(crate) fn get(&self, val: impl Hash) -> &Mutex<()> {
        &self.stripes[stripe_index::<N>(val)]
    }
}

impl<const N: usize> Default for StripedLocks<N> {
    #[inline]
    fn default() -> Self {
        Self::new()
    }
}

#[cfg(feature = "async")]
mod async_impl {
    use tokio::sync::Mutex;

    use super::{array, stripe_index, Hash, RcThreadSafety};

    #[derive(Clone)]
    pub(crate) struct StripedSharedLocks<const N: usize> {
        stripes: RcThreadSafety<[Mutex<()>; N]>,
    }

    impl<const N: usize> StripedSharedLocks<N> {
        #[inline]
        #[must_use]
        fn new() -> Self {
            Self {
                stripes: RcThreadSafety::new(array::from_fn(|_| Mutex::new(()))),
            }
        }
    }

    impl<const N: usize> StripedSharedLocks<N> {
        #[inline]
        #[must_use]
        pub(crate) fn get(&self, val: impl Hash) -> &Mutex<()> {
            &self.stripes[stripe_index::<N>(val)]
        }
    }

    impl<const N: usize> Default for StripedSharedLocks<N> {
        #[inline]
        fn default() -> Self {
            Self::new()
        }
    }
}

#[cfg(feature = "async")]
pub(crate) use async_impl::StripedSharedLocks;