Rust学习之实现命令行小工具minigrep（二）

Rust学习之实现命令行小工具minigrep（二）

Rust学习之实现命令行小工具minigrep（一）
前言

继承记载一下Rust 语言学习过程，上次写了一个命令行查找字符串的小项目minigrep。学习完了闭包（Closures）和迭代器（Iterators）之后，我们使用这两个技能来改进minigrep这个项目。
本文源码：代码堆栈
改进实现

在minigrep1 这个项目中，lib.rs文件，我们对Config的build方法中进行改进。
上次我们的代码如下：

2. pub struct Config {
3.     pub query: String,
4.     pub file_path: String,
5.     pub ignore_case: bool,
6. }
8. impl Config {
9.     pub fn build(args: &[String]) -> Result<Config, &'static str> { // 1
10.         if args.len() < 3 {
11.             return Err("Not enouth arguments"); // 传入参数不够
12.         }
13.         let query = args[1].clone();
14.         let file_path = args[2].clone();
15.         let ignore_case = env::var("IGNORE_CASE=1").is_ok();
17.         Ok(Config {
18.             query,
19.             file_path,
20.             ignore_case
21.         })
22.     }
23. }

复制代码

代码里args: &[String] 表示传进来的是字符串的切片引用，而后面的Config 须要拥有query，file_path 以及ignore_case。所以在代码中不得不克隆一份。
改进build方法

现在可以使用迭代器（Iterators）代替&[String]，对这里build方法进行改进，改进步骤如下：

• 直接将env::args() 传入 Config::build 方法，env::args() 返回一个Iterator
  
• build 方法声明中，args 参数类型改为impl Iterator<Item = String>
  
• 使用迭代器的next方法获取元素并使用match语法进行匹配
  

main.rs

2. use std::env;
3. use std::process;
5. use minigrep1::Config;
8. fn main() {
9.     let args: Vec<String> = env::args().collect();
10.     // dbg!(args); // stderr
11.     // 传入两个参数
12.     // &args 改为 env::args()
13.     let config = Config::build(env::args()).unwrap_or_else(|err| {
14.         eprintln!("Problem parsing arguments: {}", err);
15.         process::exit(1);
16.     });
18.     if let Err(e) = minigrep1::run(config) {
19.         eprintln!("Application error {}", e);
20.         process::exit(1)
21.     }
22. }

复制代码

lib.rs
改进前：

2. pub fn build(args: &Vec<String>) -> Result<Config, &'static str> {
3.     if args.len() < 3 {
4.       return Err("Not enouth arguments"); // 传入参数不够
5.     }
6.     let query = args[1].clone();
7.     let file_path = args[2].clone();
8. }

复制代码

改进后：

1. impl Config {
2.     pub fn build(mut args: impl Iterator<Item = String>) -> Result<Config, &'static str> {
3.         args.next(); // skip the first argument
4.         let query = match args.next() {
5.             Some(arg) => arg,
6.             None => return Err("Didn't get a query string"),
7.         };
9.         let file_path = match args.next() {
10.             Some(arg) => arg,
11.             None => return Err("Didn't get a file_path string"),
12.         };
14.         let ignore_case = env::var("IGNORE_CASE=1").is_ok();
16.         Ok(Config {
17.             query,
18.             file_path,
19.             ignore_case
20.         })
21.     }
22. }

复制代码

改进search方法

search方法中，是对文本进行遍历，进行判定查找。这里可以用迭代器，还可以用闭包。contents.lines() 返回一个迭代器
用迭代器把之前的for 循环都替换掉，使用filter时使用闭包

1. pub fn search<'a>(query: &str, contents: &'a str) -> Vec<&'a str> {
2.     contents
3.         .lines()
4.         .filter(|line| line.contains(query))
5.         .collect()
6. }

复制代码

同理，search_case_insensitive方法也同样使用迭代器和闭包进行修改
改进前：

1. pub fn search_case_insensitive<'a>(query: &str, contents: &'a str) -> Vec<&'a str> {
2.     let mut result = Vec::new();
3.     let query = query.to_lowercase();
4.     for line in contents.lines() {
5.         if line.to_lowercase().contains(&query) {
6.             result.push(line);
7.         }
8.     }
9.     result
10. }

复制代码

改进后：

1. pub fn search_case_insensitive<'a>(query: &str, contents: &'a str) -> Vec<&'a str> {
2.     let query: String = query.to_lowercase();
3.     contents.lines().filter(|line| line.to_lowercase().contains(&query)).collect()
4. }

复制代码

总结

项目使用迭代器和闭包优化之后首先代码效率提高了，为了将引用切片转移到Config中须要使用斲丧性能的clone方法，现在我们就不须要了。迭代器和迭代适配器filter 方法的使用让我们编写更加简明的代码。
​

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。