ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

Rust 使用 dotenv 来设置环境变量

2022-01-22 08:00:14  阅读:208  来源: 互联网

标签:unwrap DB var env dotenv 环境变量 Rust


在项目中,我们通常需要设置一些环境变量,用来保存一些凭证或其它数据,这时我们可以使用 dotenv 这个 crate。

 

首先在项目中添加 dotenv 这个依赖:

7 
8 
9 
dependencies] 
dotenv = " 
o. 15.0"

 

例如在下面这个项目中,需要设置数据库连接字符串和 Debug 等级这两个环境变量。在开发环境下,我们可以在项目根目录下创建 .env 这个文件:

EXPLORER 
v OPEN EDITORS 
x .env 
v LEARN 
v src 
@ main.rs 
target 
h! env 
.gitignore 
Cargo.lock 
Cargo.toml

 

在 .env 文件里,我们设置两个环境变量,分别是 DB_URL 和 LOG_LEVEL:

:argo.toml U 
.env U X 
.env 
1 
2 
DB_URL=postgres : // user : pwd@locathost : 5432/mydb 
LOG LEVEL-DEBUGI

 

下面来到 main.rs,想要访问系统的环境变量,我们使用标准库的 std::env 即可:

main.rs > 
use std env; 
fn main() { 
for (k, v) in env::vars() { 
print In! ("PATH: { } " , 
env :: var( " PATH" ) . unwrap( ) ) ; 
print In! ("DB: {I" , 
env :: var( "DB_URL"). unwrap()); 
println!("LOG: O" , 
env :: var( " LOG _ LEVEL " ) . unwrap( ) ) ;

这里,我们先把获取到的环境变量进行遍历和打印,然后再分别尝试打印 PATH、DB_URL 和 LOG_LEVEL 这三个环境变量的值。

 

结果如下:

USER: solenovex 
XPC FLAGS: exe 
XPC SERVICE NAME: e 
/Users/s01enovex/ . cargo/ bin/ cargo 
CE CONDA: 
CE M: 
_CFBund1eIdentifier: com.apple.Terminal 
CF USER TEXT ENCODING: 
PATH: /Users/s01enovex/.cargo/bin: / Users/ solenovex/opt/anaconda3 
bin : /Users/s01enovex/opt/anaconda3/condabin : /Users/s01enovex/ . car 
go/bin: /usr/local/bin : /usr/bin : / bin : /usr/sbin : /sbin : /usr/local/go 
/ bin : /usr/local/share/dotnet : N/ . dotnet/tools : / Users/ solenovex/go/ 
bin 
thread Imainl panicked at I called 
on an 'Err' 
src/main.rs:9:43 
value: NotPresentI, 
note: run with 'RUST BACKTRACE-I' 
environment variable to display 
a backtrace

我们可以看到,前面是遍历打印的环境变量,后边打印出了 PATH 的值。然后在尝试获取 DB_URL 值的时候程序就恐慌了,因为没有找到这个环境变量。

 

现在我们使用 dotenv 这个 crate:

use std ::env; 
use dotenv dotenv; 
fn main() { 
dotenv().ok(); 
for (k, v) in env:: vars() { 
println ! ( " { } : 
print In! ( 
print In! ( 
print In! ( 
" PATH: 0" , 
env :: var( " PATH" ) . unwrap( ) ) ; 
"DB: 
env :: var( "DB_URL") . unwrap( ) ) ; 
" LOG: 0", 
env :: var( " LOG _ LEVEL " ) . unwrap( ) ) ;

先把 dotenv 导入,然后在程序开始的地方执行 dotenv() 函数即可,这就会从当前目录或父目录中的 .env 文件中加载环境变量。

如果你想指定其它路径,可以使用 crate 中提供的 from_filename 或 from_path 这两个函数。

 

好,那么调用 dotenv() 之后为什么还要调用 ok() 方法?

首先,dotenv() 返回的是 Result<PathBuf> 类型,如果返回值不使用的话,就会发出一个警告:

fn main() { 
dotenv(); 
for 
print In! ( 
vars() {

调用 ok() 之后,会把 Result 转化为 Option,而 Option 就不会产生未使用 Result 的警告了。

 

那么,为什么不使用 unwrap()?

因为在生产环境中,你不会使用 .env 这个文件,你应该使用真实的环境变量,这时 dotenv() 函数就会加载失败,如果使用 unwrap(),那么你的程序就会停止运行。

所以这里使用 ok() 的目的就是当加载 dotenv 环境文件失败的时候可以忽略错误。

 

代码:

 1 use std::env;
 2 use dotenv::dotenv;
 3 
 4 fn main() {
 5     dotenv().ok();
 6 
 7     for (k, v) in env::vars() {
 8         println!("{}: {}", k, v);
 9     }
10 
11     println!("PATH: {}", env::var("PATH").unwrap());
12     println!("DB: {}", env::var("DB_URL").unwrap());
13     println!("LOG: {}", env::var("LOG_LEVEL").unwrap());
14 }

 

标签:unwrap,DB,var,env,dotenv,环境变量,Rust
来源: https://www.cnblogs.com/yangxu-pro/p/15831888.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有