axum中的各种响应

1375387
2021/11/13 06:05:25

本章主要讨论 axum 的响应。axum 已经实现了多种响应,比如纯文本、HTML、JSON 及 自定义响应头(response header)。除了这些 axum 内置的响应之外,我们还将讨论如何将自己定义的结构体,作为响应返回给客户端。

axum 的响应

axum 有句话说的是:

Anything that implements IntoResponse can be returned from a handler

大意就是:凡是实现了 IntoResponse trait 的,都可作为 handler 的返回值,也就是我们所说的响应

官方已经对若干常用数据类型实现了该 trait,亦是说,这些数据类型可以直接作为请求的响应进行返回。

对于我们自己的数据类型,比如自定义的 struct,也只需要实现 IntoResponse,就可以直接作为响应进行返回了。

官方提供了示例,我们将结合这个示例对官方提供的响应进行说明。

官方提供的响应

官方提供了以下响应:

类型数据类型
纯文本&strString
字节序列Vec<u8>
空响应()
仅有状态码StatusCode
带响应头的响应(HeaderMap, IntoResponse)
带响应头和状态的响应(StatusCode, HeaderMap, IntoResponse)
HTML 响应Html<IntoResponse>
JSON 响应Json<IntoResponse>
Result响应Result<T:IntoResponse, E:IntoResponse>

纯文本之 &str

async fn str_response() -> &'static str {
    "Hello, axum.rs"
}

纯文本之 String

async fn string_response() -> String{
    "Hello, axum.rs".to_string()
}

仅有状态码

async fn not_found()  -> StatusCode {
    StatusCode::NOT_FOUND
}

使用 curl 访问:

$ curl -I http://127.0.0.1:9527/404
HTTP/1.1 404 Not Found
content-length: 0
date: Mon, 15 Nov 2021 08:14:16 GMT

带响应头的响应

async fn with_headers() -> (HeaderMap, &'static str) {
    let mut headers = HeaderMap::new();
    headers.insert(
        HeaderName::from_static("x-powered");
        HeaderValue::from_static("axum.rs");
    );
    (headers, "axum.rs")
}

使用 curl 访问:

$ curl -I http://127.0.0.1:9527/with_headers
HTTP/1.1 200 OK
content-type: text/plain
x-powered: axum.rs
content-length: 0
date: Mon, 15 Nov 2021 08:18:00 GMT

带响应头和状态的响应

async fn with_headers_and_status() -> (StatusCode, HeaderMap, &'static str) {
    let mut headers = HeaderMap::new();
    headers.insert(
        HeaderName::from_static("x-powered");
        HeaderValue::from_static("axum.rs");
    );
    (StatusCode::OK, headers, "axum.rs")
}

使用 curl 访问:

$ curl -i http://127.0.0.1:9527/with_headers_and_status
HTTP/1.1 200 OK
content-type: text/plain
x-powered: axum.rs
content-length: 7
date: Mon, 15 Nov 2021 08:22:56 GMT

axum.rs

HTML 响应

async fn html() -> Html<&'static str> {
    Html("Hello, <em>axum.rs</em>")
}

JSON 响应

async fn json() -> Json<serde_json::Value> {
    Json(serde_json::json!({"hello":"axum.rs"}))
}

Result 响应

// Result<T,E> 要求T和E均实现了IntoResponse
async fn result() -> Result<&'static str, StatusCode> {
    let flag = false;
    if flag {
        Ok("Hello, axum.rs")
    } else {
        Err(StatusCode::INTERNAL_SERVER_ERROR)
    }
}

flag==false 时,使用 curl 访问:

$ curl -i http://127.0.0.1:9527/result
HTTP/1.1 500 Internal Server Error
content-length: 0
date: Mon, 15 Nov 2021 08:35:03 GMT

自定义结构体响应

响应自定义错误

自定义一个错误类型

pub struct AppError {
    pub message: String,
}

让这个自定义错误类型实现 IntoResponse

impl IntoResponse for AppError {
    type Body = Full<Bytes>;
    type BodyError = Infallible;

    fn into_response(self) -> axum::http::Response<Self::Body> {
        self.message.into_response()
    }
}

响应这个错误类型

async fn app_error() -> Result<&'static str, AppError> {
    let flag = false;
    if flag {
        Ok("Hello, axum.rs")
    } else {
        Err(AppError {
            message: "Opps!".to_string(),
        })
    }
}

本章讨论了 axum 内置的各种响应及如何响应自定义结构体和自定义错误,本章源码你可以在代码仓库中找到。

如果细心的话,你会发现在上面的例子中(axum 默认情况下),中文大概率会显示成乱码,利用本章所学知识,你能解决这个问题吗?请自行思考再看答案。

提示:HTTP 中的 content-type header 可以指定响应的类型和编码。

参考答案:

async fn cn() -> (HeaderMap, &'static str) {
    let mut headers = HeaderMap::new();
    headers.insert(
        HeaderName::from_static("content-type"),
        HeaderValue::from_static("text/plain;charset=utf-8"),
    );
    (headers, "你好,axum.rs")
}