rust - 如何让迭代器链执行for_each()并返回一个计数？

Question

我有一个长的迭代器链，该链生成的数据通过for_each插入到固定数组中。 MWE版本(playground):

fn f(xs: impl Iterator<Item = u8>) {
    let mut buf = [0u8; 32];
    xs.zip(buf.iter_mut()).for_each(|(v, b)| *b = v);
    // ... process buf
}

我想计算插入了多少个值。在 mut中更新的 for_each变量是一种可能，但我想看看是否可以避免 mut。尝试了以下内容:

fn g(xs: impl Iterator<Item = u8>) {
    let mut buf = [0u8; 32];
    let c = xs.zip(buf.iter_mut()).inspect(|(v, b)| *b = v).count();
    // ... process buf
}

但这不起作用，因为我无法在 b内修改 inspect。
迭代器链有没有办法进行数组分配并返回计数？

Answer 1

在这种情况下，您可以简单地将inspect替换为map:

fn g(xs: impl Iterator<Item = u8>) {
    let mut buf = [0u8; 32];
    let c = xs.zip(buf.iter_mut()).map(|(v, b)| *b = v).count();
    // ... process buf
}

此 map()将产生 ()的迭代器(因为赋值返回)，但是您只是在计数，因此实际值是什么都没有关系。