cycle_ptr 0.1.0

Smart pointers, with cycles
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

//! Implement the read and write locks on generation.
#[cfg(not(feature = "single_generation_mt"))]
use super::TaskExclusion;
use super::{MTControlBlock, MTGeneration, ObjectsLink};
use crate::list::List;
use crate::util::NonNull;
use std::sync::RwLockReadGuard;
#[cfg(not(feature = "single_generation_mt"))]
use std::sync::{MutexGuard, RwLockWriteGuard};

/// A shared lock on a generation.
///
/// Prevents objects in the generation from changing membership.
pub(crate) struct MTGenerationReadLock<'a> {
    /// Raw pointer to the generation, needed to implement [MTGenerationReadLock::same_generation].
    raw_generation_ptr: NonNull<MTGeneration>,
    /// Objects contained in this generation.
    _objects: RwLockReadGuard<'a, List<MTControlBlock, ObjectsLink>>,
}

/// An exclusive lock on a generation.
///
/// Prevents objects in the generation from changing membership and prevents the garbage collector from running.
#[cfg(not(feature = "single_generation_mt"))]
pub(crate) struct MTGenerationWriteLock<'a> {
    /// Task exclusion lock, to prevent the garbage collector from running.
    _task_exclusion_lock: MutexGuard<'a, TaskExclusion>,
    /// Objects contained in this generation.
    pub(super) objects: RwLockWriteGuard<'a, List<MTControlBlock, ObjectsLink>>,
}

impl<'a> MTGenerationReadLock<'a> {
    /// Instantiate a lock.
    #[allow(
        clippy::missing_const_for_fn,
        reason = "You can't create these in a const context anyway."
    )]
    #[inline]
    pub(super) fn new(
        raw_generation_ptr: NonNull<MTGeneration>,
        objects: RwLockReadGuard<'a, List<MTControlBlock, ObjectsLink>>,
    ) -> Self {
        MTGenerationReadLock {
            raw_generation_ptr,
            _objects: objects,
        }
    }

    /// Test if this lock locks the specified generation.
    #[inline]
    pub(crate) fn same_generation(&self, other: &MTGeneration) -> bool {
        self.raw_generation_ptr == NonNull::from_ref(other)
    }
}

#[cfg(not(feature = "single_generation_mt"))]
impl<'a> MTGenerationWriteLock<'a> {
    /// Instantiate a lock.
    #[allow(
        clippy::missing_const_for_fn,
        reason = "You can't create these in a const context anyway."
    )]
    #[inline]
    pub(super) fn new(
        task_exclusion_lock: MutexGuard<'a, TaskExclusion>,
        objects: RwLockWriteGuard<'a, List<MTControlBlock, ObjectsLink>>,
    ) -> Self {
        MTGenerationWriteLock {
            _task_exclusion_lock: task_exclusion_lock,
            objects,
        }
    }
}