Rust中如何实现索引范围的可变迭代器？解决filter_map捕获变量逃逸的编译错误

这个问题我之前也碰到过，编译器的错误提示看起来有点绕，但核心原因是Rust的借用检查器在严格保护我们，避免出现悬垂引用或者多个可变引用的冲突问题。

为什么会出现这个错误？ #

你用filter_map的时候，闭包捕获了&mut self，而filter_map要求闭包是FnMut类型——这类闭包的生命周期只在每次调用时有效，但你返回的&mut T的生命周期是和self的可变借用绑定的，这就导致引用“逃逸”出了闭包的生命周期范围。编译器没办法确认闭包后续调用会不会再次借用self，进而造成多个可变引用同时存在的情况，所以直接拒绝了这种写法。

安全解决方案：用from_fn手动控制迭代 #

其实不用写unsafe代码就能解决这个问题，我们可以用std::iter::from_fn来手动管理迭代过程，让借用检查器能清晰跟踪每次的可变借用：

pub fn iter_range_mut(&mut self, index: I, size: usize) -> impl Iterator<Item = &mut T> {
    let mut current = Some(index);
    let mut remaining = size;
    std::iter::from_fn(move || {
        if remaining == 0 {
            return None;
        }
        remaining -= 1;
        // 取出当前索引，没有的话直接返回None
        let current_idx = current.take()?;
        // 获取可变引用，失败的话直接返回None
        let item = self.get_mut(current_idx)?;
        // 更新下一个要迭代的索引
        current = current_idx.get_next();
        Some(item)
    })
}

这个方案的工作逻辑很直观：

• 用current跟踪当前的索引位置，remaining记录还需要迭代多少个元素
• 每次from_fn的闭包被调用时，我们先检查是否还有剩余元素要迭代
• 取出当前索引并调用get_mut获取可变引用，之后更新current为下一个索引
• 因为每次闭包调用只会对self进行一次可变借用，且借用的生命周期和self的可变借用完全对齐，借用检查器能确认不会出现引用冲突，所以不会再报错。

可选：unsafe方案（仅在必要时使用） #

如果你坚持想用successors的写法，或者有其他必须用unsafe的场景，也可以通过原始指针绕过借用检查器，但一定要严格保证安全条件：

pub fn iter_range_mut(&mut self, index: I, size: usize) -> impl Iterator<Item = &mut T> {
    // 将&mut self转换成原始指针
    let this = self as *mut Self;
    std::iter::successors(Some(index), |i| i.get_next())
        .take(size)
        // 移动捕获原始指针，在闭包里unsafe解引用
        .filter_map(move |i| unsafe { (*this).get_mut(i) })
}

必须遵守的安全前提：

1. get_mut每次返回的&mut T必须是唯一的、不重叠的，不能有两个迭代元素指向同一个内存位置
2. 迭代器的生命周期绝对不能超过self的可变借用周期，避免出现悬垂引用
3. successors生成的索引序列不会重复访问同一个内存位置，否则会导致未定义行为

因为我们绕过了Rust的借用检查器，这些安全条件必须由你手动保证，一旦违反就会引发内存安全问题。

内容来源于stack exchange