Rust环境变量管理库set_env的使用：高效设置和操作进程环境变量

set_env是一个用于高效管理进程环境变量的Rust库，它提供了比标准库更便捷的API来设置、获取和操作环境变量。

主要功能

1. 设置、获取和删除环境变量
2. 批量操作环境变量
3. 临时修改环境变量（不影响系统环境）
4. 环境变量存在性检查
5. 环境变量迭代

安装

在Cargo.toml中添加依赖：

[dependencies]
set_env = "0.3"

基本使用方法

设置和获取环境变量

use set_env::set_var;

fn main() {
    // 设置环境变量
    set_var("MY_VAR", "some_value").unwrap();
    
    // 获取环境变量
    let value = std::env::var("MY_VAR").unwrap();
    println!("MY_VAR: {}", value); // 输出: MY_VAR: some_value
}

检查环境变量是否存在

use set_env::is_present;

fn main() {
    if is_present("MY_VAR") {
        println!("MY_VAR exists");
    } else {
        println!("MY_VAR does not exist");
    }
}

删除环境变量

use set_env::remove_var;

fn main() {
    remove_var("MY_VAR").unwrap();
    println!("MY_VAR removed");
}

高级用法

批量设置环境变量

use set_env::{set_vars, remove_vars};

fn main() {
    // 批量设置
    set_vars(&[("VAR1", "value1"), ("VAR2", "value2")]).unwrap();
    
    // 批量删除
    remove_vars(&["VAR1", "VAR2"]).unwrap();
}

临时修改环境变量

use set_env::with_temp_env;

fn main() {
    // 临时设置环境变量，只在闭包内有效
    with_temp_env(&[("TEMP_VAR", "temp_value")], || {
        let value = std::env::var("TEMP_VAR").unwrap();
        println!("Inside closure: {}", value); // 输出: Inside closure: temp_value
    });
    
    // 闭包外环境变量已恢复
    let result = std::env::var("TEMP_VAR");
    println!("Outside closure: {:?}", result); // 输出: Outside closure: Err(NotPresent)
}

环境变量迭代

use set_env::vars;

fn main() {
    // 获取所有环境变量
    for (key, value) in vars() {
        println!("{}: {}", key, value);
    }
}

错误处理

所有操作都返回Result类型，可以正确处理错误情况：

use set_env::set_var;

fn main() {
    match set_var("INVALID_NAME=", "value") {
        Ok(_) => println!("Set successfully"),
        Err(e) => println!("Error: {}", e), // 会输出错误，因为变量名包含非法字符
    }
}

性能考虑

set_env库在内部进行了优化，比直接使用标准库的std::env::set_var更高效，特别是在批量操作时。

注意事项

1. 环境变量修改只影响当前进程及其子进程
2. 在Windows和Unix-like系统上行为一致
3. 变量名通常不应包含等号(=)

这个库非常适合需要频繁操作环境变量的场景，如测试框架、开发工具和配置管理系统等。

完整示例demo

下面是一个结合了set_env库多种功能的完整示例：

use set_env::{
    set_var, set_vars, remove_var, remove_vars, 
    is_present, vars, with_temp_env
};

fn main() {
    // 1. 基本设置和获取
    set_var("APP_NAME", "MyRustApp").unwrap();
    let app_name = std::env::var("APP_NAME").unwrap();
    println!("Application name: {}", app_name);

    // 2. 检查存在性
    if is_present("APP_NAME") {
        println!("APP_NAME exists");
    }

    // 3. 批量操作
    set_vars(&[
        ("DB_HOST", "localhost"),
        ("DB_PORT", "5432"),
        ("DB_USER", "admin")
    ]).unwrap();

    // 4. 临时环境变量
    with_temp_env(&[("DEBUG_MODE", "true")], || {
        if is_present("DEBUG_MODE") {
            println!("Running in debug mode");
        }
    });

    // 5. 迭代所有环境变量
    println!("\nCurrent environment variables:");
    for (key, value) in vars() {
        println!("{} = {}", key, value);
    }

    // 6. 清理环境变量
    remove_vars(&["APP_NAME", "DB_HOST", "DB_PORT", "DB_USER"]).unwrap();
    
    // 验证是否已删除
    println!("\nAfter removal:");
    println!("APP_NAME exists: {}", is_present("APP_NAME"));
}

这个完整示例展示了：

• 基本环境变量操作（设置、获取、检查、删除）
• 批量环境变量操作
• 临时环境变量的使用
• 环境变量迭代功能
• 错误处理（通过unwrap处理）

运行此程序将演示set_env库的主要功能，并展示如何在实际项目中综合使用这些功能。