如何实现Rust函数：合并迭代器切片元素并自定义组合逻辑？

原代码存在编译错误和逻辑不符合预期的问题，以下是具体修复步骤与完整实现：

1. 核心问题拆解 #

• 泛型约束非法：Rust不允许在 trait bound 中使用impl Trait作为函数参数类型，原代码F: Fn(impl Iterator<Item = Option<A>>) -> B的写法违反了这一规则。
• 迭代终止逻辑错误：当前代码会无视迭代器是否耗尽，持续生成结果，不符合“任一迭代器耗尽则停止”的常规组合迭代行为。

完整修复代码 #

pub fn combine<I, A, F, B>(iters: &mut [I], combine_fn: F) -> impl Iterator<Item = B>
where
    I: Iterator<Item = A>,
    F: Fn(&[A]) -> B,
{
    std::iter::from_fn(move || {
        // 收集当前行元素，若任一迭代器返回None则终止迭代
        let mut row = Vec::with_capacity(iters.len());
        for it in iters.iter_mut() {
            match it.next() {
                Some(val) => row.push(val),
                None => return None,
            }
        }
        // 所有元素有效时调用组合函数生成结果
        Some(combine_fn(&row))
    })
}

使用示例 #

fn main() {
    let mut it1 = vec![1, 2, 3].into_iter();
    let mut it2 = vec![4, 5, 6].into_iter();
    let mut it3 = vec![7, 8, 9].into_iter();
    
    let combined = combine(&mut [&mut it1, &mut it2, &mut it3], |vals| {
        vals.iter().sum::<i32>()
    });
    
    for val in combined {
        println!("{}", val); // 输出 12, 15, 18
    }
}

特殊场景适配（可选） #

如果确实需要让组合函数处理包含Option的迭代器（允许部分元素缺失），可以改用动态分发调整实现，代价是轻微的性能损耗：

pub fn combine_opt<I, A, F, B>(iters: &mut [I], combine_fn: F) -> impl Iterator<Item = B> + '_
where
    I: Iterator<Item = A>,
    F: Fn(&mut dyn Iterator<Item = Option<A>>) -> B,
{
    std::iter::from_fn(move || {
        let mut row = iters.iter_mut().map(|it| it.next());
        Some(combine_fn(&mut row))
    })
}

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