开发环境
- Windows 10
- Rust 1.74.1
- VS Code 1.85.1
项目工程
这次创建了新的工程minigrep.
读取文件
现在,我们将添加读取file_path参数中指定的文件的功能。首先,我们需要一个样本文件来测试它:我们将使用一个包含少量文本的文件,多行包含一些重复的单词。示例12-3中有一首艾米莉·狄金森的诗,效果很好!在项目的根级别创建一个名为poem.txt的文件,输入诗歌“我是无名小卒!你是谁?”
文件名:poem.txt
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.
How dreary to be somebody!
How public, like a frog
To tell your name the livelong day
To an admiring bog!
示例12-3:艾米莉·狄金森的一首诗是一个很好的测试案例
文本就绪后,编辑src/main.rs并添加代码来读取该文件,如示例12-4所示。
文件名:src/main.rs
use std::env;
use std::fs;
fn main() {
let args: Vec<String> = env::args().collect();
let file_path = &args[2];
// --snip--
println!("In file {}", file_path);
let contents = fs::read_to_string(file_path)
.expect("Should have been able to read the file");
println!("With text:\n{contents}");
}
示例12-4:读取第二个参数指定的文件内容
首先,我们引入标准库的相关部分和一个use声明:我们需要std::fs来处理文件。
在main中,新语句fs::read_to_string接受file_path,打开该文件,并返回文件内容的std::io::Result<String >。
之后,我们再次添加一个临时的println!语句,该语句在文件被读取后打印contents的值,因此我们可以检查程序到目前为止是否工作正常。
让我们运行这段代码,将任意字符串作为第一个命令行参数(因为我们还没有实现搜索部分),将poem.txt文件作为第二个参数:
$ cargo run -- the poem.txt
Compiling minigrep v0.1.0 (file:///projects/minigrep)
Finished dev [unoptimized + debuginfo] target(s) in 0.0s
Running `target/debug/minigrep the poem.txt`
Searching for the
In file poem.txt
With text:
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.
How dreary to be somebody!
How public, like a frog
To tell your name the livelong day
To an admiring bog!
太好了!代码读取然后打印文件的内容。但是代码有一些缺陷。目前,main有多重职责:一般来说,如果每个功能只负责一个想法,功能会更清晰,更容易维护。另一个问题是我们没有尽可能好地处理错误。程序仍然很小,所以这些缺陷不是大问题,但是随着程序的增长,干净地修复它们会变得更加困难。在开发程序的早期开始重构是一个很好的实践,因为重构少量的代码要容易得多。我们接下来会这么做。
本章重点
- 如何获取命令行参数
- 如何读取文件