配置文件

989228
2021/11/25 18:27:34

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

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

.env

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

配置文件的结构体

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

mod config;
/// 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:

pub async fn usage() -> &'static str {
    r#"
        ### USAGE ###
        - GET /todo -- get all todo list
        - POST /todo -- create a todo list
        - GET /todo/:list_id -- get detail for a todo list
        - DELETE /todo/:list_id -- delete a todo list, include it's items
        - PUT /todo/:list_id -- edit a todo list
        - GET /todo/:list_id/items -- get items from todo list
        - GET /todo/:list_id/items/:item_id -- get detail for a todo item
        - PUT /todo/:list_id/items/:item_id -- edit a todo item(set the item to checked)
        - DELETE /todo/:list_id/items/:item_id -- delete a todo item
    "#
}

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

mod handler;

解析.env并初始化配置

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

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

创建 axum 服务

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

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