Rust HashMap中的默认可变值

发布于06月30日

假设我有一个HashMap，我想得到一个条目的可变引用，或者如果这个条目不存在，我想得到一个新对象的可变引用，我该怎么做呢？我试过用unwrap_or()，类似这样的:

fn foo() {
    let mut map: HashMap<&str, Vec<&str>> = HashMap::new();

    let mut ref = map.get_mut("whatever").unwrap_or( &mut Vec::<&str>::new() );

    // Modify ref.
}

但这是行不通的，因为Vec人的生命周期不够长.有没有办法告诉Rust我希望返回的Vec和foo()的生命周期相同？我的意思是有一个显而易见的解决方案，但我觉得应该有更好的方法:

fn foo() {
    let mut map: HashMap<&str, Vec<&str>> = HashMap::new();

    let mut dummy: Vec<&str> = Vec::new();
    let mut ref = map.get_mut("whatever").unwrap_or( &dummy );

    // Modify ref.
}

推荐答案

正如Shepmaster所提到的，下面是一个使用输入模式的示例.一开始看起来很冗长，但这避免了分配一个除非需要否则可能不会使用的array.我相信你可以做一个通用函数来减少聊天:)

use std::collections::HashMap;
use std::collections::hash_map::Entry::{Occupied, Vacant};

fn foo() {
    let mut map = HashMap::<&str, Vec<&str>>::new();
    let mut result = match map.entry("whatever") {
       Vacant(entry) => entry.insert(Vec::new()),
       Occupied(entry) => entry.into_mut(),
    };

    // Do the work
    result.push("One thing");
    result.push("Then another");
}

正如我刚刚发现的，这也可以缩短到or_insert！

use std::collections::HashMap;

fn foo() {
    let mut map = HashMap::<&str, Vec<&str>>::new();
    let mut result = map.entry("whatever").or_insert(Vec::new());

    // Do the work
    result.push("One thing");
    result.push("Then another");
}