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
use {
crate::escape::{Escape, Handle, Terminal},
smallvec::SmallVec,
std::collections::VecDeque,
};
#[doc(hidden)]
#[derive(Clone, Debug)]
pub struct Epochs {
pub values: SmallVec<[SmallVec<[u64; 8]>; 4]>,
}
impl Epochs {
fn is_before(&self, other: &Self) -> bool {
debug_assert_eq!(self.values.len(), other.values.len());
self.values.iter().zip(other.values.iter()).all(|(a, b)| {
debug_assert_eq!(a.len(), b.len());
a.iter().zip(b.iter()).all(|(a, b)| a < b)
})
}
}
#[derive(Debug, derivative::Derivative)]
#[derivative(Default(bound = ""))]
pub struct ResourceTracker<T> {
terminal: Terminal<T>,
dropped: VecDeque<(Epochs, T)>,
}
impl<T> ResourceTracker<T> {
pub fn new() -> Self {
Self::default()
}
pub fn escape(&self, resource: T) -> Escape<T>
where
T: Sized,
{
Escape::escape(resource, &self.terminal)
}
pub fn handle(&self, resource: T) -> Handle<T>
where
T: Sized,
{
self.escape(resource).into()
}
pub fn cleanup(&mut self, mut dispose: impl FnMut(T), next: &Epochs, complete: &Epochs) {
while let Some((epoch, resource)) = self.dropped.pop_front() {
if !epoch.is_before(complete) {
self.dropped.push_front((epoch, resource));
break;
}
dispose(resource);
}
self.dropped
.extend(self.terminal.drain().map(|res| (next.clone(), res)));
}
pub fn dispose(&mut self, dispose: impl FnMut(T)) {
self.dropped
.drain(..)
.map(|(_, res)| res)
.chain(self.terminal.drain())
.for_each(dispose);
}
}