尝试实现查找或插入时 HashMap 借用问题

我尝试实现自己的类似物find_or_insert方法看起来像这样：

use std::collections::HashMap;

pub struct SomeManager {
    next: i32,
    types: HashMap<i32, i32>,
}

impl SomeManager {
    pub fn get_type<'a>(&'a mut self, k: i32) -> &'a i32 {
        match self.types.get(&k) {
            Some(ref x) => return *x,
            None => {
                self.types.insert(k, self.next);
                self.next += 1;
                return self.types.get(&k).unwrap();
            }
        }
    }
}

fn main() {}

Error:

error[E0502]: cannot borrow `self.types` as mutable because it is also borrowed as immutable
  --> src/main.rs:13:17
   |
10 |         match self.types.get(&k) {
   |               ---------- immutable borrow occurs here
...
13 |                 self.types.insert(k, self.next);
   |                 ^^^^^^^^^^ mutable borrow occurs here
...
18 |     }
   |     - immutable borrow ends here

我知道有一些标准方法可以实现此功能，但我希望此方法尽可能轻 - 它将被非常频繁地调用，并且几乎所有时间值都已经存在。

据我了解，当我们打电话时self.types.get我们借用它的范围match声明，所以我们无法调用self.types.insert这里。我尝试过将方法从None分支出match声明，但它也失败了。

我发现的唯一可行的解​​决方案需要调用get twice:

pub fn get_type<'a>(&'a mut self, k: i32) -> &'a i32 {
    let is_none = match self.types.get(&k) {
        Some(ref x) => false,
        None => true,
    };
    if is_none {
        self.types.insert(k, self.next);
        self.next += 1;
    }
    self.types.get(&k).unwrap()
}

我该如何解决这种情况？

有以下几种方法HashMap来实现这些复杂的情况。最值得注意的是，对于你的情况，HashMap::entry https://doc.rust-lang.org/std/collections/struct.HashMap.html#method.entry and Entry::or_insert_with https://doc.rust-lang.org/std/collections/hash_map/enum.Entry.html#method.or_insert_with:

pub fn get_type<'a>(&'a mut self, k: i32) -> &'a i32 {
    self.types.entry(k).or_insert_with(|| {
        let value = self.next;
        self.next += 1;
        value
    })
}

然而，就你而言，有借用self里面，所以不行。

因此，我们将借用self.next在闭包之外，因此编译器可以将其推断为与self.types。问题应该只通过一次查找就解决了。

pub fn get_type<'a>(&'a mut self, k: i32) -> &'a i32 {
    let next = &mut self.next;

    self.types.entry(k).or_insert_with(|| {
        let value = *next;
        *next += 1;
        value
    })
}