配置文件

本章我们将实现配置文件的加载。我们将对 axum 的监听地址和 PostgreSQL 相关的信息进行配置,并保存到.env文件中。

.env

先看一下.env文件的内容:

配置文件的结构体

首先,我们要在 src/main.rs 声明一个 config 模块:

mod config;

然后,创建一个 src/config.rs 文件,并定义相关结构体:

/// Web 配置
#[derive(Deserialize)]
pub struct WebConfig {
    pub addr: String,
}

/// 应用配置
#[derive(Deserialize)]
pub struct Config {
    pub web: WebConfig,
}

最后,实现从环境变量中初始化配置的方法:

impl Config {
    /// 从环境变量中初始化配置
    pub fn from_env() -> Result<Self, config::ConfigError> {
        let mut cfg = config::Config::new();
        cfg.merge(config::Environment::new())?;
        cfg.try_into()
    }
}

handler 模块

别忘了在 src/main.rs 声明这个模块:

mod handler;

现在可以在main函数中解析.env并初始化配置了:

// 解析 .env 文件
dotenv().ok();
// 初始化配置
let cfg = config::Config::from_env().expect("初始化配置失败");

创建 axum 服务

接下来,我们可以创建 axum 服务了

// 路由
let app = Router::new().route("/", get(handler::usage));
// 绑定到配置文件设置的地址
axum::Server::bind(&cfg.web.addr.parse().unwrap())
    .serve(app.into_make_service())
    .await
    .unwrap();

下一章,我们将进行错误处理。

要查看完整内容,请先登录