Rust从入门到实战系列三百五十二:直接使用 env::args 返回的迭代器

打开 I∕O 项目的 src∕main.rs 文件,它看起来应该像这样:# use std::env;

use std::process;

use minigrep::Config;

fn main() {
let args: Vec = env::args().collect();
let config = Config::new(&args).unwrap_or_else(|err| {
eprintln!(“Problem parsing arguments: {}”, err);
process::exit(1);
});
// --snip–

if let Err(e) = minigrep::run(config) {

eprintln!(“Application error: {}”, e);

process::exit(1);

}

}
修改第十二章结尾示例 12-24 中的 main 函数的开头为示例 13-25 中的代码。在更新 Config::new 之前
这些代码还不能编译:

use std::env;

use std::process;

use minigrep::Config;

fn main() {
let config = Config::new(env::args()).unwrap_or_else(|err| {
eprintln!(“Problem parsing arguments: {}”, err);
process::exit(1);
});
// --snip–

if let Err(e) = minigrep::run(config) {

eprintln!(“Application error: {}”, e);

process::exit(1);

}

}
将 env::args 的返回值传递给 Config::new
env::args 函数返回一个迭代器!不同于将迭代器的值收集到一个 vector 中接着传递一个 slice 给
Config::new,现在我们直接将 env::args 返回的迭代器的所有权传递给 Config::new。
接下来需要更新 Config::new 的定义。在 I∕O 项目的 src∕lib.rs 中,将 Config::new 的签名改为
这仍然不能编译因为我们还需更新函数体:

use std::env;

use std::error::Error;

use std::fs;

pub struct Config {

pub query: String,

pub filename: String,

pub case_sensitive: bool,

}

impl Config {
pub fn new(mut args: env::Args) -> Result<Config, &'static str> {
// --snip–

if args.len() < 3 {

return Err(“not enough arguments”);

}

let query = args[1].clone();

let filename = args[2].clone();

let case_sensitive = env::var(“CASE_INSENSITIVE”).is_err();

Ok(Config {

query,

filename,

case_sensitive,

})

}

}

pub fn run(config: Config) -> Result<(), Box> {

let contents = fs::read_to_string(config.filename)?;

let results = if config.case_sensitive {

search(&config.query, &contents)

} else {

search_case_insensitive(&config.query, &contents)

};

for line in results {

println!(“{}”, line);

}

Ok(())

}

pub fn search<'a>(query: &str, contents: &'a str) -> Vec<&'a str> {

let mut results = Vec::new();

for line in contents.lines() {

if line.contains(query) {

results.push(line);

}

}

results

}

pub fn search_case_insensitive<'a>(

query: &str,

contents: &'a str,

) -> Vec<&'a str> {

let query = query.to_lowercase();

let mut results = Vec::new();

for line in contents.lines() {

if line.to_lowercase().contains(&query) {

results.push(line);

}

}

results

}

#[cfg(test)]

mod tests {

use super:😗;

#[test]

fn case_sensitive() {

let query = “duct”;

let contents = "\

Rust:

safe, fast, productive.

Pick three.

Duct tape.";

assert_eq!(vec![“safe, fast, productive.”], search(query, contents));

}

#[test]

fn case_insensitive() {

let query = “rUsT”;

let contents = "\

Rust:

safe, fast, productive.

Pick three.

Trust me.";

assert_eq!(

vec![“Rust:”, “Trust me.”],

search_case_insensitive(query, contents)

);

}

}

以迭代器作为参数更新 Config::new 的签名
env::args 函数的标准库文档显示,它返回的迭代器的类型为 std:: env::Args。我们已经更新了 Config :: new
函数的签名,因此参数 args 的类型为 std:: env::Args 而不是 &[String]。因为我们拥有 args 的所有权,并
且将通过对其进行迭代来改变 args ,所以我们可以将 mut 关键字添加到 args 参数的规范中以使其可变。
现在我们还需指定字符串 slice 错误类型只能有 ’ static 生命周期。因为我们之前只会返回字符串 slice,
所以这是成立的。然而,当参数中有一个引用的时候,返回类型的引用有可能与参数的引用有着相同的
生命周期。之前第十章 ” 生命周期省略” 部分讨论的规则生效,因此无需注明 &str 的生命周期。随着对
args 的修改,生命周期省略规则不再适用,所以必须指定 ’ static 生命周期。

  • 23
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值