mirror of https://github.com/fafhrd91/actix-web
add boxed middleware
This commit is contained in:
parent
d40b6748bc
commit
07dbd96fcf
|
@ -0,0 +1,91 @@
|
|||
//! For middleware documentation, see [`Boxed`].
|
||||
|
||||
use actix_service::{
|
||||
boxed::{self, BoxService},
|
||||
Service, Transform,
|
||||
};
|
||||
use futures_core::future::LocalBoxFuture;
|
||||
|
||||
/// Middleware for boxing another middleware's output. It would do type earse for the final middleware service
|
||||
/// and reduce type complexity for potential faster compile time in exchange for extra overhead at runtime.
|
||||
pub struct Boxed<T> {
|
||||
transform: T,
|
||||
}
|
||||
|
||||
impl<T> Boxed<T> {
|
||||
/// Wrap a middleware to erase it's type signature and reduce type complexity.
|
||||
pub fn new(middleware: T) -> Self {
|
||||
Self {
|
||||
transform: middleware,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl<S, T, Req> Transform<S, Req> for Boxed<T>
|
||||
where
|
||||
S: Service<Req> + 'static,
|
||||
T: Transform<S, Req> + 'static,
|
||||
Req: 'static,
|
||||
{
|
||||
type Response = T::Response;
|
||||
type Error = T::Error;
|
||||
type Transform = BoxedMiddleware<Req, T::Response, T::Error>;
|
||||
type InitError = T::InitError;
|
||||
type Future = LocalBoxFuture<'static, Result<Self::Transform, Self::InitError>>;
|
||||
|
||||
fn new_transform(&self, service: S) -> Self::Future {
|
||||
let fut = self.transform.new_transform(service);
|
||||
Box::pin(async move {
|
||||
let service = fut.await?;
|
||||
Ok(BoxedMiddleware {
|
||||
service: boxed::service(service),
|
||||
})
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
pub struct BoxedMiddleware<Req, Res, Err> {
|
||||
service: BoxService<Req, Res, Err>,
|
||||
}
|
||||
|
||||
impl<Req, Res, Err> Service<Req> for BoxedMiddleware<Req, Res, Err> {
|
||||
type Response = Res;
|
||||
type Error = Err;
|
||||
type Future = LocalBoxFuture<'static, Result<Self::Response, Self::Error>>;
|
||||
|
||||
actix_service::forward_ready!(service);
|
||||
|
||||
fn call(&self, req: Req) -> Self::Future {
|
||||
self.service.call(req)
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
use actix_service::IntoService;
|
||||
|
||||
use crate::dev::ServiceRequest;
|
||||
use crate::http::StatusCode;
|
||||
use crate::middleware::Logger;
|
||||
use crate::test::{call_service, TestRequest};
|
||||
use crate::HttpResponse;
|
||||
|
||||
#[actix_rt::test]
|
||||
async fn test_boxed_logger_middleware() {
|
||||
let srv = |req: ServiceRequest| {
|
||||
Box::pin(async move {
|
||||
Ok(req.into_response(HttpResponse::InternalServerError().finish()))
|
||||
})
|
||||
};
|
||||
|
||||
let mw = Boxed::new(Logger::default())
|
||||
.new_transform(srv.into_service())
|
||||
.await
|
||||
.unwrap();
|
||||
|
||||
let resp = call_service(&mw, TestRequest::default().to_srv_request()).await;
|
||||
assert_eq!(resp.status(), StatusCode::INTERNAL_SERVER_ERROR);
|
||||
}
|
||||
}
|
|
@ -1,5 +1,6 @@
|
|||
//! Commonly used middleware.
|
||||
|
||||
mod boxed;
|
||||
mod compat;
|
||||
mod condition;
|
||||
mod default_headers;
|
||||
|
@ -7,6 +8,7 @@ mod err_handlers;
|
|||
mod logger;
|
||||
mod normalize;
|
||||
|
||||
pub use self::boxed::Boxed;
|
||||
pub use self::compat::Compat;
|
||||
pub use self::condition::Condition;
|
||||
pub use self::default_headers::DefaultHeaders;
|
||||
|
|
Loading…
Reference in New Issue