itsscb/axum
1
0
Fork 0
mirror of https://github.com/tokio-rs/axum.git synced 2025-09-30 22:41:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

axum-extra/scheme: Use rejection macros for SchemeMissing rejection (#3113)

Browse Source
This commit is contained in:
Tobias Bieniek 2024-12-27 10:01:59 +01:00 committed by GitHub
parent 8def6782fb
commit 8e78a37660
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
axum-extra/src/extract/scheme.rs
View File

@ -1,10 +1,8 @@
//! Extractor that parses the scheme of a request. //! Extractor that parses the scheme of a request.
//! See [`Scheme`] for more details. //! See [`Scheme`] for more details.
use axum::{ use axum::extract::FromRequestParts;
extract::FromRequestParts, use axum_core::__define_rejection as define_rejection;
response::{IntoResponse, Response},
};
use http::{ use http::{
header::{HeaderMap, FORWARDED}, header::{HeaderMap, FORWARDED},
request::Parts, request::Parts,
@ -23,15 +21,12 @@ const X_FORWARDED_PROTO_HEADER_KEY: &str = "X-Forwarded-Proto";
#[derive(Debug, Clone)] #[derive(Debug, Clone)]
pub struct Scheme(pub String); pub struct Scheme(pub String);
/// Rejection type used if the [`Scheme`] extractor is unable to define_rejection! {
/// resolve a scheme. #[status = BAD_REQUEST]
#[derive(Debug)] #[body = "No scheme found in request"]
pub struct SchemeMissing; /// Rejection type used if the [`Scheme`] extractor is unable to
/// resolve a scheme.
impl IntoResponse for SchemeMissing { pub struct SchemeMissing;
fn into_response(self) -> Response {
(http::StatusCode::BAD_REQUEST, "No scheme found in request").into_response()
}
} }
impl<S> FromRequestParts<S> for Scheme impl<S> FromRequestParts<S> for Scheme