Rust中如何在闭包中无克隆消费HashMap内的Vector？

首先明确：你的需求完全合理！既然已经保证bucket后续不会再被访问，用drain转移元素所有权来避免不必要的clone，无论是从性能还是内存效率上看都是非常正确的选择，完全没必要为了迁就编译器而保留冗余的拷贝操作。

为什么会出现编译错误？ #

你遇到的错误核心是Rust的借用检查器在阻止可变引用逃逸出FnMut闭包。flat_map返回的迭代器会持有对bucket的可变引用（来自get_mut），而这个迭代器会在闭包执行结束后被cluster.extend使用——相当于把bucket的可变引用从闭包里“带了出去”。

Rust对FnMut闭包有严格限制：它允许闭包多次调用，但不能让捕获的可变引用逃逸，否则后续调用闭包时可能会产生悬垂引用或者数据竞争，这是内存安全的红线。

不需要unsafe，调整逻辑即可解决问题 #

我们可以换一种方式处理，避免让引用逃逸。核心思路是：先把cluster里的所有键收集起来，再逐个从bucket中drain对应的值并扩展到cluster中。修改后的代码如下：

use std::collections::HashMap;
fn main() {
    let mut bucket: HashMap<&str, Vec<&str>> = HashMap::new();
    bucket.insert("a", vec!["hello", "good morning"]);
    bucket.insert("b", vec!["bye", "ciao"]);
    bucket.insert("c", vec!["good"]);
    let cluster = vec!["a", "b"];
    let cluster2 = vec!["c"];
    let mut clusters = [cluster, cluster2];

    clusters.iter_mut().for_each(|cluster| {
        // 先把cluster中的所有键取出来，避免迭代时修改cluster本身
        let keys: Vec<_> = cluster.drain(..).collect();
        // 遍历每个键，从bucket中drain对应的值并扩展到cluster
        for key in keys {
            if let Some(values) = bucket.get_mut(key) {
                cluster.extend(values.drain(..));
            }
        }
    });

    println!("{:?}", clusters);
}

代码解释 #

1. cluster.drain(..).collect()：把cluster中原有的所有键转移到keys向量中，这样我们就可以安全地遍历keys，同时修改原cluster（因为原cluster此时是空的）。
2. 逐个处理键：对每个键，用bucket.get_mut(key)获取对应的可变向量，然后调用drain(..)把向量中的所有元素转移出来，通过cluster.extend添加到cluster中。
3. 整个过程完全符合Rust的借用规则：没有同时持有多个可变引用，也没有让引用逃逸出闭包的生命周期，完全不需要unsafe代码。

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