Jak efektywnie wyszukiwać i wstawiać do HashMap?

102

Chciałbym wykonać następujące czynności:

  • Wyszukaj Vecokreślony klucz i zapisz go do późniejszego użytku.
  • Jeśli nie istnieje, utwórz pusty Vecdla klucza, ale nadal przechowuj go w zmiennej.

Jak to zrobić skutecznie? Oczywiście pomyślałem, że przydałoby się 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
    }
};

Kiedy go wypróbowałem, dało mi to błędy takie jak:

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

Skończyło się na tym, że zrobiłem coś takiego, ale nie podoba mi się to, że wykonuje wyszukiwanie dwukrotnie ( map.contains_keyi 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!");
    }
};

Czy istnieje bezpieczny sposób na zrobienie tego tylko z jednym match?

Yusuke Shinyama
źródło

Odpowiedzi:

120

entryAPI jest przeznaczony do tego celu. W formie ręcznej może to wyglądać

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)
};

Lub można skorzystać z krótszej formy:

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

Jeśli defaultjest OK / tani do obliczenia, nawet jeśli nie jest wstawiony, może to być również:

map.entry(key).or_insert(default)
huon
źródło
Dzięki za szybką odpowiedź! Teraz dowiedziałem się, że powinienem zagłębić się w dokumenty.
Yusuke Shinyama
22
problem z entry () polega na tym, że zawsze musisz sklonować klucz, czy jest sposób, aby tego uniknąć?
Pascalius
@Pascalius możesz ustawić swój typ klucza &T(jeśli klucze przeżywają mapę, np. Ciągi statyczne) lub Rc<T>zamiast T- ale nie jest to ładne w obu przypadkach
kbolino
@Pascalius: możesz użyć v.key()w wyrażeniu for default, a wtedy otrzyma odniesienie do klucza istniejącego w hashmap, więc możesz w ten sposób uniknąć klona
Chris Beck,