Rust中flatten对嵌套Vec与Result/Option迭代器的行为差异及原理

1. flatten处理Vec<Vec<_>>与Vec<Result<T,E>>/Vec<Option<T>>的行为差异 #

flatten的核心逻辑是对迭代器中的每个元素调用IntoIterator，再将子迭代器的元素依次输出，仅展开一层嵌套。两者的差异源于元素类型的IntoIterator实现不同：

• 对于Vec<Vec<_>>：每个内层Vec<_>本身实现了IntoIterator，会迭代出自身包含的所有元素。因此flatten会把两层嵌套的Vec展开成一层，输出所有内层元素，不会丢弃任何内容。
  示例：

  let v = vec![vec![1,2], vec![3,4]];
  let flattened: Vec<_> = v.into_iter().flatten().collect();
  // flattened = [1,2,3,4]
  

• 对于Vec<Result<T,E>>或Vec<Option<T>>：Result和Option的IntoIterator实现是条件性迭代：
  • Result<T,E>：仅当值为Ok(T)时，迭代出一个元素T；若为Err(E)，则产生空迭代（无元素输出）。
  • Option<T>：仅当值为Some(T)时，迭代出一个元素T；若为None，则产生空迭代。
    因此flatten在这里的作用是过滤掉错误/空值，并提取有效元素，同时丢弃Err和None的部分。
    示例：

  let results = vec![Ok(1), Err("oops"), Ok(2)];
  let flattened: Vec<_> = results.into_iter().flatten().collect();
  // flattened = [1,2]
  
  let options = vec![Some(1), None, Some(2)];
  let flattened: Vec<_> = options.into_iter().flatten().collect();
  // flattened = [1,2]
  

2. ReadDir迭代器调用flatten()的作用与原理 #

fs::read_dir返回的ReadDir是一个迭代器，其元素类型为Result<DirEntry, io::Error>——也就是说，它本身就是一个产生Result的迭代器。

Clippy建议用entries.flatten()替代嵌套if-let，本质是利用Result的IntoIterator实现：

• 当对ReadDir调用flatten()时，迭代器会遍历每个Result<DirEntry, io::Error>：
  • 若为Ok(DirEntry)，则将DirEntry作为元素输出；
  • 若为Err(io::Error)，则直接跳过该错误（不产生任何元素）。
• 这等价于手动用if let Ok(entry) = entry过滤错误的逻辑，但代码更简洁。

示例对比：
原嵌套if-let代码：

use std::fs;

fn main() -> std::io::Result<()> {
    let entries = fs::read_dir(".")?;
    for entry in entries {
        if let Ok(entry) = entry {
            println!("{}", entry.file_name().to_string_lossy());
        }
    }
    Ok(())
}

用flatten()简化后的代码：

use std::fs;

fn main() -> std::io::Result<()> {
    let entries = fs::read_dir(".")?;
    for entry in entries.flatten() {
        println!("{}", entry.file_name().to_string_lossy());
    }
    Ok(())
}

这里不需要ReadDir有专属的flatten实现，因为flatten是迭代器的通用方法（定义在Iterator trait中），只要迭代器的元素类型实现了IntoIterator（Result正好满足），就能调用flatten()。

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