在Axum Rust中，內容協商是通過使用Accept頭字段來實現的。當客戶端發送請求時，它會在請求頭中包含一個Accept字段，用于指定它期望接收的內容類型。服務器根據這個字段來決定返回哪種內容類型的數據。
要在Axum Rust中進行內容協商，你需要使用tower-http庫中的ContentNegotiation特性。首先，在你的Cargo.toml文件中添加以下依賴：
[dependencies]
axum = "0.6"
tower-http = "0.2"
接下來，在你的Axum應用中配置內容協商。這里有一個簡單的例子：
use axum::prelude::*;
use tower_http::content_negotiation::{ContentNegotiation, Negotiated};
use tower_http::service::{make_service_fn, service_fn};
use std::convert::Infallible;
async fn handle(req: Request<Body>) -> Result<Response<Negotiated>, Infallible> {
// 獲取請求頭中的Accept字段
let accept = req.headers().get("Accept").unwrap().to_str().unwrap();
// 根據Accept字段選擇合適的內容類型
let content_type = match accept {
"application/json" => "application/json",
"application/xml" => "application/xml",
_ => "application/octet-stream",
};
// 創建一個Negotiated響應
let response = Response::builder()
.status(200)
.header("Content-Type", content_type)
.body(Body::from("Hello, world!"))
.expect("Failed to build response");
Ok(response)
}
#[tokio::main]
async fn main() {
// 創建一個內容協商中間件
let negotiation = ContentNegotiation::new(vec![
("application/json", serde_json::MediaType::parse("application/json").unwrap()),
("application/xml", tower_http::mime::XML.parse().unwrap()),
]);
// 創建一個Axum服務
let make_svc = make_service_fn(|_conn| async {
Ok::<_, Infallible>(service_fn(handle))
});
// 將內容協商中間件應用到Axum服務
let app = Axum::new()
.layer(tower_http::middleware::ContentNegotiationLayer::new(negotiation))
.serve(make_svc);
// 運行Axum應用
if let Err(e) = app.await {
eprintln!("server error: {}", e);
}
}
在這個例子中，我們首先從請求頭中獲取Accept字段，然后根據這個字段的值選擇合適的內容類型。接下來，我們創建一個Negotiated響應，并將其發送給客戶端。最后，我們將內容協商中間件應用到Axum服務上。
