Rust：如何将条目 API 与拥有的数据结合起来？

Question

我有一个 HashMap 并且想更新一个值（如果存在），否则添加一个默认值。通常我会这样做：

some_map.entry(some_key)
    .and_modify(|e| modify(e))
    .or_insert(default)

但是现在我的 modify 有类型 fn(T)->T，但是借用检查器显然不允许我写：

some_map.entry(some_key)
    .and_modify(|e| *e = modify(*e))
    .or_insert(default)

在 Rust 中执行此操作的首选方法是什么？我应该只使用 remove 和 insert 吗？

Answer 1

假设您可以廉价创建 T 的空版本，您可以使用 mem::replace:

some_map.entry(some_key)
    .and_modify(|e| {
        // swaps the second parameter in and returns the value which was there
        let mut v = mem::replace(e, T::empty());
        v = modify(v);
        // puts the value back in and discards the empty one
        mem::replace(e, v);
    })
    .or_insert(default)

这假定 modify 不会恐慌，否则您会发现自己的地图中保留着“空”值。但是 remove / insert.

也会有类似的问题

Rust：如何将条目 API 与拥有的数据结合起来？

Rust: How to combine the Entry API with owned data?

hashmap

ownership

rust