当前位置：首页 > news >正文

Rust 文件读取：实现我们的 “迷你 grep”

news 2025/6/19 21:43:15

1. 准备示例文件

首先，在项目根目录（与 Cargo.toml 同级）下新建一个名为 poem.txt 的文件。示例内容可参考 Emily Dickinson 的诗：

I'm nobody! Who are you?
Are you nobody, too?
Then there's a pair of us — don't tell!
They'd banish us, you know.

这为我们后续的搜索及测试提供了多行文本。

2. 在 `src/main.rs` 中读入文件

让我们打开 src/main.rs，在前面已经实现的“获取命令行参数”逻辑后，添加读取文件内容的功能。完整示例如下：

use std::env;
use std::fs;fn main() {let args: Vec<String> = env::args().collect();let query = &args[1];        // 搜索关键字（暂时没用到）let file_path = &args[2];    // 文件路径println!("Searching for: {}", query);println!("In file: {}", file_path);let contents = fs::read_to_string(file_path).expect("Should have been able to read the file");println!("File contents:\n{contents}");
}

代码解读

use std::fs;
- 我们从标准库引入了 fs 模块来完成文件操作。
fs::read_to_string(file_path)
- 该函数以只读方式打开指定文件并读取其完整内容。它返回一个 Result<String>，如果读取成功即包含文本内容，否则包含错误信息。
expect("...")
- 简化的错误处理方式。如果无法读取文件，程序会在此处退出并打印自定义错误信息。
println!("File contents:\n{contents}")
- 打印出文件全部内容，以便确认我们成功地读取到文件。

3. 运行示例

我们假设要在文件中搜索“searchstring”（功能尚未实现，但先用它做参数示例）：

$ cargo run -- searchstring poem.txtCompiling minigrep v0.1.0 (~/minigrep)Finished dev [unoptimized + debuginfo] target(s) in 0.74sRunning `target/debug/minigrep searchstring poem.txt`
Searching for: searchstring
In file: poem.txt
File contents:
I'm nobody! Who are you?
Are you nobody, too?
Then there's a pair of us — don't tell!
They'd banish us, you know.

如输出所示，minigrep 成功打印了 poem.txt 的全部文本。

4. 潜在问题及后续改进

当前代码虽能正常运行，但仍有一些可优化之处：

main 函数职责过多
- main 同时负责解析参数、读取文件、打印内容，如果将来添加更多功能，会越来越臃肿。
- 我们可以将配置解析与文件搜索等操作拆分到单独的函数或模块。
错误处理不够灵活
- 这里我们使用 expect 简化了错误处理，一旦出错程序就立刻崩溃。
- 若是生产环境，可考虑返回 Result 并进行更完善的错误提示与恢复机制。
搜索功能尚未实现
- 虽然可以成功读取文件，但还无法对文件内容进行关键字查找。
- 后续我们会在搜索逻辑、大小写处理、行输出等方面展开更多讨论。