trait-map-derive 0.1.2

Derive the TraitMapEntry trait for trait_map crate
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

use trait_map::{TraitMap, TraitMapEntry};

trait MyTraitOne {
  fn test_one(&self) -> u32;
}

trait MyTraitTwo {
  fn test_two(&self) -> u32;
}

trait MyTraitThree {
  fn test_three(&self) -> u32;
}

#[derive(TraitMapEntry)]
#[trait_map(MyTraitOne)]
#[trait_map(MyTraitTwo, MyTraitThree)]
struct MyStruct {
  value: u32,
}

impl MyTraitOne for MyStruct {
  fn test_one(&self) -> u32 {
    self.value
  }
}

impl MyTraitTwo for MyStruct {
  fn test_two(&self) -> u32 {
    self.value * self.value
  }
}

impl MyTraitThree for MyStruct {
  fn test_three(&self) -> u32 {
    self.value / 2
  }
}

fn main() {
  let mut trait_map = TraitMap::new();
  trait_map.add_entry(MyStruct { value: 5 });

  let entries = trait_map.get_entries::<dyn MyTraitOne>();
  assert_eq!(entries.len(), 1);
  assert_eq!(entries.values().next().unwrap().test_one(), 5);

  let entries = trait_map.get_entries::<dyn MyTraitTwo>();
  assert_eq!(entries.len(), 1);
  assert_eq!(entries.values().next().unwrap().test_two(), 25);

  let entries = trait_map.get_entries::<dyn MyTraitThree>();
  assert_eq!(entries.len(), 1);
  assert_eq!(entries.values().next().unwrap().test_three(), 2);
}