Rust中统计文件行数后无法迭代文件行的原因及解决方法

你遇到的问题核心在于文件句柄的读取指针状态被修改了。当你第一次用BufReader::new(&file)创建读取器并调用.count()时，底层的文件句柄已经把读取指针移动到了文件末尾。后续创建的第二个BufReader基于同一个文件引用&file，会从当前指针位置（也就是文件末尾）开始读取，自然读不到任何内容——这和你创建多少个独立的BufReader无关，因为它们共享同一个底层文件句柄的状态。

这里提供三种可行的解决思路，你可以根据场景选择：

1. 重新打开文件 #

每次打开文件都会获得一个全新的文件句柄，读取指针默认从头开始。这种方式适合大文件，不需要占用太多内存：

let file_path = Path::new(&args[1]);

// 第一次打开文件统计行数
let lines_amount = {
    let file = OpenOptions::new().read(true).open(file_path)
        .expect("failed to open file");
    BufReader::new(file).lines().count()
};

if lines_amount == 0 {
    panic!("The file has no lines");
}

println!("{}", lines_amount);

// 第二次打开文件迭代内容
let file = OpenOptions::new().read(true).open(file_path)
    .expect("failed to open file");
for (i, line_result) in BufReader::new(file).lines().enumerate() {
    let line = line_result.expect("failed to read line");
    // 处理逻辑...
}

2. 将文件内容读入内存（适合小文件） #

一次性把整个文件读取到内存中，然后拆分成行处理。这种方式只需要读取一次文件，操作更高效，但会占用和文件大小相当的内存：

use std::fs::read_to_string;

let file_path = Path::new(&args[1]);
let content = read_to_string(file_path)
    .expect("failed to read file");

let lines: Vec<&str> = content.lines().collect();
let lines_amount = lines.len();

if lines_amount == 0 {
    panic!("The file has no lines");
}

println!("{}", lines_amount);

for (i, line) in lines.iter().enumerate() {
    // 处理每一行...
}

3. 重置文件指针位置 #

利用Seek trait将文件指针移回文件开头，这样可以复用同一个文件句柄。注意这种方式只适用于支持随机访问的文件（比如本地磁盘文件，管道、套接字等流类型不行）：

use std::io::{Seek, SeekFrom};

let file_path = Path::new(&args[1]);
let mut file = OpenOptions::new().read(true).open(file_path)
    .expect("failed to open file");

// 统计行数
let lines_amount = BufReader::new(&file).lines().count();
if lines_amount == 0 {
    panic!("The file has no lines");
}

println!("{}", lines_amount);

// 将文件指针移回开头
file.seek(SeekFrom::Start(0))
    .expect("failed to seek file");

// 重新迭代文件内容
for (i, line_result) in BufReader::new(&file).lines().enumerate() {
    let line = line_result.expect("failed to read line");
    // 处理逻辑...
}

内容的提问来源于stack exchange，提问作者Jonas Grønbek