Rust中如何实现按条件截断迭代器的until(f)功能

嘿，你这个需求其实在Rust里很好解决，不管是用标准库现成的方法，还是自己实现一个语义更直观的until迭代器适配器都可以，咱们一步步来看！

用标准库快速解决你的质数检查问题 #

你的核心诉求是：遍历质数列表时，一旦遇到满足某个条件的元素就立刻停止迭代，只检查前面的元素。其实Rust标准库的Iterator::take_while方法正好能帮你做到这一点——它会持续迭代并返回元素，直到传入的条件返回false为止。

不过要注意，你想要的是“直到条件f为true就停止”，所以咱们把条件取反传给take_while就行。比如你的质数检查代码可以改成这样：

// 修改extend_by_one方法里的循环判断部分
while self.primes.iter()
    .take_while(|&&x| x * x <= next_option) // 只保留x² ≤ next_option的质数
    .any(|&x| next_option % x == 0)
{
    next_option += 2;
}

这样一来，迭代器会在遇到第一个x*x > next_option的质数时立刻停止，不会再遍历后面的元素，完美符合你想要的“截断”效果，同时也优化了质数检查的效率。

自定义实现until(f)迭代器适配器 #

如果你觉得take_while加取反的写法不够直观，想拥有一个语义更清晰的until(f)方法（直接表达“直到f为true就停止”），那可以自己给迭代器实现一个扩展适配器。

步骤1：定义Until迭代器结构体 #

首先我们需要一个结构体来包装原迭代器和判断函数：

use std::iter::Iterator;

// 包装原迭代器和终止条件函数
struct Until<I, F> {
    iter: I,
    f: F,
    stopped: bool, // 标记是否已经触发终止条件
}

步骤2：实现Iterator trait #

接下来为Until结构体实现Iterator trait，定义它的迭代逻辑：

impl<I, F> Iterator for Until<I, F>
where
    I: Iterator,
    F: FnMut(&I::Item) -> bool,
{
    type Item = I::Item;

    fn next(&mut self) -> Option<Self::Item> {
        // 如果已经停止，直接返回None
        if self.stopped {
            return None;
        }

        match self.iter.next() {
            Some(item) => {
                // 检查当前元素是否满足终止条件
                if (self.f)(&item) {
                    self.stopped = true;
                    None // 触发终止，不返回当前元素
                } else {
                    Some(item) // 未终止，返回当前元素
                }
            }
            None => None, // 原迭代器已经耗尽
        }
    }
}

步骤3：为所有迭代器添加until方法 #

最后，我们通过trait给所有实现了Iterator的类型添加until扩展方法：

// 定义扩展trait
trait UntilIterator: Iterator {
    fn until<F>(self, f: F) -> Until<Self, F>
    where
        F: FnMut(&Self::Item) -> bool,
        Self: Sized,
    {
        Until {
            iter: self,
            f,
            stopped: false,
        }
    }
}

// 为所有Iterator实现这个trait
impl<I: Iterator> UntilIterator for I {}

现在你就可以像最开始设想的那样写代码了：

while self.primes.iter()
    .until(|&&x| x * x > next_option) // 直到x² > next_option就停止迭代
    .any(|&x| next_option % x == 0)
{
    next_option += 2;
}

这个自定义的until方法语义非常清晰，完全符合你对“逆filter，遇到第一个true就停止迭代”的定义。

备注：内容来源于stack exchange，提问作者iHnR