o3 0.26.0

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

use crate::{
    buffer::{
        segments,
        segments::{
            Chain, Key, Pool,
            chain::{edit, edit::join},
            handles,
        },
    },
    depot,
    depot::{
        arena::linked::{fifo::ends, slots},
        capacity,
    },
};

pub(in crate::buffer::segments::chain::edit::join) struct Incoming<'a, T> {
    pub(in crate::buffer::segments::chain::edit::join) source: &'a mut Chain<T, Pool<T>>,
    pub(in crate::buffer::segments::chain::edit::join) input: join::Input,
}

struct Drain<'a, T> {
    store: &'a slots::Store<segments::Item<T>>,
    key: &'a mut Key,
    links: ends::Exclusive<slots::Index>,
    remaining: capacity::Count,
}

// SAFETY: the input variant and count belong to this exclusively borrowed source.
unsafe impl<T> depot::raw::Input<segments::Item<T>> for Incoming<'_, T> {
    type Item = segments::Item<T>;
    type Error = edit::Cause;
    type Context = ();
    type Output = ();

    fn count(&self) -> capacity::Count {
        capacity::Count::new(match self.input {
            join::Input::Single(_) => 1,
            join::Input::Multi { count, .. } => count.get(),
        })
    }

    fn reject(self, error: capacity::Shortage) -> Self::Error {
        edit::Cause::Nodes(error)
    }
}

// SAFETY: the exclusively retained input already owns every node to detach.
unsafe impl<T> depot::raw::Start<segments::Item<T>> for Incoming<'_, T> {
    fn start(
        self,
    ) -> (
        impl depot::raw::Exact<Item = segments::Item<T>, Output = ()>,
        (),
    ) {
        let store = handles::nodes(&self.source.handle);
        let (links, count) = match self.input {
            join::Input::Single(index) => {
                let mut links = ends::Exclusive::empty();
                unsafe { links.append_end(store, index) };
                (links, 1)
            }
            join::Input::Multi { index, count } => (
                unsafe { handles::descriptors(&self.source.handle).detached(index) }
                    .take()
                    .0
                    .links,
                count.get(),
            ),
        };
        (
            Drain {
                store,
                key: &mut self.source.key,
                links,
                remaining: capacity::Count::new(count),
            },
            (),
        )
    }
}

impl<T> Iterator for Drain<'_, T> {
    type Item = segments::Item<T>;

    fn next(&mut self) -> Option<segments::Item<T>> {
        // SAFETY: admission transferred this exact source partition into the drain.
        let entry = unsafe { slots::Front::new(self.store, &mut self.links) }?;
        self.remaining = capacity::Count::new(self.remaining.raw().wrapping_sub(1));
        Some(entry.pop())
    }
}

// SAFETY: the links and count are detached together from the admitted source.
unsafe impl<T> depot::raw::Exact for Drain<'_, T> {
    type Output = ();
    fn remaining(&self) -> capacity::Count {
        self.remaining
    }
    unsafe fn finish(self) {
        *self.key = Key::EMPTY;
    }
}