[meta] Add a MapWithDefault trait;

This traits augments HashMap so they have a `get_or_default` method that
can be used to avoid boilerplate around usage of `entry.or_default`.
This commit is contained in:
Benjamin Bouvier
2019-06-21 12:08:43 +02:00
parent 18fb87cd9c
commit 9dcc185264
2 changed files with 20 additions and 0 deletions

View File

@@ -0,0 +1,19 @@
use std::collections::HashMap;
use std::hash::Hash;
pub trait MapWithDefault<K, V: Default> {
fn get_or_default(&mut self, k: K) -> &mut V;
}
impl<K: Eq + Hash, V: Default> MapWithDefault<K, V> for HashMap<K, V> {
fn get_or_default(&mut self, k: K) -> &mut V {
self.entry(k).or_insert_with(|| V::default())
}
}
#[test]
fn test_default() {
let mut hash_map = HashMap::new();
hash_map.insert(42, "hello");
assert_eq!(*hash_map.get_or_default(43), "");
}

View File

@@ -12,6 +12,7 @@ mod gen_settings;
mod gen_types; mod gen_types;
mod constant_hash; mod constant_hash;
mod default_map;
mod shared; mod shared;
mod unique_table; mod unique_table;