o3 0.31.1

shared-nothing primitives
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
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144

use std::marker;

use hash::errors;
use runtime::entry;

use crate::depot::{
    arena::{
        hash::{self, runtime},
        slab::key,
    },
    map,
    queue::links,
};

#[repr(transparent)]
pub struct Store<T, Tag, P> {
    pub(in crate::depot::arena::hash) core: hash::Core<T, Tag>,
    phase: marker::PhantomData<fn(P) -> P>,
}

pub type Map<T, Tag = ()> = Store<T, Tag, entry::Occupied>;

impl<T, Tag, P> Store<T, Tag, P> {
    pub fn try_new(plan: hash::Plan) -> Result<Self, crate::alloc::Error> {
        Ok(Self {
            core: hash::Core::try_new(hash::Parts::try_new(plan.table)?)?,
            phase: marker::PhantomData,
        })
    }

    pub const fn capacity(&self) -> usize {
        self.core.capacity()
    }

    pub const fn is_empty(&self) -> bool {
        self.core.len() == 0
    }

    pub fn get(&self, key: key::Handle<runtime::Domain<Tag>>) -> Option<&T> {
        self.core.get(key)
    }

    pub fn key_value_at(&self, index: usize) -> Option<(key::Handle<runtime::Domain<Tag>>, &T)> {
        self.core.key_value_at(index)
    }

    pub fn key(
        &self,
        hash: u64,
        eq: impl Fn(&T) -> bool,
    ) -> Option<key::Handle<runtime::Domain<Tag>>> {
        map::raw::Probe::find(&self.core, hash, eq)
    }
}

impl<T, Tag> Map<T, Tag> {
    pub const fn len(&self) -> usize {
        self.core.len()
    }

    pub fn key_at(&self, index: usize) -> Option<key::Handle<runtime::Domain<Tag>>> {
        self.core.key_at(index)
    }

    pub fn try_key<E>(
        &self,
        hash: u64,
        eq: impl Fn(&T) -> Result<bool, E>,
    ) -> Result<Option<key::Handle<runtime::Domain<Tag>>>, E> {
        map::raw::Probe::try_find(&self.core, hash, eq)
    }

    pub fn try_entry<E>(
        &mut self,
        hash: u64,
        eq: impl Fn(&T) -> Result<bool, E>,
    ) -> Entry<'_, T, Tag, E> {
        Ok(
            match map::raw::Probe::try_new(&mut self.core, hash, eq)?.entry() {
                map::Entry::Occupied(entry) => map::Entry::Occupied(entry.with_phase()),
                map::Entry::Vacant(entry) => map::Entry::Vacant(entry),
            },
        )
    }

    pub fn values(&self) -> impl Iterator<Item = &T> {
        self.core.values()
    }

    pub fn try_insert(
        &mut self,
        hash: u64,
        value: T,
        eq: impl Fn(&T, &T) -> bool,
    ) -> Result<key::Handle<runtime::Domain<Tag>>, errors::Insert<T, Tag>> {
        let (place, value) = entry::Vacancy::try_new(&mut self.core, hash, value, eq)?;
        Ok(place.insert(value))
    }

    pub fn try_create(
        &mut self,
        hash: u64,
        eq: impl Fn(&T) -> bool,
        make: impl FnOnce(links::Index) -> T,
    ) -> Result<key::Handle<runtime::Domain<Tag>>, errors::Create<Tag>> {
        Ok(self.try_live(hash, eq, make)?.key())
    }

    /// Creates one resident and retains its occupied source loan without a key lookup.
    pub fn try_live(
        &mut self,
        hash: u64,
        eq: impl Fn(&T) -> bool,
        make: impl FnOnce(links::Index) -> T,
    ) -> Result<entry::Live<'_, T, Tag, entry::Occupied>, errors::Create<Tag>> {
        let capacity = self.capacity();
        match map::raw::Probe::new(&mut self.core, hash, eq).entry() {
            map::Entry::Occupied(entry) => Err(errors::Create::Occupied { key: entry.key() }),
            map::Entry::Vacant(None) => Err(errors::Create::Full { capacity }),
            map::Entry::Vacant(Some(entry)) => {
                let value = make(entry.index());
                Ok(entry.insert_entry(value).with_phase())
            }
        }
    }

    pub fn get_mut(&mut self, key: key::Handle<runtime::Domain<Tag>>) -> Option<&mut T> {
        Some(self.core.entry(key)?.into_mut())
    }

    pub fn remove(&mut self, key: key::Handle<runtime::Domain<Tag>>) -> Option<T> {
        Some(
            self.core
                .entry(key)?
                .with_phase::<entry::Occupied>()
                .remove(),
        )
    }
}

type Entry<'a, T, Tag, E> = Result<
    map::Entry<entry::Live<'a, T, Tag, entry::Occupied>, Option<entry::Vacancy<'a, T, Tag>>>,
    E,
>;