rust从0开始写项目-读取配置文件
一个项目初始化,总是有几个元素是必不可少的、框架、日志、配置文件等等基本元素。
今天我们主要介绍下怎么获取配置并在全局使用
更多好文。vx. golang技术实验室
专注分享 golang、rust等多语言、中间件及大数据相关内容
# 一、读取cargo.toml文件内容
`Cargo.toml` 文件配置如下:
```
[package]
name = "my_test"
version = "0.1.0"
edition = "2021"
authors = ["zhangql"]
```
代码:
```
pub fn cargo_file(){
let name = env!("CARGO_PKG_NAME");
let version = env!("CARGO_PKG_VERSION");
let authors = env!("CARGO_PKG_AUTHORS");
println!("{name}\n {version}\n {authors}")
}
my_test
0.1.0
zhangql
```
# 二、读取.env文件
项目[根目录](https://so.csdn.net/so/search?q=根目录&spm=1001.2101.3001.7020)新建 `.env` 文件,写入如下代码:
```
DATABASE_URL=mysql://postgres:123456@localhost:3306/test
```
Cargo.toml 文件配置导入以下第三方库:
```
[dependencies]
dotenv = "0.15.0"
```
main.rs
```
use dotenv::dotenv;
use std::env;
fn main() {
// 在访问环境变量之前检查一下,防止因读取环境变量失败导致程序恐慌。
// 先把 dotenv 导入,然后在程序开始的地方执行 dotenv() 函数即可,这就会从当前目录或父目录中的 .env 文件中加载环境变量。
// 如果你想指定其它路径,可以使用 crate 中提供的 from_filename 或 from_path 这两个函数。
// 好,那么调用 dotenv() 之后为什么还要调用 ok() 方法?
// 首先,dotenv() 返回的是 Result 类型,如果返回值不使用的话,就会发出一个警告:
// 调用 ok() 之后,会把 Result 转化为 Option,而 Option 就不会产生未使用 Result 的警告了。
// 那么,为什么不使用 unwrap()?
// 因为在生产环境中,你不会使用 .env 这个文件,你应该使用真实的环境变量,这时 dotenv() 函数就会加载失败,如果使用 unwrap(),那么你的程序就会停止运行。
// 所以这里使用 ok() 的目的就是当加载 dotenv 环境文件失败的时候可以忽略错误。
dotenv().ok();
let database_url = env::var("DATABASE_URL").expect("DATABASE_URL 没有在 .env 文件里设置");
print!("{:?}", database_url);
}
```
运行结果:
```
"mysql://postgres:123456@localhost:3306/test"
```
# 三、读取自定义toml文件
项目根目录新建 config.toml 文件,写入如下代码:
```
[database]
url = "tcp://mysql:123456@localhost/test"
[log]
debug = true
debug_sql = false
log_root = "/tmp"
```
Cargo.toml 文件配置导入以下第三方库:
```
[dependencies]
config = "0.13.1"
toml = "0.5.9"
lazy_static = "1.4"
serde = "1.0"
serde_derive = "1.0"
```
```
use std::fs::File;
use std::io::Read;
use serde::Deserialize;
#[macro_use]
extern crate lazy_static;
pub fn read_toml(){
let setting = Setting::default();
println!("{:#?}",setting);
let c = Setting::get();
println!("{:#?}",c);
}
#[derive(Debug,Deserialize)]
pub struct Log{
pub debug : bool,
pub debug_sql : bool,
pub log_root : String
}
#[derive(Debug,Deserialize)]
pub struct Database{
pub url : String
}
#[derive(Debug,Deserialize)]
pub struct Setting{
pub database : Database,
pub log : Log
}
impl Default for Setting{
fn default() -> Self {
let file_path = "/Users/zhangql/Desktop/rust/my_test/config.toml";
let mut file = match File::open(file_path){
Ok(f)=> f,
Err(e)=>panic!("error is op en config {e}")
};
let mut str = String::new();
println!("{}",str);
match file.read_to_string(&mut str) {
Ok(s)=>s,
Err(e)=>panic!("error read str {}",e)
};
toml::from_str(&str).expect("Parsing the configuration file failed")
}
}
impl Setting{
pub fn get<'a>()->&'a Self{
lazy_static!{
static ref CACHE : Setting = Setting::default();
}
&CACHE
}
}
```
````
Setting {
database: Database {
url: "tcp://mysql:123456@localhost/test",
},
log: Log {
debug: true,
debug_sql: false,
log_root: "/tmp",
},
}
Setting {
database: Database {
url: "tcp://mysql:123456@localhost/test",
},
log: Log {
debug: true,
debug_sql: false,
log_root: "/tmp",
},
}
````
现在,`CACHE`变量将在首次访问`Setting::get()`方法时进行初始化,并且后续的调用将重复使用已初始化的值。
请注意,为了使`lazy_static`宏正常工作,您需要在代码的顶部添加`#[macro_use] extern crate lazy_static;`。这使得宏在编译时能够正确展开。
使用这种方式,您可以通过调用`Setting::get()`方法来获取全局的`Setting`实例,而无需将其作为全局变量直接暴露出来。
参考文章:
https://blog.csdn.net/xiaohuihui1400/article/details/130801880
本文由[mdnice](https://mdnice.com/?platform=6)多平台发布