use iddqd::{BiHashItem, BiHashMap, bi_hash_map::Entry, bi_upcast};
use std::path::{Path, PathBuf};
#[derive(Clone, Debug, PartialEq, Eq)]
struct MyStruct {
a: String,
b: usize,
c: PathBuf,
d: Vec<usize>,
}
#[derive(Clone, Debug, Hash, Eq, PartialEq)]
struct MyKey1<'a> {
b: usize,
c: &'a Path,
}
#[derive(Clone, Debug, Hash, Eq, PartialEq)]
struct MyKey2<'a> {
c: &'a Path,
d: &'a [usize],
}
impl BiHashItem for MyStruct {
type K1<'a> = MyKey1<'a>;
type K2<'a> = MyKey2<'a>;
fn key1(&self) -> Self::K1<'_> {
MyKey1 { b: self.b, c: &self.c }
}
fn key2(&self) -> Self::K2<'_> {
MyKey2 { c: &self.c, d: &self.d }
}
bi_upcast!();
}
fn main() {
let mut map = BiHashMap::new();
let item = MyStruct {
a: "example".to_owned(),
b: 20,
c: PathBuf::from("/"),
d: Vec::new(),
};
map.insert_unique(item.clone()).unwrap();
map.insert_unique(MyStruct {
a: "something-else".to_owned(),
b: 20,
c: PathBuf::from("/"),
d: vec![1],
})
.unwrap_err();
let item2 = MyStruct {
a: "example".to_owned(),
b: 10,
c: PathBuf::from("/home"),
d: Vec::new(),
};
map.insert_unique(item2.clone()).unwrap();
assert_eq!(map.get1(&MyKey1 { b: 20, c: Path::new("/") }), Some(&item));
for item in map.iter() {
println!("{item:?}");
}
let item3 = MyStruct {
a: "example".to_owned(),
b: 20,
c: PathBuf::from("/"),
d: vec![1, 2, 3],
};
let item4 = MyStruct {
a: "example".to_owned(),
b: 20,
c: PathBuf::from("/home"),
d: vec![1, 2, 3],
};
for item in [item, item2, item3, item4] {
let entry = map.entry(item.key1(), item.key2());
match entry {
Entry::Occupied(entry) => {
let item = entry.get();
println!("occupied: {item:?}");
}
Entry::Vacant(entry) => {
let item_ref = entry.insert(item);
println!("inserted: {item_ref:?}");
}
}
}
}