配置文件

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

.env

先看一下.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 文件，并定义相关结构体：

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 服务

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

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

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