使用迭代器优化IO项目¶

在前面两节，我们开发了一下给予IO操作的示例项目，我们现在来优化之前的项目。本节对应的代码目录src/s43_minigrep_optmize。

一、使用迭代器优化Config结构体方法¶

回顾之前的一个代码片段，在Config的new函数中包含如下代码

impl Config {
    pub fn new(args: &[String]) -> Result<Config, &'static str> {
        if args.len() < 3 {
            return Err("not enought arguments");
        }

        let query = args[1].clone();
        let filename = args[2].clone();
        // CASE_SENSITIVE 环境变量出现则代表区分大小写，而并不关心起具体值
        let case_sensitive = env::var("CASE_SENSITIVE").is_err();

        Ok(Config { query, filename, case_sensitive })
    }
}

在去参数的时候，之所以使用clone方法，是因为new函数并不拥有args参数的所有权，传进来仅仅只字符串切片，而返回的Config需要拥有字符串的所有权。针对query和filename两个字段，必须克隆出一份，这样Config才拥有它两的所有权。

而迭代器有用数据的所有权，这样就不需要使用切片了，我门还可以使用迭代器自带的功能进行长度检查和索引。在main函数中，env::args()产生了个迭代器，我们可以直接将迭代器传给new函数。改造如下

impl Config {
    pub fn new(mut args: std::env::Args) -> Result<Config, &'static str> {
        if args.len() < 3 {
            return Err("not enought arguments");
        }

        args.next(); // 放出第一个元素

        // 取第二个元素
        let query = match args.next() {
            Some(arg) => arg,
            None => return Err("Didn't get a query string"),
        };

        // 取第三个元素
        let filename = match args.next() {
            Some(arg) => arg,
            None => return Err("Didn't get a query string"),
        };

        // CASE_SENSITIVE 环境变量出现则代表区分大小写，而并不关心起具体值
        let case_sensitive = env::var("CASE_SENSITIVE").is_err();

        Ok(Config {
            query,
            filename,
            case_sensitive,
        })
    }
}

二、使用迭代器优化搜索函数¶

我们还可以优化search方法，使用迭代器替换原来的逻辑，如下示例代码

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

    contents
        .lines()
        .filter(|line| line.contains(query))
        .collect()
}

使用迭代器的好处是让开发者能专注于高层的业务逻辑，而不必陷入写循环、维护临时变量的细节工作里。

search_case_insensitive函数也可以优化成如下示例代码

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

    contents
        .lines()
        .filter(|line| line.contains(&query))
        .collect()
}

三、零开销抽象¶

在上面的“二”中，如果我们把一本小说的内容放到一个String里面，搜索“the”。实际上测试的结果是，使用迭代器比手写循环还要快。

不过重点不在这，而我们要知道的是：迭代器在rust里是一种高层次的抽象，但是它在编译后，生成的逻辑和我们手写底层代码几乎一样的。这套机制在rust里叫做**零开销抽象**（Zero-Cost Abstraction），意味着使用抽象时不会引入额外的运行时开销。所以我们可以尽情使用类似迭代器这样高层次的抽象。

实际上，rust使用了很多优化，使得产出的代码非常地高效，我门可以无所畏惧地使用闭包或者迭代器，它们既能让代码在感观上保持高层次的抽象，又不会因此带来任何的运行时性能的损失。