7.0 KiB
接受命令行参数
ch12-01-accepting-command-line-arguments.md
commit b8e4fcbf289b82c12121b282747ce05180afb1fb
第一个任务是让greprs能够接受两个命令行参数：文件名和要搜索的字符串。也就是说希望能够使用cargo run，要搜索的字符串和被搜索的文件的路径来运行程序，像这样：
$ cargo run searchstring example-filename.txt
现在cargo new生成的程序忽略任何传递给它的参数。crates.io 上有一些现存的可以帮助我们接受命令行参数的库，不过因为我们正在学习，让我们实现一个。
读取参数值
为了能够获取传递给程序的命令行参数的值，我们需要调用一个 Rust 标准库提供的函数：std::env::args。这个函数返回一个传递给程序的命令行参数的迭代器（iterator）。我们还未讨论到迭代器，第十三章会全面的介绍他们。但是对于我们现在的目的来说只需要明白两点：
- 迭代器生成一系列的值。
- 在迭代器上调用
collect方法可以将其生成的元素转换为一个 vector。
让我们尝试一下：使用列表 12-1 中的代码来读取任何传递给greprs的命令行参数并将其收集到一个 vector 中。
Filename: src/main.rs
use std::env;
fn main() {
let args: Vec<String> = env::args().collect();
println!("{:?}", args);
}
Listing 12-1: Collect the command line arguments into a vector and print them out
首先使用use语句来将std::env模块引入作用域以便可以使用它的args函数。注意std::env::args函数嵌套进了两层模块中。如第七章讲到的，当所需函数嵌套了多于一层模块时，通常将父模块引入作用域，而不是其自身。这便于我们利用std::env中的其他函数。这比增加了use std::env::args;后仅仅使用args调用函数要更明确一些；这样看起来好像一个定义于当前模块的函数。
注意：
std::env::args在其任何参数包含无效 Unicode 字符时会 panic。如果你需要接受包含无效 Unicode 字符的参数，使用std::env::args_os代替。这个函数返回OsString值而不是String值。出于简单考虑这里使用std::env::args，因为OsString值每个平台都不一样而且比String值处理起来更复杂。
在main函数的第一行，我们调用了env::args，并立即使用collect来创建了一个包含迭代器所有值的 vector。collect可以被用来创建很多类型的集合，所以这里显式注明的args类型来指定我们需要一个字符串 vector。虽然在 Rust 中我们很少会需要注明类型，collect就是一个经常需要注明类型的函数，因为 Rust 不能推断出你想要什么类型的集合。
最后，我们使用调试格式:?打印出 vector。让我们尝试不用参数运行代码，接着用两个参数：
$ cargo run
["target/debug/greprs"]
$ cargo run needle haystack
...snip...
["target/debug/greprs", "needle", "haystack"]
你可能注意到了 vector 的第一个值是"target/debug/greprs"，它是我们二进制文件的名称。其原因超出了本章介绍的范围，不过需要记住的是我们保存了所需的两个参数。
将参数值保存进变量
打印出参数 vector 中的值仅仅展示了可以访问程序中指定为命令行参数的值。但是这并不是我们想要做的，我们希望将这两个参数的值保存进变量这样就可以在程序使用这些值。让我们如列表 12-2 这样做：
Filename: src/main.rs
use std::env;
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);
}
Listing 12-2: Create variables to hold the query argument and filename argument
正如我们在打印出 vector 时所看到的，程序的名称占据了 vector 的第一个值args[0]，所以我们从索引1开始。第一个参数greprs是需要搜索的字符串，所以将其将第一个参数的引用存放在变量query中。第二个参数将是文件名，所以将第二个参数的引用放入变量filename中。
我们将临时打印出出这些变量的值，再一次证明代码如我们期望的那样工作。让我们使用参数test和sample.txt再次运行这个程序：
$ cargo run test sample.txt
Finished debug [unoptimized + debuginfo] target(s) in 0.0 secs
Running `target/debug/greprs test sample.txt`
Searching for test
In file sample.txt
好的，它可以工作！我们将所需的参数值保存进了对应的变量中。之后会增加一些错误处理来应对类似用户没有提供参数的情况，不过现在我们将忽略他们并开始增加读取文件功能。