alex/agb
1
0
Fork 0
mirror of https://github.com/italicsjenga/agb.git synced 2024-12-24 00:31:34 +11:00
Code Issues Packages Projects Releases Wiki Activity

Implement insert correctly

Browse source
This commit is contained in:
Gwilym Kuiper 2022-03-21 20:42:07 +00:00
parent 24b61f51fe
commit c7db20c1df
1 changed files with 18 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
agb/src/hash_map.rs
View file

@ -102,7 +102,23 @@ impl<K, V> HashMap<K, V>
where
K: Eq + Hash,
{
pub fn insert(&mut self, key: K, value: V) -> &mut V {
pub fn insert(&mut self, key: K, value: V) -> Option<V> {
let hash = self.hash(&key);
if let Some(location) = self.nodes.get_location(&key, hash) {
Some(self.nodes.replace_at_location(location, key, value))
} else {
if self.nodes.capacity() * 85 / 100 <= self.len() {
self.resize(self.nodes.capacity() * 2);
}
self.nodes.insert_new(key, value, hash);
None
}
}
fn insert_and_get(&mut self, key: K, value: V) -> &'_ mut V {
let hash = self.hash(&key);
let location = if let Some(location) = self.nodes.get_location(&key, hash) {
@ -254,7 +270,7 @@ impl<'a, K: 'a, V: 'a> VacantEntry<'a, K, V> {
where
K: Hash + Eq,
{
self.map.insert(self.key, value)
self.map.insert_and_get(self.key, value)
}
}