Как эффективно искать и вставлять в хэш-карту?

Я хотел бы сделать следующее:

  • Поиск a 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?

1 60
name-lookup hashmap rust

1 ответ:

The 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)
76