Rust之构建命令行程序(五)：环境变量

开发环境

Windows 11
Rust 1.77.0
VS Code 1.87.2

项目工程

这次创建了新的工程minigrep.

使用环境变量

我们将通过添加一个额外的功能来改进minigrep:一个不区分大小写的搜索选项，用户可以通过环境变量打开该选项。我们可以将此功能设置为命令行选项，并要求用户在每次想要应用它时输入它，但通过将其设置为环境变量，我们允许用户设置一次环境变量，并在该终端会话中使他们的所有搜索不区分大小写。

为不区分大小写的search函数编写失败测试

我们首先添加一个新的search_case_insensitive 函数，当环境变量有值时将调用该函数。我们将继续遵循TDD过程，所以第一步是再次编写一个失败的测试。我们将为新的search_case_insensitive 函数添加一个新测试，并将我们的旧测试从one_result 重命名为case_sensitive，以阐明两个测试之间的差异，如示例12-20所示。

文件名:src/lib.rs

rust 复制代码

#[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)
        );
    }
}

示例12-20:为我们将要添加的不区分大小写的函数添加新的失败测试

请注意，我们还编辑了旧测试的contents 。我们添加了一个新行，文本为"Duct tape "当我们以区分大小写的方式进行搜索时，使用不应该与查询"duct"匹配的大写D。以这种方式更改旧测试有助于确保我们不会意外破坏已经实现的区分大小写的搜索功能。这个测试现在应该通过了，并且在我们进行不区分大小写的搜索时应该会继续通过。

不区分大小写搜索的新测试使用"rUsT "作为查询。在我们即将添加的search_case_insensitive 函数中，查询"Rust: "应该与包含大写R的行"rUsT :"匹配，并与行"Trust me "匹配。即使两者的大小写与查询不同。这是我们失败的测试，它将无法编译，因为我们还没有定义search_case_insensitive 函数。随意添加一个总是返回空向量的框架实现，类似于我们对示例12-16中的search函数所做的那样，以查看测试编译和失败。

实现search_case_insensitive函数

示例12-21所示的search_case_insensitive 函数与search 函数几乎相同。唯一的区别是我们将query 和line都小写，因此无论输入参数的大小写如何，当我们检查该行是否包含查询时，它们都是相同的大小写。

文件名:src/lib.rs

rust 复制代码

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
}

示例12-21:定义search_case_insensitive函数在比较查询和行之前将其小写

首先，我们将query 字符串小写，并将其存储在同名的隐藏变量中。对查询调用to_lowercase 是必要的，因此无论用户的查询是"rust "、"Rust "、"rUsT "还是"rust "，我们都将查询视为"RUST "，并且不区分大小写。虽然to_lowercase可以处理基本的Unicode，但它不会100%准确。如果我们正在编写一个真正的应用程序，我们会想在这里多做一点工作，但这一节是关于环境变量的，而不是Unicode，所以我们将把它留在这里。

请注意，query 现在是String 而不是字符串片段，因为调用to_lowercase 会创建新数据而不是引用现有数据。例如，假设查询是"rUsT ":该字符串片段不包含供我们使用的小写u 或t ，因此我们必须分配一个包含"rust "的新String 。当我们现在将query 作为参数传递给contains 方法时，我们需要添加一个&符号，因为contains的签名被定义为接受一个字符串切片。

接下来，我们在line 添加一个对to_lowercase 的调用，以小写所有字符。现在我们已经将line 和query转换为小写，无论查询的大小写如何，我们都将找到匹配项。

让我们看看这个实现是否通过了测试:

bash 复制代码

$ cargo test
   Compiling minigrep v0.1.0 (file:///projects/minigrep)
    Finished test [unoptimized + debuginfo] target(s) in 1.33s
     Running unittests src/lib.rs (target/debug/deps/minigrep-9cd200e5fac0fc94)

running 2 tests
test tests::case_insensitive ... ok
test tests::case_sensitive ... ok

test result: ok. 2 passed; 0 failed; 0 ignored; 0 measured; 0 filtered out; finished in 0.00s

     Running unittests src/main.rs (target/debug/deps/minigrep-9cd200e5fac0fc94)

running 0 tests

test result: ok. 0 passed; 0 failed; 0 ignored; 0 measured; 0 filtered out; finished in 0.00s

   Doc-tests minigrep

running 0 tests

test result: ok. 0 passed; 0 failed; 0 ignored; 0 measured; 0 filtered out; finished in 0.00s

太好了！他们通过了。现在，让我们从run 函数中调用新的search_case_insensitive 函数。首先，我们将向Config结构添加一个配置选项，以在区分大小写和不区分大小写的搜索之间切换。添加此字段将导致编译器错误，因为我们还没有在任何地方初始化此字段:

文件名:src/lib.rs

rust 复制代码

pub struct Config {
    pub query: String,
    pub file_path: String,
    pub ignore_case: bool,
}

我们添加了包含布尔值的ignore_case 字段。接下来，我们需要run 函数来检查ignore_case 字段的值，并使用它来决定是调用search 函数还是search_case_insensitive函数，如示例12-22所示。这仍然无法编译。

文件名:src/lib.rs

rust 复制代码

pub fn run(config: Config) -> Result<(), Box<dyn Error>> {
    let contents = fs::read_to_string(config.file_path)?;

    let results = if config.ignore_case {
        search_case_insensitive(&config.query, &contents)
    } else {
        search(&config.query, &contents)
    };

    for line in results {
        println!("{line}");
    }

    Ok(())
}

清单12-22:根据config.ignore_case 中的值调用search 或search_case_insensitive

最后，我们需要检查环境变量。处理环境变量的函数在标准库中的env 模块中，因此我们将该模块纳入src/lib.rs的顶部。然后我们将使用env 模块中的var 函数来检查是否为名为IGNORE_CASE的环境变量设置了任何值，如示例12-23所示。

文件名:src/lib.rs

rust 复制代码

use std::env;
// --snip--

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

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

        let ignore_case = env::var("IGNORE_CASE").is_ok();

        Ok(Config {
            query,
            file_path,
            ignore_case,
        })
    }
}

清单12-23:检查名为IGNORE_CASE的环境变量中的任何值

在这里，我们创建了一个新变量ignore_case 。为了设置其值，我们调用env::var 函数并向其传递IGNORE_CASE 环境变量的名称。如果环境变量设置为任意值，env::var 函数返回的Result 将是成功的Ok 变量，该变量包含环境变量的值。如果未设置环境变量，它将返回Err变量。

我们对Result 使用is_ok 方法来检查是否设置了环境变量，这意味着程序应该进行不区分大小写的搜索。如果IGNORE_CASE 环境变量未设置为任何值，is_ok 将返回false ，程序将执行区分大小写的搜索。我们不关心环境变量的值，只关心它是设置的还是未设置的，因此我们检查的是is_ok ，而不是使用unwrap 、expect 或我们在Result上看到的任何其他方法。

我们将ignore_case 变量中的值传递给Config 实例，这样run 函数就可以读取该值并决定是调用search_case_insensitive 还是search，如示例12-22所示。

完整源码

文件名:src/lib.rs

rust 复制代码

pub mod cfg {
    use std::error::Error;
    use std::env;
    use std::fs;

    pub struct Config {
        pub query: String,
        pub file_path: String,
        pub ignore_case: bool,
    }
    
    impl Config {
        pub fn build(args: &[String]) -> Result<Config, &'static str> {
            if args.len() < 3 {
                return Err("not enough arguments");
            }
    
            let query = args[1].clone();
            let file_path = args[2].clone();
    
            let ignore_case = env::var("IGNORE_CASE").is_ok();
    
            Ok(Config {
                query,
                file_path,
                ignore_case,
            })
        }
    }


#[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)
        );
    }
}

    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
    }

    pub fn run(config: Config) -> Result<(), Box<dyn Error>> {
        let contents = fs::read_to_string(config.file_path)?;

        let results = if config.ignore_case {
            search_case_insensitive(&config.query, &contents)
        } else {
            search(&config.query, &contents)
        };

        for line in results {
            println!("{line}");
        }

        Ok(())
    }
}

文件名:src/main.rs

rust 复制代码

use std::env;
use std::process;
use minigrep::cfg;
 
 
fn main() {

    let args: Vec<String> = env::args().collect();
 
    let config = cfg::Config::build(&args).unwrap_or_else(|err| {
        println!("Problem parsing arguments: {err}");
        process::exit(1);
    });
 
    println!("Searching for {}", config.query);
    println!("In file {}", config.file_path);
 
    if let Err(e) = cfg::run(config) {
        println!("Application error: {e}");
        process::exit(1);
    } 
}

让我们试一试吧！首先，我们将在不设置环境变量的情况下运行我们的程序，并使用to查询，该查询应匹配包含全部小写单词"to"的任何行:

bash 复制代码

$ cargo run -- to poem.txt
   Compiling minigrep v0.1.0 (file:///projects/minigrep)
    Finished dev [unoptimized + debuginfo] target(s) in 0.0s
     Running `target/debug/minigrep to poem.txt`
Are you nobody, too?
How dreary to be somebody!

看来这仍然有效！现在，让我们运行程序，将IGNORE_CASE 设置为1 ，但仍然使用to。

bash 复制代码

$ IGNORE_CASE=1 cargo run -- to poem.txt

如果您使用的是PowerShell，则需要设置环境变量并作为单独的命令运行程序:

bash 复制代码

PS> $Env:IGNORE_CASE=1; cargo run -- to poem.txt

这将使IGNORE_CASE 在shell会话的剩余部分持续存在。可以使用Remove-Item cmdlet取消设置它:

bash 复制代码

PS> Remove-Item Env:IGNORE_CASE

我们应该得到包含"to"的行，这些行可能包含大写字母:

bash 复制代码

Are you nobody, too?
How dreary to be somebody!
To tell your name the livelong day
To an admiring bog!

太好了，我们还得到了包含"To"的行！我们的minigrep程序现在可以进行由环境变量控制的不区分大小写的搜索。现在您知道了如何使用命令行参数或环境变量来管理选项集。

一些程序允许相同配置的参数和环境变量。在这些情况下，程序会决定其中一个优先。对于您自己的另一个练习，尝试通过命令行参数或环境变量来控制大小写。如果程序运行时一个设置为区分大小写，一个设置为忽略大小写，请确定是命令行参数还是环境变量优先。

std::env模块包含许多更有用的处理环境变量的特性:查看其文档以了解可用的特性。

本章重点

环境变量的意义
编写不区分大小写的search函数
不区分大小写search函数的实现思路
通过添加环境变量的方式实现