配置文件

使用axum构建todo服务
989587
2021/11/25 18:27:34
axum配置configtodo

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

本章代码在01/配置文件分支。

.env

WEB.ADDR=0.0.0.0:9527
PG.USER=axum_rs
PG.PASSWORD=axum.rs
PG.DBNAME=axum_rs
PG.PORT=5432
PG.HOST=pg.axum.rs
PG.POOL.MAX_SIZE=30

配置文件的结构体

首先，我们要在 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/handler.rs 文件，并定义一个 handler：

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

mod handler;

解析.env并初始化配置

现在可以在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();

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