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
use std::sync::Arc;
use std::ops::{Deref, DerefMut};
use parking_lot::{RwLock, RwLockReadGuard, RwLockWriteGuard};
use arena::Arena;
use super::NodeEntry;
#[derive(Debug)]
pub struct NodeArena<T> {
inner: Arc<RwLock<Arena<NodeEntry<T>>>>,
}
impl<T> NodeArena<T> {
pub fn new() -> NodeArena<T> {
NodeArena {
inner: Arc::new(RwLock::new(Arena::new())),
}
}
pub fn with_capacity(capacity: usize) -> NodeArena<T> {
NodeArena {
inner: Arc::new(RwLock::new(Arena::with_capacity(capacity))),
}
}
pub fn into_inner(self) -> Arena<NodeEntry<T>> {
if let Ok(inner) = Arc::try_unwrap(self.inner) {
inner.into_inner()
} else {
panic!("NodeArena::into_inner called while having more than one strong reference");
}
}
pub fn ptr_eq(&self, other: &NodeArena<T>) -> bool {
Arc::ptr_eq(&self.inner, &other.inner)
}
pub fn read<'a>(&'a self) -> impl Deref<Target = Arena<NodeEntry<T>>> + 'a {
self.inner.read()
}
pub fn write<'a>(&'a self) -> impl DerefMut<Target = Arena<NodeEntry<T>>> + 'a {
self.inner.write()
}
pub(crate) fn read_guard<'a>(&'a self) -> RwLockReadGuard<'a, Arena<NodeEntry<T>>> {
self.inner.read()
}
pub(crate) fn write_guard<'a>(&'a self) -> RwLockWriteGuard<'a, Arena<NodeEntry<T>>> {
self.inner.write()
}
}
impl<T> Clone for NodeArena<T> {
fn clone(&self) -> NodeArena<T> {
NodeArena {
inner: self.inner.clone(),
}
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn ptr_eq() {
let arena = NodeArena::<i32>::new();
let copy = arena.clone();
assert!(arena.ptr_eq(©));
}
#[test]
fn read() {
let arena = NodeArena::<i32>::with_capacity(1024);
assert_eq!(arena.read().capacity(), 1024);
}
#[test]
fn into_inner() {
let arena = NodeArena::<i32>::new();
assert_eq!(arena.into_inner().capacity(), 0);
}
#[test]
fn write() {
let arena = NodeArena::<i32>::with_capacity(1024);
arena.write().clear();
}
}