add boxed middleware

This commit is contained in:
fakeshadow 2021-10-23 10:28:50 +08:00
parent d40b6748bc
commit 07dbd96fcf
2 changed files with 93 additions and 0 deletions

91
src/middleware/boxed.rs Normal file
View File

@ -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);
}
}

View File

@ -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;