以编程方式从 Rust 中的映射函数创建迭代器

Question

我正在尝试编写如下代码，但我应用f1和f2的次数不定：

#![feature(impl_trait_in_bindings)]

fn f1(c: char) -> impl IntoIterator<Item = char> {
    vec!['A', c]
}

fn f2(c: char) -> impl IntoIterator<Item = char> {
    vec!['C', 'D', c] 
}

fn main() {
    let x = vec!['X', 'X', 'X'];

    let v: impl Iterator<Item = char> = x.into_iter();
    let v = v.flat_map(f1);
    let v = v.flat_map(f2);

    println!("Force evaluation of five elements: {:?}", v.take(5).collect::<Vec<_>>());
}

我想用迭代重新分配 v 的循环替换let v =...行，例如

    let mut v: impl Iterator<Item = char> = x.into_iter();
    for i in 0..f1Times {
      v = v.flat_map(f1);
    }
    for i in 0..f2Times {
      v = v.flat_map(f2);
    }

...例如，对于我的用例，我可能有多个功能，但我不知道要提前应用哪些功能（或应用多少次）。 我希望结果是一个迭代器，我只能从中take有限数量的项目，并且我想避免调用生成这些项目不需要的任何函数。

我无法让这些类型工作。 例如，对于我上面提出的let mut块，我得到：

mismatched types
expected opaque type `impl Iterator`
        found struct `FlatMap<impl Iterator, impl IntoIterator, fn(char) -> impl IntoIterator {f1}>`

有没有一种以编程方式构建这种迭代器的好方法？

Answer 1

我发现这种模式有效，但仍然不知道它是惯用的还是推荐的......

#![feature(impl_trait_in_bindings)]

fn f1(c: char) -> impl Iterator<Item = char> {
    Box::new(vec!['A', c].into_iter())
}

fn f2(c: char) -> impl Iterator<Item = char> {
    Box::new(vec!['C', 'D', c].into_iter())
}

fn main() {
    let x = vec!['X', 'X', 'X'];

    let mut v: Box<dyn Iterator<Item = char>> = Box::new(x.into_iter());
    let f1_ntimes = 2;
    for _i in 0..f1NTimes {
        v = Box::new(v.into_iter().flat_map(f1));
    }

    let f2_ntimes = 2;
    for _i in 0..f2_ntimes {
        v = Box::new(v.into_iter().flat_map(f2));
    }

    println!("Force evaluation of five elements: {:?}", v.take(5).collect::<Vec<_>>());
}