本章我们将实现配置文件的加载。我们将对 axum 的监听地址和 PostgreSQL 相关的信息进行配置,并保存到.env
文件中。
.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
文件,并定义相关结构体:
/// Web 配置
#[derive(Deserialize)]
pub struct WebConfig {
pub addr: String,
}
/// 应用配置
#[derive(Deserialize)]
pub struct Config {
pub web: WebConfig,
}
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();
下一章,我们将进行错误处理。