summaryrefslogtreecommitdiffstats
path: root/vendor/elsa/examples/mutable_arena.rs
blob: d5db2d3314ecfb49955c3ebb858706d56e512389 (plain)
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
use elsa::FrozenVec;

fn main() {
    let arena = Arena::new();
    let lonely = arena.add_person("lonely", vec![]);
    let best_friend = arena.add_person("best friend", vec![lonely]);
    let threes_a_crowd = arena.add_person("threes a crowd", vec![lonely, best_friend]);
    let rando = arena.add_person("rando", vec![]);
    let _everyone = arena.add_person(
        "follows everyone",
        vec![rando, threes_a_crowd, lonely, best_friend],
    );
    arena.dump();
}

struct Arena<'arena> {
    people: FrozenVec<Box<Person<'arena>>>,
}

struct Person<'arena> {
    pub follows: FrozenVec<PersonRef<'arena>>,
    pub reverse_follows: FrozenVec<PersonRef<'arena>>,
    pub name: &'static str,
}

type PersonRef<'arena> = &'arena Person<'arena>;

impl<'arena> Arena<'arena> {
    fn new() -> Arena<'arena> {
        Arena {
            people: FrozenVec::new(),
        }
    }

    fn add_person(
        &'arena self,
        name: &'static str,
        follows: Vec<PersonRef<'arena>>,
    ) -> PersonRef<'arena> {
        let idx = self.people.len();
        self.people.push(Box::new(Person {
            name,
            follows: follows.into(),
            reverse_follows: Default::default(),
        }));
        let me = &self.people[idx];
        for friend in &me.follows {
            friend.reverse_follows.push(me)
        }
        me
    }

    fn dump(&'arena self) {
        for thing in &self.people {
            println!("{} network:", thing.name);
            println!("\tfollowing:");
            for friend in &thing.follows {
                println!("\t\t{}", friend.name);
            }
            println!("\tfollowers:");
            for friend in &thing.reverse_follows {
                println!("\t\t{}", friend.name);
            }
        }
    }
}

// Note that the following will cause the above code to stop compiling
// since non-eyepatched custom destructors can potentially
// read deallocated data.
//
// impl<'arena> Drop for Person<'arena> {
//     fn drop(&mut self) {
//         println!("goodbye {:?}", self.name);
//         for friend in &self.follows {
//             println!("\t\t{}", friend.name);
//         }
//     }
// }