Rust读取配置文件的实现步骤

 更新时间:2023年12月08日 10:51:37   作者:一碗情深  
任何项目都离不开对于配置文件的读取和解析,rust项目也一样,本文主要介绍了Rust读取配置文件的实现步骤,具有一定的参考价值,感兴趣的可以了解一下

1. 读取Cargo.toml文件

Cargo.toml 文件配置如下:

[package] 
name = "axum" 
version = "0.1.0" 
authors = ["chh <1400152400@qq.com>"] 
edition = "2021"

main.rs

fn main() {
    let name = env!("CARGO_PKG_NAME");
    let version = env!("CARGO_PKG_VERSION");
    let author = env!("CARGO_PKG_AUTHORS");
    println!("{} {} {}", &name, &version, &author);
}

运行结果:

axum 0.1.0 chh <1400152400@qq.com>

2. 读取.env文件

项目根目录新建 .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<PathBuf> 类型,如果返回值不使用的话,就会发出一个警告:
    // 调用 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"

3. 读取自定义toml文件

项目根目录新建 config.toml 文件,写入如下代码:

[database]
url = "postgres://postgres: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"

新建 settings.rs ,写入如下代码:

use std::{fs::File, io::Read};
use lazy_static::lazy_static;
use serde::Deserialize;

#[derive(Debug, Deserialize)]
pub struct Database {
    pub url: String,
}

#[derive(Debug, Deserialize)]
pub struct Log {
    pub debug: bool,
    pub debug_sql: bool,
    pub log_root: String,
}

#[derive(Debug, Deserialize)]
pub struct Settings {
    pub database: Database,
    pub log: Log,
}

impl Default for Settings {
    fn default() -> Self {
        let file_path = "config.toml";
        let mut file = match File::open(file_path) {
            Ok(f) => f,
            Err(e) => panic!("no such file {} exception:{}", file_path, e)
        };
        let mut str_val = String::new();
        match file.read_to_string(&mut str_val) {
            Ok(s) => s,
            Err(e) => panic!("Error Reading file: {}", e)
        };
        toml::from_str(&str_val).expect("Parsing the configuration file failed");
    }
}

impl Settings {
    pub fn get<'a>() -> &'a Self {
        // 给静态变量延迟赋值的宏
        lazy_static! {
            static ref CACHE: Settings = Settings::default();
        }
        &CACHE
    }
}

main.rs 代码如下:

use crate::settings::Settings;

mod settings;

fn main() {
    let setting = Settings::get();
    println!("{:?}", setting.database.url);
    println!("{:?}", setting.log);
}

运行结果:

"postgres://postgres:123456@localhost/test"
Log { debug: true, debug_sql: false, log_root: "/tmp" }

到此这篇关于Rust读取配置文件的实现步骤的文章就介绍到这了,更多相关Rust读取配置文件内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

相关文章

  • rust 如何使用 cargo-nextest 替代 cargo test

    rust 如何使用 cargo-nextest 替代 cargo te

    cargo-nextest 是新一代的rust测试程序,能够极大提升测试性能,可以完全替代 cargo test 命令,这篇文章主要介绍了rust 如何使用 cargo-nextest 替代 cargo test,需要的朋友可以参考下
    2024-05-05
  • Rust anyhow 简明示例教程

    Rust anyhow 简明示例教程

    anyhow 是 Rust 中的一个库,旨在提供灵活的、具体的错误处理能力,建立在 std::error::Error 基础上,主要用于那些需要简单错误处理的应用程序和原型开发中,本文给大家分享Rust anyhow 简明教程,一起看看吧
    2024-06-06
  • 关于使用rust调用c++静态库并编译nodejs包的问题

    关于使用rust调用c++静态库并编译nodejs包的问题

    这篇文章主要介绍了使用rust调用c++静态库并编译nodejs包的问题,本文通过实例代码给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
    2022-08-08
  • rust的package,crate,module示例解析

    rust的package,crate,module示例解析

    rust提供了非常优秀的包管理器cargo,我们可以使用crate,module,package来组织代码,这篇文章主要介绍了rust的package,crate,module相关知识,本文给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
    2023-04-04
  • Rust实现构建器模式和如何使用Bon库中的构建器

    Rust实现构建器模式和如何使用Bon库中的构建器

    这篇文章主要介绍了Rust实现构建器模式和如何使用Bon库中的构建器,本文给大家介绍的非常详细,感兴趣的朋友跟随小编一起看看吧
    2024-08-08
  • Rust duckdb和polars读csv文件比较情况

    Rust duckdb和polars读csv文件比较情况

    duckdb在数据分析上,有非常多不错的特质,1、快;2、客户体验好,特别是可以同时批量读csv在一个目录下的csv等文件,今天来比较下Rust duckdb和polars读csv文件比较的情况,感兴趣的朋友一起看看吧
    2024-06-06
  • Rust遍历 BinaryHeap的示例代码

    Rust遍历 BinaryHeap的示例代码

    Rust 的 BinaryHeap 结构体实现了迭代器接口,因此你可以遍历它,如果你想要遍历 BinaryHeap 中的所有元素,你可以使用 .into_iter() 方法将其转换为迭代器,并遍历其中的元素,本文通过实例介绍Rust遍历 BinaryHeap的相关知识,感兴趣的朋友一起看看吧
    2024-04-04
  • Rust中的不安全代码详解

    Rust中的不安全代码详解

    这篇文章主要为大家介绍了Rust中的不安全代码详解,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
    2023-04-04
  • Rust使用Sqlx连接Mysql的实现

    Rust使用Sqlx连接Mysql的实现

    数据库在编程中是一个很重要的环节,本文主要介绍了Rust使用Sqlx连接Mysql的实现,记录rust如何操作数据库并以mysql为主的做简单的使用说明,感兴趣的可以了解一下
    2024-03-03
  • 一步到位,教你如何在Windows成功安装Rust

    一步到位,教你如何在Windows成功安装Rust

    一步到位:轻松学会在Windows上安装Rust!想快速掌握Rust编程语言?别再为复杂教程头疼!这份指南将手把手带你顺利完成Windows平台上的Rust安装全过程,从此编码之旅更加顺畅无阻,立即阅读,开始你的Rust编程旅程吧!
    2024-01-01

最新评论