mirror of
https://github.com/fafhrd91/actix-net
synced 2024-11-24 05:52:59 +01:00
Simplify transform trait, remove map_init_err
This commit is contained in:
parent
8bf8ad86d6
commit
5b74c79cf9
@ -298,5 +298,5 @@ pub mod dev {
|
|||||||
pub use crate::map_err::{MapErr, MapErrServiceFactory};
|
pub use crate::map_err::{MapErr, MapErrServiceFactory};
|
||||||
pub use crate::map_init_err::MapInitErr;
|
pub use crate::map_init_err::MapInitErr;
|
||||||
pub use crate::then::{ThenService, ThenServiceFactory};
|
pub use crate::then::{ThenService, ThenServiceFactory};
|
||||||
pub use crate::transform::{ApplyTransform, TransformMapInitErr};
|
pub use crate::transform::ApplyTransform;
|
||||||
}
|
}
|
||||||
|
@ -1,5 +1,4 @@
|
|||||||
use std::future::Future;
|
use std::future::Future;
|
||||||
use std::marker::PhantomData;
|
|
||||||
use std::pin::Pin;
|
use std::pin::Pin;
|
||||||
use std::rc::Rc;
|
use std::rc::Rc;
|
||||||
use std::sync::Arc;
|
use std::sync::Arc;
|
||||||
@ -37,17 +36,6 @@ pub trait Transform<S> {
|
|||||||
|
|
||||||
/// Creates and returns a new Service component, asynchronously
|
/// Creates and returns a new Service component, asynchronously
|
||||||
fn new_transform(&self, service: S) -> Self::Future;
|
fn new_transform(&self, service: S) -> Self::Future;
|
||||||
|
|
||||||
/// Map this service's factory error to a different error,
|
|
||||||
/// returning a new transform service factory.
|
|
||||||
fn map_init_err<F, E>(self, f: F) -> TransformMapInitErr<Self, S, F, E>
|
|
||||||
where
|
|
||||||
Self: Sized,
|
|
||||||
Self::Future: Unpin,
|
|
||||||
F: Fn(Self::InitError) -> E + Unpin + Clone,
|
|
||||||
{
|
|
||||||
TransformMapInitErr::new(self, f)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<T, S> Transform<S> for Rc<T>
|
impl<T, S> Transform<S> for Rc<T>
|
||||||
@ -187,90 +175,3 @@ where
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Transform for the `map_err` combinator, changing the type of a new
|
|
||||||
/// transform's init error.
|
|
||||||
///
|
|
||||||
/// This is created by the `Transform::map_err` method.
|
|
||||||
pub struct TransformMapInitErr<T, S, F, E> {
|
|
||||||
t: T,
|
|
||||||
f: F,
|
|
||||||
e: PhantomData<(S, E)>,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl<T, S, F, E> TransformMapInitErr<T, S, F, E> {
|
|
||||||
/// Create new `TransformMapErr` new transform instance
|
|
||||||
pub(crate) fn new(t: T, f: F) -> Self
|
|
||||||
where
|
|
||||||
T: Transform<S>,
|
|
||||||
T::Future: Unpin,
|
|
||||||
F: Fn(T::InitError) -> E + Unpin + Clone,
|
|
||||||
{
|
|
||||||
Self {
|
|
||||||
t,
|
|
||||||
f,
|
|
||||||
e: PhantomData,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl<T, S, F, E> Clone for TransformMapInitErr<T, S, F, E>
|
|
||||||
where
|
|
||||||
T: Clone,
|
|
||||||
F: Clone,
|
|
||||||
{
|
|
||||||
fn clone(&self) -> Self {
|
|
||||||
Self {
|
|
||||||
t: self.t.clone(),
|
|
||||||
f: self.f.clone(),
|
|
||||||
e: PhantomData,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl<T, S, F, E> Transform<S> for TransformMapInitErr<T, S, F, E>
|
|
||||||
where
|
|
||||||
T: Transform<S>,
|
|
||||||
T::Future: Unpin,
|
|
||||||
F: Fn(T::InitError) -> E + Unpin + Clone,
|
|
||||||
{
|
|
||||||
type Request = T::Request;
|
|
||||||
type Response = T::Response;
|
|
||||||
type Error = T::Error;
|
|
||||||
type Transform = T::Transform;
|
|
||||||
|
|
||||||
type InitError = E;
|
|
||||||
type Future = TransformMapInitErrFuture<T, S, F, E>;
|
|
||||||
|
|
||||||
fn new_transform(&self, service: S) -> Self::Future {
|
|
||||||
TransformMapInitErrFuture {
|
|
||||||
fut: self.t.new_transform(service),
|
|
||||||
f: self.f.clone(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub struct TransformMapInitErrFuture<T, S, F, E>
|
|
||||||
where
|
|
||||||
T: Transform<S>,
|
|
||||||
T::Future: Unpin,
|
|
||||||
F: Fn(T::InitError) -> E + Unpin,
|
|
||||||
{
|
|
||||||
fut: T::Future,
|
|
||||||
f: F,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl<T, S, F, E> Future for TransformMapInitErrFuture<T, S, F, E>
|
|
||||||
where
|
|
||||||
T: Transform<S>,
|
|
||||||
T::Future: Unpin,
|
|
||||||
F: Fn(T::InitError) -> E + Unpin + Clone,
|
|
||||||
{
|
|
||||||
type Output = Result<T::Transform, E>;
|
|
||||||
|
|
||||||
fn poll(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
|
|
||||||
let this = self.get_mut();
|
|
||||||
|
|
||||||
Pin::new(&mut this.fut).poll(cx).map_err(&this.f)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
Loading…
Reference in New Issue
Block a user