如何复用已推进n步的Rust迭代器实现分阶段任务？

在Rust遍历文件行时，需要先执行task_A再执行task_B：前n行数据存入某数据结构（task_A），剩余行用来描述该数据结构的操作逻辑（task_B）。当前通过带enumerate的for循环结合if-else实现：

let file = File::open("./example.txt").unwrap();
let reader = BufReader::new(file);

for (i, lines) in reader.lines().map(|l| l.unwrap()).enumerate() {
    if i < n {
        do_task_a(&lines);
    } else {
        do_task_b(&lines);
    }
} 

虽然可用take_while方法，但仅能解决部分需求。希望将迭代器先传入task_A处理n步，再传入task_B处理剩余部分，且仅遍历文件一次（此需求为优化Advent of Code 2022第5题的实现）。是否存在这样的方法？

可以通过迭代器的by_ref()方法实现单次遍历下的分阶段处理，具体代码如下：

use std::fs::File;
use std::io::{BufRead, BufReader};

fn do_task_a(line: &str) {
    // 你的task_A逻辑，比如初始化数据结构
}

fn do_task_b(line: &str) {
    // 你的task_B逻辑，比如执行数据结构操作
}

fn main() {
    let n = 5; // 示例的前n行数量
    let file = File::open("./example.txt").unwrap();
    let mut reader = BufReader::new(file).lines().map(|l| l.unwrap());

    // 借用迭代器处理前n行，不转移所有权
    for line in reader.by_ref().take(n) {
        do_task_a(&line);
    }

    // 处理剩余所有行
    for line in reader {
        do_task_b(&line);
    }
}

原理说明 #

• by_ref()会返回迭代器的可变引用，调用take(n)时只会消耗前n个元素，但原迭代器的所有权仍保留在reader变量中。
• 后续遍历reader时，迭代器会从第n+1行开始继续读取，整个过程仅对文件做一次遍历，完全符合性能和逻辑分离的需求。

针对Advent of Code 2022第5题的场景，这种写法可以清晰分离栈结构初始化和移动指令执行两个阶段，代码可读性和执行效率都会更优。

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