環(huán)境變量是存儲(chǔ)在操作系統(tǒng)中的一組鍵值對(duì)。它們用于存儲(chǔ)系統(tǒng)和其他應(yīng)用程序所需的配置信息。本文我們將探索如何在Rust中使用標(biāo)準(zhǔn)庫(kù)以及dotenv crate來處理環(huán)境變量。

環(huán)境變量

環(huán)境變量提供了一種靈活的方式來配置應(yīng)用程序，而無需直接在源代碼中硬編碼配置值。這可以為不同的環(huán)境（開發(fā)、測(cè)試和生產(chǎn)）管理不同的配置，或讓敏感信息（如API密鑰）的安全變得更加容易。

讓我們看3個(gè)不同的例子來說明如何在Rust中使用環(huán)境變量。

1. 使用std::env模塊
2. 使用命令行（Windows和Linux）
3. 使用dotenv crate

使用std::env模塊

Rust提供了std::env模塊來與環(huán)境變量交互。該模塊可以讀取、設(shè)置和刪除環(huán)境變量。

首先，從標(biāo)準(zhǔn)庫(kù)中導(dǎo)入env模塊：

use std::env;
fn main(){let key: &str = "AAA";std::env::set_var(key, "123");match env::var(key){OK(val:String) => println!("():{:?}", key, val),Err(e: VarError) => println!("Error{}:{}", key, e),}
}

在終端中輸入cargo run來測(cè)試程序，輸出應(yīng)該是這樣的：

AAA:"123"

• 設(shè)置刪除變量

可以使用set_var函數(shù)設(shè)置環(huán)境變量，刪除環(huán)境變量使用remove_var函數(shù)。下面是示例，我們將變量AAA設(shè)置為值123：

fn main() {let key = "AAA";std::env::set_var(key, "123"); // Sets AAA to 123
}fn main() {let key = "AAA";env::remove_var(key); // Removes the variable AAA
}

• 檢測(cè)變量存在

要檢查環(huán)境變量是否存在，請(qǐng)使用env::var函數(shù)，該函數(shù)返回Result。你可以用match語(yǔ)句來處理：

fn main() {let key = "AAA";match env::var(key) {Ok(val) => println!("{}: {:?}", key, val),Err(e) => println!("Error {}: {}", key, e),}
}

使用命令行環(huán)境變量

你可以在運(yùn)行Rust程序時(shí)直接從命令行傳遞環(huán)境變量。下面是如何讀取CLI_ARG環(huán)境變量：

fn main() {let cli_arg = env::var("CLI_ARG");match cli_arg {Ok(val) => println!("CLI_ARG: {:?}", val),Err(e) => println!("Error CLI_ARG: {}", e),}
}

要從命令行讀取它們，可以使用以下命令：

在Linux / macOS:

CLI_ARG=TEST cargo run

在Windows(powershell):

$env:CLI_ARG="TEST"; cargo run

使用dotenv Crate

除了標(biāo)準(zhǔn)庫(kù)之外，您還可以使用dotenv crate從.env文件加載環(huán)境變量。這在開發(fā)環(huán)境中特別有用。首先將dotenv添加到Cargo.toml文件:

[dependencies]
dotenv = "0.15.0"

然后在項(xiàng)目根目錄下創(chuàng)建.env文件：

API_KEY=1234567890

• 從.env文件加載環(huán)境變量：

use dotenv::dotenv;
use std::env;fn main() {dotenv().ok(); // Reads the .env filelet api_key = env::var("API_KEY");match api_key {Ok(val) => println!("API_KEY: {:?}", val),Err(e) => println!("Error API_KEY: {}", e),}//Simulate the execution of the rest of the programprintln!("...program continues...");
}

在上面的例子中，dotenv().ok()；行代碼從項(xiàng)目根目錄中的.env文件加載變量。然后，程序嘗試讀取API_KEY變量并打印其值，如果未設(shè)置則打印錯(cuò)誤。

最后總結(jié)

這只是在Rust中使用環(huán)境變量的三個(gè)基本示例。希望讓對(duì)你學(xué)習(xí)rust有點(diǎn)幫助，一起rust！