使用環(huán)境變量實現(xiàn)Rust程序中的不區(qū)分大小寫搜索方式
更新時間：2025年02月26日 09:36:20 作者：Hello.Reader
本文介紹了如何在Rust中實現(xiàn)不區(qū)分大小寫的搜索功能,并通過測試驅(qū)動開發(fā)（TDD）方法逐步實現(xiàn)該功能,通過修改運行函數(shù)和獲取環(huán)境變量,程序可以根據(jù)環(huán)境變量控制搜索模式
步驟 1：編寫失敗的測試
我們遵循測試驅(qū)動開發(fā)（TDD）方法，首先編寫一個測試來驗證不區(qū)分大小寫搜索功能。由于我們尚未實現(xiàn)該功能，因此這個測試會失敗。
以下是測試代碼：
#[cfg(test)]
mod tests {
use super::*;
// 測試：區(qū)分大小寫的搜索
#[test]
fn case_sensitive() {
let query = "duct";
let contents = "\
Rust:
safe, fast, productive.
Pick three.
Duct tape.";
// 期望返回大小寫敏感的匹配結(jié)果
assert_eq!(vec!["safe, fast, productive."], search(query, contents));
}
// 測試：不區(qū)分大小寫的搜索
#[test]
fn case_insensitive() {
let query = "rUsT"; // 查詢字符串的大小寫不一致
let contents = "\
Rust:
safe, fast, productive.
Pick three.
Trust me.";
// 期望返回不區(qū)分大小寫的匹配結(jié)果
assert_eq!(
vec!["Rust:", "Trust me."],
search_case_insensitive(query, contents)
);
}
}在上述代碼中：
case_sensitive測試驗證了原始的區(qū)分大小寫的搜索。case_insensitive測試驗證了我們將在后面實現(xiàn)的不區(qū)分大小寫的搜索。此時，由于我們還沒有實現(xiàn)search_case_insensitive函數(shù)，編譯將失敗。
步驟 2：實現(xiàn) search_case_insensitive 函數(shù)
為了實現(xiàn)不區(qū)分大小寫的搜索，我們需要修改 search 函數(shù)，使其能夠忽略大小寫。
以下是 search_case_insensitive 函數(shù)的實現(xiàn)：
// 不區(qū)分大小寫的搜索函數(shù)
pub fn search_case_insensitive<'a>(
query: &str, // 查詢字符串
contents: &'a str, // 文件內(nèi)容
) -> Vec<&'a str> {
let query = query.to_lowercase(); // 將查詢字符串轉(zhuǎn)換為小寫
let mut results = Vec::new(); // 存儲匹配結(jié)果
// 遍歷文件中的每一行
for line in contents.lines() {
// 將每一行轉(zhuǎn)換為小寫再進行比較
if line.to_lowercase().contains(&query) {
results.push(line); // 如果匹配，加入結(jié)果列表
}
}
results // 返回匹配的行
}代碼解釋：
query.to_lowercase()將查詢字符串轉(zhuǎn)換為小寫，以確保不區(qū)分大小寫地進行匹配。line.to_lowercase()將每一行內(nèi)容轉(zhuǎn)換為小寫，并檢查是否包含查詢字符串。
步驟 3：修改 run 函數(shù)
run 函數(shù)是程序的核心，它負責執(zhí)行搜索并輸出結(jié)果。我們需要修改 run 函數(shù)，使其根據(jù) Config 結(jié)構(gòu)體中的 ignore_case 字段來決定是使用區(qū)分大小寫的 search 函數(shù)，還是不區(qū)分大小寫的 search_case_insensitive 函數(shù)。
// 根據(jù)配置運行搜索
pub fn run(config: Config) -> Result<(), Box<dyn Error>> {
let contents = fs::read_to_string(config.file_path)?; // 讀取文件內(nèi)容
// 根據(jù) ignore_case 字段決定使用哪種搜索
let results = if config.ignore_case {
search_case_insensitive(&config.query, &contents) // 不區(qū)分大小寫的搜索
} else {
search(&config.query, &contents) // 區(qū)分大小寫的搜索
};
// 打印匹配的行
for line in results {
println!("{line}");
}
Ok(())
}代碼解釋：
Config結(jié)構(gòu)體包含一個ignore_case字段，用于控制是否啟用不區(qū)分大小寫的搜索。- 根據(jù)
ignore_case的值，我們決定調(diào)用search還是search_case_insensitive。
步驟 4：獲取環(huán)境變量
現(xiàn)在，我們需要從環(huán)境變量中獲取 IGNORE_CASE 的值，來決定是否啟用不區(qū)分大小寫的搜索。
我們使用 Rust 標準庫中的 env 模塊來訪問環(huán)境變量。
use std::env;
impl Config {
// 構(gòu)建 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();
// 檢查環(huán)境變量 IGNORE_CASE 是否設(shè)置
let ignore_case = env::var("IGNORE_CASE").is_ok(); // 如果環(huán)境變量存在，則啟用不區(qū)分大小寫的搜索
Ok(Config {
query,
file_path,
ignore_case,
})
}
}代碼解釋：
- 我們使用
env::var("IGNORE_CASE").is_ok()來檢查環(huán)境變量IGNORE_CASE是否被設(shè)置。 - 如果設(shè)置了該環(huán)境變量，我們將
ignore_case設(shè)置為true，否則默認為false。
步驟 5：運行程序
現(xiàn)在，我們的程序已經(jīng)支持根據(jù)環(huán)境變量來控制搜索模式。你可以通過以下命令來運行程序：
- 不使用環(huán)境變量，執(zhí)行區(qū)分大小寫的搜索：
$ cargo run -- to poem.txt
- 設(shè)置環(huán)境變量
IGNORE_CASE=1，啟用不區(qū)分大小寫的搜索：
$ IGNORE_CASE=1 cargo run -- to poem.txt
對于 PowerShell 用戶，可以使用以下命令：
PS> $Env:IGNORE_CASE=1; cargo run -- to poem.txt
總結(jié)
通過使用環(huán)境變量，我們成功地在Rust程序中實現(xiàn)了不區(qū)分大小寫的搜索功能。
這種方式讓用戶可以通過簡單的環(huán)境變量配置來改變程序的行為，而不需要每次運行程序時都指定命令行參數(shù)。
通過這種方法，我們可以在命令行工具中靈活地控制不同的搜索模式，使程序更加友好和易于配置。
在本教程中，我們：
- 通過 TDD 編寫了失敗的測試，并逐步實現(xiàn)了功能。
- 學會了如何使用 Rust 標準庫中的
env模塊獲取環(huán)境變量。 - 通過環(huán)境變量控制程序行為，讓命令行工具更加靈活。
通過這個示例，希望你對 Rust 中如何使用環(huán)境變量有所了解，并能夠在自己的項目中靈活應用。
以上為個人經(jīng)驗，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
這篇文章主要介紹了Rust語言的println!宏的格式占位符,這只是格式說明符的一部分清單,Rust?的格式化系統(tǒng)非常靈活和強大,支持更多的選項和組合,需要的朋友可以參考下2024-03-03
在Rust語言中，泛型是一種強大的工具，它允許我們編寫可復用且靈活的代碼，本文主要介紹了Rust中泛型的學習筆記，具有一定的參考價值，感興趣的可以了解一下2024-03-03
Sauron?是一個多功能的?Web?框架和庫，用于構(gòu)建客戶端和/或服務器端?Web?應用程序，重點關(guān)注人體工程學、簡單性和優(yōu)雅性,這篇文章主要介紹了Rust使用Sauron實現(xiàn)Web界面交互,需要的朋友可以參考下2024-03-03
在Rust中,async/await用于編寫異步代碼,使得異步操作更易于理解和編寫,通過使用await,在async函數(shù)或代碼塊中等待Future完成,而不會阻塞線程,允許同時執(zhí)行其他Future,這種機制簡化了異步編程的復雜性,使代碼更加直觀2024-10-10
Rust語言開發(fā)環(huán)境搭建詳細教程(圖文教程)
本文主要介紹了rust編程語言在windows上開發(fā)環(huán)境的搭建方法,文中通過圖文的非常詳細,對大家的學習或者工作具有一定的參考學習價值,需要的朋友們下面隨著小編來一起學習學習吧2024-02-02
Rust可迭代類型迭代器正確創(chuàng)建自定義可迭代類型的方法
在 Rust 中, 如果一個類型實現(xiàn)了 Iterator, 那么它會被同時實現(xiàn) IntoIterator, 具體邏輯是返回自身, 因為自身就是迭代器,這篇文章主要介紹了Rust可迭代類型迭代器正確創(chuàng)建自定義可迭代類型的方法,需要的朋友可以參考下2023-12-12
libbpf和Rust開發(fā)ebpf程序?qū)崙?zhàn)示例
這篇文章主要為大家介紹了libbpf和Rust開發(fā)ebpf程序?qū)崙?zhàn)示例,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進步,早日升職加薪2023-12-12
先前我們提到過?trait，那么Rust中的trait?是啥呢？本文將通過一些示例為大家詳細講講Rust中trait的使用，感興趣的小伙伴可以了解一下2022-11-11
哈希表也是集合中的一種,也是最常用的集合形式,目前Rust語言核心部分沒有對哈希表進行實現(xiàn),是使用標準庫提供的,這篇文章主要介紹了Rust語言之哈希表,需要的朋友可以參考下2024-02-02
任何項目都離不開對于配置文件的讀取和解析,rust項目也一樣,本文主要介紹了Rust讀取配置文件的實現(xiàn)步驟,具有一定的參考價值,感興趣的可以了解一下2023-12-12