tokio-rs/axum
1
0
Fork 0
mirror of https://github.com/tokio-rs/axum.git synced 2026-08-19 00:00:14 +02:00
Code Issues Packages Projects Releases Wiki Activity

Use JsonRejection::{status, body_text} in customize extractor error example (#1790)

Browse Source
This commit is contained in:
David Pedersen
2023-02-25 15:02:02 +01:00
committed by GitHub
parent 37e2a7d5e7
commit 1dc4b44472
3 changed files with 17 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files
examples/customize-extractor-error/src/with_rejection.rs
+11 -11
View File
@@ -13,7 +13,7 @@
//! [`thiserror`]: https://crates.io/crates/thiserror
//! [#1116]: https://github.com/tokio-rs/axum/issues/1116#issuecomment-1186197684
use axum::{extract::rejection::JsonRejection, http::StatusCode, response::IntoResponse, Json};
use axum::{extract::rejection::JsonRejection, response::IntoResponse, Json};
use axum_extra::extract::WithRejection;
use serde_json::{json, Value};
use thiserror::Error;
@@ -37,21 +37,21 @@ pub enum ApiError {
#[error(transparent)]
JsonExtractorRejection(#[from] JsonRejection),
}
// We implement `IntoResponse` so ApiError can be used as a response
impl IntoResponse for ApiError {
fn into_response(self) -> axum::response::Response {
let (status, message) = match self {
ApiError::JsonExtractorRejection(json_rejection) => {
(json_rejection.status(), json_rejection.body_text())
}
};
let payload = json!({
"message": self.to_string(),
"message": message,
"origin": "with_rejection"
});
let code = match self {
ApiError::JsonExtractorRejection(x) => match x {
JsonRejection::JsonDataError(_) => StatusCode::UNPROCESSABLE_ENTITY,
JsonRejection::JsonSyntaxError(_) => StatusCode::BAD_REQUEST,
JsonRejection::MissingJsonContentType(_) => StatusCode::UNSUPPORTED_MEDIA_TYPE,
_ => StatusCode::INTERNAL_SERVER_ERROR,
},
};
(code, Json(payload)).into_response()
(status, Json(payload)).into_response()
}
}