Rust中从HashMap获取已有值或创建新值的优雅实现方式

你的问题其实Rust标准库的HashMap已经为我们准备了完美的解决方案——entry API，不仅能彻底避免提前声明冗余变量的繁琐，还能优化性能、解决生命周期问题，同时让代码逻辑更清晰。

先聊聊你遇到的问题根源：

• 最初的代码里，match的None分支创建的Test是局部变量，一旦离开分支就会被销毁，返回的引用自然就成了"悬垂引用"，这就是编译器抛出生命周期错误的原因。
• 你临时用提前声明变量的方法虽然绕开了生命周期问题，但不仅代码不够简洁，还有个隐藏的逻辑漏洞：新创建的Test并没有被插入到HashMap中，后续无法再通过这个key获取到它，这大概率不是你想要的结果。

最优实现：用entry API完成"获取或插入" #

Rust的HashMap::entry方法专门为这种"检查存在性，不存在则插入"的场景设计，它只做一次哈希查找（手动get+insert会做两次），而且能自动处理所有权和生命周期。

针对你的需求，完整的优雅代码如下：

use std::collections::HashMap;

struct Test { name: String, valid: bool }

fn main() {
    let mut tests: HashMap<String, Test> = HashMap::new();
    let key = String::from("a");
    // 先克隆key：因为HashMap会拿走key的所有权，而我们需要用它初始化Test的name
    let key_clone = key.clone();

    // entry(key)会尝试获取对应key的条目，or_insert_with会在条目不存在时执行闭包创建新值并插入
    let test: &Test = tests.entry(key).or_insert_with(|| {
        Test {
            name: key_clone,
            valid: false
        }
    });

    // 现在test就是对应key的引用，不管是原本存在的还是新创建的
    println!("Test name: {}, valid: {}", test.name, test.valid);
}

为什么这更优？ #

• 无冗余变量：不需要提前声明额外的t变量，逻辑连贯一气呵成。
• 性能更好：只执行一次哈希查找，比手动get后再insert的两次查找效率更高。
• 生命周期安全：编译器能正确推导引用的生命周期，不会出现悬垂引用问题。
• 语义清晰：entry+or_insert_with的组合直接表达了"获取已有值，没有则创建插入"的意图，可读性拉满。

额外场景补充 #

如果你的Test实例不需要用原key初始化name，代码会更简洁：

let test = tests.entry(key).or_insert_with(|| {
    Test {
        name: String::from("default_test"),
        valid: false
    }
});

要是你确实不需要把新创建的Test插入到HashMap（这种场景很少见），那不能返回引用（会悬垂），可以直接返回实例本身（需要Test实现Clone）：

let test: Test = match tests.get(&key) {
    Some(existing) => existing.clone(),
    None => Test { name: key, valid: false }
};

内容的提问来源于stack exchange，提问作者Horowitzathome