cranpose_core/slot/
table.rs

1use std::rc::Rc;
2
3use super::{
4    AnchorRegistry, DeferredDrop, GroupRecord, MovableIndex, NodeRecord, PayloadAnchorRegistry,
5    SlotLifecycleCoordinator, SlotWriteSessionState, debug::SlotTableDiagnostics,
6    growth::GrowthSlack, payload_store::PayloadStore,
7};
8
9mod metadata;
10mod mutation;
11mod values;
12
13#[cfg(test)]
14pub(crate) use values::ValueSlotError;
15
16struct SlotStorageIdentity(Rc<()>);
17
18impl SlotStorageIdentity {
19    fn new() -> Self {
20        Self(Rc::new(()))
21    }
22
23    fn id(&self) -> usize {
24        Rc::as_ptr(&self.0).addr()
25    }
26}
27
28pub(crate) struct SlotWriteSession<'a> {
29    pub(super) table: &'a mut SlotTable,
30    pub(in crate::slot) lifecycle: &'a mut SlotLifecycleCoordinator,
31    pub(in crate::slot) state: &'a mut SlotWriteSessionState,
32}
33
34pub struct SlotTable {
35    storage_id: SlotStorageIdentity,
36    pub(super) groups: Vec<GroupRecord>,
37    pub(super) payloads: PayloadStore,
38    pub(super) nodes: Vec<NodeRecord>,
39    pub(super) anchors: AnchorRegistry,
40    pub(super) payload_anchors: PayloadAnchorRegistry,
41    pub(super) movables: MovableIndex,
42    pub(super) diagnostics: SlotTableDiagnostics,
43    stale_group_indexes: metadata::StaleGroupIndexes,
44    next_group_generation: u32,
45}
46
47impl SlotTable {
48    pub fn new() -> Self {
49        Self {
50            storage_id: SlotStorageIdentity::new(),
51            groups: Vec::new(),
52            payloads: PayloadStore::default(),
53            nodes: Vec::new(),
54            anchors: AnchorRegistry::new(),
55            payload_anchors: PayloadAnchorRegistry::new(),
56            movables: MovableIndex::default(),
57            diagnostics: SlotTableDiagnostics::default(),
58            stale_group_indexes: metadata::StaleGroupIndexes::default(),
59            next_group_generation: 1,
60        }
61    }
62
63    pub(crate) fn write_session<'a>(
64        &'a mut self,
65        lifecycle: &'a mut SlotLifecycleCoordinator,
66        state: &'a mut SlotWriteSessionState,
67    ) -> SlotWriteSession<'a> {
68        SlotWriteSession {
69            table: self,
70            lifecycle,
71            state,
72        }
73    }
74
75    pub(crate) fn storage_id(&self) -> usize {
76        self.storage_id.id()
77    }
78
79    #[cfg(test)]
80    pub(in crate::slot) fn set_next_group_generation_for_test(&mut self, generation: u32) {
81        self.next_group_generation = generation;
82    }
83
84    #[cfg(test)]
85    pub(in crate::slot) fn set_next_group_anchor_for_test(&mut self, next_anchor: usize) {
86        self.anchors.set_next_anchor_for_test(next_anchor);
87    }
88
89    #[cfg(test)]
90    pub(in crate::slot) fn set_next_payload_anchor_for_test(&mut self, next_anchor: usize) {
91        self.payload_anchors.set_next_id_for_test(next_anchor);
92    }
93
94    pub(crate) fn compact_storage(&mut self) {
95        self.groups.shrink_to_fit();
96        self.payloads.compact();
97        self.nodes.shrink_to_fit();
98        self.anchors.shrink_to_fit();
99        self.payload_anchors.shrink_to_fit();
100        self.movables.shrink_to_fit();
101    }
102
103    pub(crate) fn storage_capacity(&self) -> usize {
104        self.groups.capacity() + self.payloads.capacity() + self.nodes.capacity()
105    }
106
107    pub(crate) fn trim_growth_slack(&mut self) {
108        self.groups.trim_growth_slack();
109        self.payloads.trim_growth_slack();
110        self.nodes.trim_growth_slack();
111        self.anchors.trim_growth_slack();
112        self.payload_anchors.trim_growth_slack();
113    }
114
115    pub(crate) fn take_effect_drops(&mut self) -> Vec<DeferredDrop> {
116        let mut drops = Vec::new();
117        self.payloads.for_each_mut(|payload| {
118            if let Some(fresh) = payload.payload_type.fresh {
119                let old = std::mem::replace(&mut payload.value, fresh());
120                drops.push(DeferredDrop::payload(old));
121            }
122        });
123        drops
124    }
125
126    pub(crate) fn take_all_drops(&mut self) -> Vec<DeferredDrop> {
127        let mut drops = Vec::with_capacity(self.payloads.len());
128        self.payloads
129            .drain_rev(|payload| drops.push(payload.into_deferred_drop()));
130        self.groups.clear();
131        self.stale_group_indexes = metadata::StaleGroupIndexes::default();
132        self.nodes.clear();
133        self.anchors.clear();
134        self.payload_anchors.clear();
135        self.movables.clear();
136        drops
137    }
138}
139
140impl Default for SlotTable {
141    fn default() -> Self {
142        Self::new()
143    }
144}
cranpose_core/slot/table.rs

cranpose_core/slot/
table.rs
