我想做以下几件事:

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

当我try 时,它会给我以下错误:

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_keymap.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

推荐答案

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)

Rust相关问答推荐

在Rust中,在实现特征`Display`时,如何获取调用方指定的格式?

有没有办法避免在While循环中多次borrow `*分支`

铁 rust 中的共享对象实现特征

当对VEC;U8>;使用serde_json时,Base64编码是保护空间的好方法吗?

Rust 的多态现象.AsRef与Derf

将serde_json读入`VEC<;T&>;`( rust 色)时出现问题

在Rust 中移动原始指针的靶子安全吗

为什么`str`类型可以是任意大小(未知大小),而`string`类型的大小应该是已知的?

在运行特定测试时,如何 suppress cargo test 的空输出?

为什么HashMap::get和HashMap::entry使用不同类型的密钥?

使用 pyo3 将 Rust 转换为 Python 自定义类型

Rust 并行获取对 ndarray 的每个元素的可变引用

将引用移动到线程中

为什么这个闭包没有实现Fn?

如何基于常量在Rust中跳过一个测试

实现AsyncWrite到hyper Sender时发生生命周期错误

Rust/Serde/HTTP:序列化`Option`

用逗号分隔字符串,但在标记中使用逗号

有没有办法阻止 rust-analyzer 使非活动代码变暗?

有没有办法隐藏类型定义?