Question

我想做以下事情：

查找某个密钥的Vec，并存储该密钥以供日后使用。
如果它不存在，请为该密钥创建一个空的Vec，但仍将其保留在变量中。

如何有效地做到这一点？我当然认为我可以使用match：

use std::collections::HashMap;

// This code doesn't compile.
let mut map = HashMap::new();
let key = "foo";
let values: &Vec<isize> = match map.get(key) {
    Some(v) => v,
    None => {
        let default: Vec<isize> = Vec::new();
        map.insert(key, default);
        &default
    }
};

当我尝试它时，它给了我错误，如：

error[E0502]: cannot borrow `map` as mutable because it is also borrowed as immutable
  --> src/main.rs:11:13
   |
7  |     let values: &Vec<isize> = match map.get(key) {
   |                                     --- immutable borrow occurs here
...
11 |             map.insert(key, default);
   |             ^^^ mutable borrow occurs here
...
15 | }
   | - immutable borrow ends here

我最终做了类似的事情，但我不喜欢它执行两次查询（map.contains_key和map.get）的事实：

// This code does compile.
let mut map = HashMap::new();
let key = "foo";
if !map.contains_key(key) {
    let default: Vec<isize> = Vec::new();
    map.insert(key, default);
}
let values: &Vec<isize> = match map.get(key) {
    Some(v) => v,
    None => {
        panic!("impossiburu!");
    }
};

只用一个match是否可以安全地执行此操作？

Answer 1

entry API专为此而设计。在手动表单中，它可能看起来像

use std::collections::hash_map::Entry;

let values: &Vec<isize> = match map.entry(key) {
    Entry::Occupied(o) => o.into_mut(),
    Entry::Vacant(v) => v.insert(default)
};

或者可以使用简短形式：

map.entry(key).or_insert_with(|| default)

如果default即使没有插入就可以计算好/便宜，它也可以只是：

map.entry(key).or_insert(default)

Answer 2

我使用了@huon 和@Shepmaster 的回答，并将其作为一个特征来实现：

use std::collections::HashMap;
use std::hash::Hash;

pub trait InsertOrGet<K: Eq + Hash, V: Default> {
    fn insert_or_get(&mut self, item: K) -> &mut V;
}

impl<K: Eq + Hash, V: Default> InsertOrGet<K, V> for HashMap<K, V> {
    fn insert_or_get(&mut self, item: K) -> &mut V {
        return match self.entry(item) {
            std::collections::hash_map::Entry::Occupied(o) => o.into_mut(),
            std::collections::hash_map::Entry::Vacant(v) => v.insert(V::default()),
        };
    }
}

然后我可以在其他地方做：

use crate::utils::hashmap::InsertOrGet;

let new_or_existing_value: &mut ValueType = my_map.insert_or_get(my_key.clone());

如何有效地从HashMap中查找和插入？

2 个答案: