4.2 KiB
读取文件
ch12-02-reading-a-file.md
commit b8e4fcbf289b82c12121b282747ce05180afb1fb
接下来我们将读取由命令行文件名参数指定的文件。首先,需要一个用来测试的示例文件——用来确保greprs
正常工作的最好的文件是拥有少量文本和多个行且有一些重复单词的文件。列表 12-3 是一首艾米莉·狄金森(Emily Dickinson)的诗,它正适合这个工作!在项目根目录创建一个文件poem.txt
,并输入诗 "I'm nobody! Who are you?":
Filename: 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!
Listing 12-3: The poem "I'm nobody! Who are you?" by Emily Dickinson that will make a good test case
创建完这个文件之后,修改 src/main.rs 并增加如列表 12-4 所示的打开文件的代码:
Filename: src/main.rs
use std::env;
use std::fs::File;
use std::io::prelude::*;
fn main() {
let args: Vec<String> = env::args().collect();
let query = &args[1];
let filename = &args[2];
println!("Searching for {}", query);
println!("In file {}", filename);
let mut f = File::open(filename).expect("file not found");
let mut contents = String::new();
f.read_to_string(&mut contents).expect("something went wrong reading the file");
println!("With text:\n{}", contents);
}
Listing 12-4: Reading the contents of the file specified by the second argument
首先,增加了更多的use
语句来引入标准库中的相关部分:需要std::fs::File
来处理文件,而std::io::prelude::*
则包含许多对于 I/O 包括文件 I/O 有帮助的 trait。类似于 Rust 有一个通用的 prelude 来自动引入特定内容,std::io
也有其自己的 prelude 来引入处理 I/O 时所需的通用内容。不同于默认的 prelude,必须显式use
位于std::io
中的 prelude。
在main
中,我们增加了三点内容:第一,通过传递变量filename
的值调用File::open
函数的值来获取文件的可变句柄。创建了叫做contents
的变量并将其设置为一个可变的,空的String
。它将会存放之后读取的文件的内容。第三,对文件句柄调用read_to_string
并传递contents
的可变引用作为参数。
在这些代码之后,我们再次增加了临时的println!
打印出读取文件后contents
的值,这样就可以检查目前为止的程序能否工作。
尝试运行这些代码,随意指定一个字符串作为第一个命令行参数(因为还未实现搜索功能的部分)而将 poem.txt 文件将作为第二个参数:
$ cargo run the poem.txt
Finished debug [unoptimized + debuginfo] target(s) in 0.0 secs
Running `target/debug/greprs 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
函数有着多个功能,同时也没有处理可能出现的错误。虽然我们的程序还很小,这些瑕疵并不是什么大问题。不过随着程序功能的丰富,将会越来越难以用简单的方法修复他们。在开发程序时,及早开始重构是一个最佳实践,因为重构少量代码时要容易的多,所以让我们现在就开始吧。