如何为Rust迭代器实现peek_while方法且不消耗未匹配元素？

现有代码的问题 #

第一版死循环的原因 #

你第一版实现里的while循环完全是冗余的：迭代器的next方法每次只需要返回单个元素，消费者（比如collect）会自动重复调用next直到返回None。虽然这版逻辑本身不会直接导致死循环，但多余的判断逻辑可能和业务侧的谓词逻辑产生意外交互，同时开头不必要的let n = self.iter.peek()?;也属于冗余代码。

第二版类型错误的原因 #

Peekable::peek()方法返回的是Option<&I::Item>类型，是对元素的只读引用，而你定义的迭代器关联类型Item = I::Item要求返回元素所有权，直接返回Some(n)相当于返回引用，自然会报类型不匹配错误。符合条件时你需要调用self.iter.next()取出元素的所有权再返回。

正确实现 #

核心迭代器逻辑调整 #

只需要修改PeekWhile的next方法即可：

impl<I, P> Iterator for PeekWhile<I, P>
where
    I: Iterator,
    P: FnMut(&I::Item) -> bool,
{
    type Item = I::Item;
    fn next(&mut self) -> Option<I::Item> {
        // 先偷看元素判断是否符合条件，不消耗元素
        if self.iter.peek().map_or(false, |item| (self.predicate)(item)) {
            // 符合条件才消费该元素返回
            self.iter.next()
        } else {
            // 不符合直接返回None，不消耗元素
            None
        }
    }
}

用法调整说明 #

你给出的示例代码会编译失败，因为peek_while默认会拿走原迭代器的所有权，后续无法再调用next。如果需要保留原迭代器的使用权，调整使用方式即可：

// 调整后的正确用法
let mut chars = "keyword:".chars().peekable();
assert_eq!(chars.by_ref().peek_while(|c| c.is_alphabetic()).collect::<String>(), "keyword");
assert_eq!(chars.next().unwrap(), ':');

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