1
0
mirror of https://github.com/fafhrd91/actix-web synced 2024-11-28 01:52:57 +01:00

improve NormalizePath docs (#1839)

This commit is contained in:
Rob Ede 2020-12-23 00:19:20 +00:00 committed by GitHub
parent 4dccd092f3
commit 05f104c240
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -1,10 +1,11 @@
//! `Middleware` to normalize request's URI //! For middleware documentation, see [`NormalizePath`].
use std::task::{Context, Poll}; use std::task::{Context, Poll};
use actix_http::http::{PathAndQuery, Uri}; use actix_http::http::{PathAndQuery, Uri};
use actix_service::{Service, Transform}; use actix_service::{Service, Transform};
use bytes::Bytes; use bytes::Bytes;
use futures_util::future::{ok, Ready}; use futures_util::future::{ready, Ready};
use regex::Regex; use regex::Regex;
use crate::service::{ServiceRequest, ServiceResponse}; use crate::service::{ServiceRequest, ServiceResponse};
@ -17,10 +18,12 @@ pub enum TrailingSlash {
/// Always add a trailing slash to the end of the path. /// Always add a trailing slash to the end of the path.
/// This will require all routes to end in a trailing slash for them to be accessible. /// This will require all routes to end in a trailing slash for them to be accessible.
Always, Always,
/// Only merge any present multiple trailing slashes. /// Only merge any present multiple trailing slashes.
/// ///
/// Note: This option provides the best compatibility with the v2 version of this middlware. /// Note: This option provides the best compatibility with the v2 version of this middleware.
MergeOnly, MergeOnly,
/// Trim trailing slashes from the end of the path. /// Trim trailing slashes from the end of the path.
Trim, Trim,
} }
@ -32,28 +35,53 @@ impl Default for TrailingSlash {
} }
#[derive(Default, Clone, Copy)] #[derive(Default, Clone, Copy)]
/// `Middleware` to normalize request's URI in place /// Middleware to normalize a request's path so that routes can be matched less strictly.
/// ///
/// Performs following: /// # Normalization Steps
/// /// - Merges multiple consecutive slashes into one. (For example, `/path//one` always
/// - Merges multiple slashes into one. /// becomes `/path/one`.)
/// - Appends a trailing slash if one is not present, removes one if present, or keeps trailing /// - Appends a trailing slash if one is not present, removes one if present, or keeps trailing
/// slashes as-is, depending on the supplied `TrailingSlash` variant. /// slashes as-is, depending on which [`TrailingSlash`] variant is supplied
/// to [`new`](NormalizePath::new()).
/// ///
/// # Default Behavior
/// The default constructor chooses to strip trailing slashes from the end
/// ([`TrailingSlash::Trim`]), the effect is that route definitions should be defined without
/// trailing slashes or else they will be inaccessible.
///
/// # Example
/// ```rust /// ```rust
/// use actix_web::{web, http, middleware, App, HttpResponse}; /// use actix_web::{web, middleware, App};
/// ///
/// # fn main() { /// # #[actix_rt::test]
/// # async fn normalize() {
/// let app = App::new() /// let app = App::new()
/// .wrap(middleware::NormalizePath::default()) /// .wrap(middleware::NormalizePath::default())
/// .service( /// .route("/test", web::get().to(|| async { "test" }))
/// web::resource("/test") /// .route("/unmatchable/", web::get().to(|| async { "unmatchable" }));
/// .route(web::get().to(|| HttpResponse::Ok())) ///
/// .route(web::method(http::Method::HEAD).to(|| HttpResponse::MethodNotAllowed())) /// use actix_web::http::StatusCode;
/// ); /// use actix_web::test::{call_service, init_service, TestRequest};
///
/// let mut app = init_service(app).await;
///
/// let req = TestRequest::with_uri("/test").to_request();
/// let res = call_service(&mut app, req).await;
/// assert_eq!(res.status(), StatusCode::OK);
///
/// let req = TestRequest::with_uri("/test/").to_request();
/// let res = call_service(&mut app, req).await;
/// assert_eq!(res.status(), StatusCode::OK);
///
/// let req = TestRequest::with_uri("/unmatchable").to_request();
/// let res = call_service(&mut app, req).await;
/// assert_eq!(res.status(), StatusCode::NOT_FOUND);
///
/// let req = TestRequest::with_uri("/unmatchable/").to_request();
/// let res = call_service(&mut app, req).await;
/// assert_eq!(res.status(), StatusCode::NOT_FOUND);
/// # } /// # }
/// ``` /// ```
pub struct NormalizePath(TrailingSlash); pub struct NormalizePath(TrailingSlash);
impl NormalizePath { impl NormalizePath {
@ -76,11 +104,11 @@ where
type Future = Ready<Result<Self::Transform, Self::InitError>>; type Future = Ready<Result<Self::Transform, Self::InitError>>;
fn new_transform(&self, service: S) -> Self::Future { fn new_transform(&self, service: S) -> Self::Future {
ok(NormalizePathNormalization { ready(Ok(NormalizePathNormalization {
service, service,
merge_slash: Regex::new("//+").unwrap(), merge_slash: Regex::new("//+").unwrap(),
trailing_slash_behavior: self.0, trailing_slash_behavior: self.0,
}) }))
} }
} }
@ -160,9 +188,11 @@ mod tests {
use actix_service::IntoService; use actix_service::IntoService;
use super::*; use super::*;
use crate::dev::ServiceRequest; use crate::{
use crate::test::{call_service, init_service, TestRequest}; dev::ServiceRequest,
use crate::{web, App, HttpResponse}; test::{call_service, init_service, TestRequest},
web, App, HttpResponse,
};
#[actix_rt::test] #[actix_rt::test]
async fn test_wrap() { async fn test_wrap() {
@ -244,7 +274,7 @@ mod tests {
} }
#[actix_rt::test] #[actix_rt::test]
async fn keep_trailing_slash_unchange() { async fn keep_trailing_slash_unchanged() {
let mut app = init_service( let mut app = init_service(
App::new() App::new()
.wrap(NormalizePath(TrailingSlash::MergeOnly)) .wrap(NormalizePath(TrailingSlash::MergeOnly))
@ -279,7 +309,7 @@ mod tests {
async fn test_in_place_normalization() { async fn test_in_place_normalization() {
let srv = |req: ServiceRequest| { let srv = |req: ServiceRequest| {
assert_eq!("/v1/something/", req.path()); assert_eq!("/v1/something/", req.path());
ok(req.into_response(HttpResponse::Ok().finish())) ready(Ok(req.into_response(HttpResponse::Ok().finish())))
}; };
let mut normalize = NormalizePath::default() let mut normalize = NormalizePath::default()
@ -310,7 +340,7 @@ mod tests {
let srv = |req: ServiceRequest| { let srv = |req: ServiceRequest| {
assert_eq!(URI, req.path()); assert_eq!(URI, req.path());
ok(req.into_response(HttpResponse::Ok().finish())) ready(Ok(req.into_response(HttpResponse::Ok().finish())))
}; };
let mut normalize = NormalizePath::default() let mut normalize = NormalizePath::default()
@ -324,12 +354,12 @@ mod tests {
} }
#[actix_rt::test] #[actix_rt::test]
async fn should_normalize_notrail() { async fn should_normalize_no_trail() {
const URI: &str = "/v1/something"; const URI: &str = "/v1/something";
let srv = |req: ServiceRequest| { let srv = |req: ServiceRequest| {
assert_eq!(URI.to_string() + "/", req.path()); assert_eq!(URI.to_string() + "/", req.path());
ok(req.into_response(HttpResponse::Ok().finish())) ready(Ok(req.into_response(HttpResponse::Ok().finish())))
}; };
let mut normalize = NormalizePath::default() let mut normalize = NormalizePath::default()