Rust标准库中是否存在类似Java Stream.mapMulti的等价方法？

Rust标准库中没有直接对应Java mapMulti的方法，但可以通过flat_map实现完全相同的逻辑，而且写法更简洁。

针对你的需求的优化实现 #

你的核心需求是筛选出数组中的偶数和1并求和，用flat_map可以直接实现：

fn main() {
    let buf = vec![1,2,3,4,5];
    let sum: i32 = buf.iter()
        .flat_map(|&x| {
            // 符合条件的元素返回Some(x)，否则返回None，None会被flat_map忽略
            if x % 2 == 0 || x == 1 {
                Some(x)
            } else {
                None
            }
        })
        .sum();
    
    println!("sum: {:?}", sum);
}

更复杂场景的扩展 #

如果需要一个元素生成多个输出元素（这也是mapMulti的核心能力之一），flat_map同样支持——只要让闭包返回一个迭代器即可：

fn main() {
    let buf = vec![1,2,3,4,5];
    let result: Vec<i32> = buf.iter()
        .flat_map(|&x| {
            match x {
                1 => vec![x].into_iter(), // 只输出自身
                x if x % 2 == 0 => vec![x, x + 1].into_iter(), // 输出自身和自身+1
                _ => std::iter::empty(), // 不输出任何元素
            }
        })
        .collect();
    
    println!("result: {:?}", result); // 输出: result: [1, 2, 3, 4, 5]
}

原理说明 #

flat_map的作用是将原迭代器的每个元素转换为一个子迭代器，然后将所有子迭代器的元素平铺成一个新的迭代器。这和Java mapMulti通过消费函数来发射0到多个元素的逻辑完全等价，是Rust标准库中处理这类“一对多”迭代转换的标准方式。

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