Rust中peek()导致迭代器元素被消耗的场景及问题咨询

问题原因 #

你代码里的问题出在每次循环都创建了一个新的Peekable包装器：

match i1.peekable().peek() { ... }

Peekable的工作逻辑是：调用peek()时，它会从底层迭代器取出一个元素缓存起来，后续peek()可直接返回缓存值，next()则优先返回缓存元素再取新值。但你这里每次创建的Peekable都是临时对象，调用peek()后就被销毁，缓存的元素（比如第一次的0）不会放回原迭代器i1。当你接着调用i1.next()时，原迭代器已经被临时Peekable取走了当前元素，只能返回下一个（1），这就造成了“peek消耗元素”的假象。

解决方案 #

核心是只创建一次Peekable，而非每次循环临时包装。你的迭代器是impl Iterator且不可复制，直接将原迭代器转换成Peekable即可，后续所有操作基于这个实例：

修改后的代码：

fn main() {
    let v = vec![0, 1, 2, 3, 4, 5];
    let mut iter = v.into_iter().peekable();
    loop {
        match iter.peek() {
            Some(p) => println!("peek {}", p),
            None => println!("peek None"),
        }

        if let Some(i) = iter.next() {
            println!("next: {}", i);
        } else {
            break;
        }
    }
}

运行结果：

peek 0
next: 0
peek 1
next: 1
peek 2
next: 2
peek 3
next: 3
peek 4
next: 4
peek 5
next: 5
peek None

这样peek()只会查看当前元素，不会消耗原迭代器元素——缓存的元素由Peekable自行管理，后续next()会优先返回缓存值，完全符合预期。

补充说明 #

如果你的迭代器是以&mut impl Iterator传递的，无法直接转换为Peekable（会拿走所有权），可以考虑将Peekable作为包装类型持有，或者调整函数参数类型为impl Iterator（若场景允许），就能直接转换为Peekable操作。

内容的提问来源于stack exchange，提问作者compilation-error