在 Rust 中更新 HashMap 的值：从新手到专家

如何在 Rust 中更新可变 HashMap 的值：从新手到专家

您是否曾遇到过这样的情况：需要在 Rust 中更新 HashMap 的值，但编译器却固执地拒绝您的代码？如果是这样——您并不孤单。Stack Overflow 上关于 "How can I update a value in a mutable HashMap?" 的问题已获得数千次浏览，这意味着即使是经验丰富的开发人员有时也会在操作此集合的细节上感到困惑。

在本文中，我们将探讨在 Rust 中更新 HashMap 值的所有方法：从基本的 insert 到高级的 entry().and_modify()。您将学习如何正确地修改数据、避免常见错误以及编写地道的代码。开始吧！

为什么在 HashMap 中更新不仅仅是“拿来就改”？

与许多语言不同，Rust 要求您显式处理所有权和借用。当您想要更新 HashMap 中的值时，需要解决三个问题：

\n
• 检查键是否存在——是否已有该值？
\n
• 修改值——如果键存在，则更新它
\n
• 插入新值——如果键不存在，则添加它
\n

听起来很简单？让我们看看代码。

方法 1：经典的 insert（但有陷阱）

最直接的方法是使用 insert。如果键存在，它会覆盖值；如果键不存在，它会插入一个新的键值对。

use std::collections::HashMap;\n\nlet mut scores = HashMap::new();\nscores.insert(String::from("Alice"), 10);\nscores.insert(String::from("Bob"), 20);\n\n// 更新 Alice 的值\nscores.insert(String::from("Alice"), 15);\n\nprintln!("{:?}", scores); // {"Alice": 15, "Bob": 20}

有效吗？是的。但有一个细节：insert 返回 Option<V>——旧值（如果存在）。如果您想保留之前的值，这很有用。但是，如果您需要基于旧值更新值呢？例如，将分数增加 1？

// 想要将 Alice 的分数增加 1\nif let Some(score) = scores.get_mut("Alice") {\n    *score += 1;\n}

这有效，但需要两个操作：检查存在性和修改。如果键不存在呢？您必须单独进行插入。不太优雅。

方法 2：Entry API——更新之王

Rust 有一个强大的工具——Entry API。它允许您在一个表达式中检查键是否存在、更新或插入值。这是操作 HashMap 的地道方式。

use std::collections::HashMap;\n\nlet mut word_count = HashMap::new();\n\n// 统计句子中的单词\nfor word in "hello world hello Rust".split_whitespace() {\n    let count = word_count.entry(word.to_string()).or_insert(0);\n    *count += 1;\n}\n\nprintln!("{:?}", word_count);\n// {"hello": 2, "world": 1, "Rust": 1}

让我们解析一下魔法：

\n
• entry(key)——为键创建一个条目
\n
• or_insert(0)——如果键不存在，插入值 0 并返回一个可变引用
\n
• *count += 1——解引用并增加值
\n

瞧！一行代码就解决了问题。但这还不是全部。

方法 3：and_modify——当只需要更新时

如果您只想在键存在时更新值，而如果不存在则什么都不做呢？或者插入另一个值？为此，可以在链中使用 and_modify 和 or_insert。

use std::collections::HashMap;\n\nlet mut inventory = HashMap::new();\ninventory.insert("apples".to_string(), 10);\ninventory.insert("bananas".to_string(), 5);\n\n// 如果苹果存在，数量增加 1\ninventory.entry("apples".to_string())\n    .and_modify(|