Rust Axum 路由处理函数神奇参数特性揭秘

2023-10-01 17:51:25

Rust Axum 框架中有一个奇妙的特性，就是路由处理函数可以支持不同的参数，并且可以根据需要按需提取请求字段。这在其他框架中通常需要大量的代码才能实现，但在 Axum 中却非常简单。

为了理解这个特性是如何工作的，我们来看一个简单的例子。假设我们有一个 API 端点，它接受一个查询参数 name 和一个路径参数 id。我们可以使用以下代码来定义这个端点：

use axum::{routing::get, Router};

#[get("/users/:id")]
async fn get_user(id: String) -> String {
    format!("Hello, {}!", id)
}

#[tokio::main]
async fn main() {
    let app = Router::new().route("/users/:id", get(get_user));

    axum::Server::bind(&"0.0.0.0:3000").serve(app).await.unwrap();
}

在这个例子中，get_user 函数接收一个 id 参数，它是从路径中提取的。我们也可以使用查询参数，如下所示：

use axum::{extract::Query, routing::get, Router};

#[get("/users")]
async fn get_users(Query(params): Query<HashMap<String, String>>) -> String {
    let name = params.get("name").unwrap_or(&"".to_string());
    format!("Hello, {}!", name)
}

#[tokio::main]
async fn main() {
    let app = Router::new().route("/users", get(get_users));

    axum::Server::bind(&"0.0.0.0:3000").serve(app).await.unwrap();
}