Add example for implementing IntoResponseParts (#825)

This commit is contained in:
David Pedersen 2022-03-04 10:53:17 +01:00 committed by GitHub
parent 1614ef7a22
commit ae22154979
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -9,6 +9,67 @@ use std::{
}; };
/// Trait for adding headers and extensions to a response. /// Trait for adding headers and extensions to a response.
///
/// # Example
///
/// ```rust
/// use axum::{
/// response::{ResponseParts, IntoResponse, IntoResponseParts, Response},
/// http::{StatusCode, header::{HeaderName, HeaderValue}},
/// };
///
/// // Hypothetical helper type for setting a single header
/// struct SetHeader<'a>(&'a str, &'a str);
///
/// impl<'a> IntoResponseParts for SetHeader<'a> {
/// type Error = (StatusCode, String);
///
/// fn into_response_parts(self, mut res: ResponseParts) -> Result<ResponseParts, Self::Error> {
/// match (self.0.parse::<HeaderName>(), self.1.parse::<HeaderValue>()) {
/// (Ok(name), Ok(value)) => {
/// res.headers_mut().insert(name, value);
/// },
/// (Err(_), _) => {
/// return Err((
/// StatusCode::INTERNAL_SERVER_ERROR,
/// format!("Invalid header name {}", self.0),
/// ));
/// },
/// (_, Err(_)) => {
/// return Err((
/// StatusCode::INTERNAL_SERVER_ERROR,
/// format!("Invalid header value {}", self.1),
/// ));
/// },
/// }
///
/// Ok(res)
/// }
/// }
///
/// // Its also recommended to implement `IntoResponse` so `SetHeader` can be used on its own as
/// // the response
/// impl<'a> IntoResponse for SetHeader<'a> {
/// fn into_response(self) -> Response {
/// // This gives an empty response with the header
/// (self, ()).into_response()
/// }
/// }
///
/// // We can now return `SetHeader` in responses
/// async fn handler() -> impl IntoResponse {
/// (
/// SetHeader("server", "axum"),
/// SetHeader("x-foo", "custom"),
/// "body",
/// )
/// }
///
/// // Or on its own as the whole response
/// async fn other_handler() -> impl IntoResponse {
/// SetHeader("x-foo", "custom")
/// }
/// ```
pub trait IntoResponseParts { pub trait IntoResponseParts {
/// The type returned in the event of an error. /// The type returned in the event of an error.
/// ///