1
0
mirror of https://github.com/fafhrd91/actix-net synced 2024-12-03 19:42:13 +01:00
actix-net/actix-service/src/map_err.rs

243 lines
5.1 KiB
Rust
Raw Normal View History

2018-11-30 03:56:15 +01:00
use std::marker::PhantomData;
2018-08-25 18:02:14 +02:00
use futures::{Async, Future, Poll};
2018-09-11 18:30:22 +02:00
use super::{NewService, Service};
2018-08-25 18:02:14 +02:00
2018-09-12 22:34:53 +02:00
/// Service for the `map_err` combinator, changing the type of a service's
/// error.
///
/// This is created by the `ServiceExt::map_err` method.
2018-11-30 03:56:15 +01:00
pub struct MapErr<A, F, E> {
2018-09-12 22:34:53 +02:00
service: A,
2018-08-25 18:02:14 +02:00
f: F,
2018-11-30 03:56:15 +01:00
_t: PhantomData<E>,
2018-08-25 18:02:14 +02:00
}
2018-11-30 03:56:15 +01:00
impl<A, F, E> MapErr<A, F, E> {
2018-08-25 18:02:14 +02:00
/// Create new `MapErr` combinator
pub fn new(service: A, f: F) -> Self
2018-11-30 03:56:15 +01:00
where
A: Service,
2018-11-30 03:56:15 +01:00
F: Fn(A::Error) -> E,
{
Self {
service,
f,
_t: PhantomData,
}
2018-08-25 18:02:14 +02:00
}
}
impl<A, F, E> Clone for MapErr<A, F, E>
where
2018-11-30 03:56:15 +01:00
A: Clone,
F: Clone,
{
fn clone(&self) -> Self {
MapErr {
2018-09-12 22:34:53 +02:00
service: self.service.clone(),
f: self.f.clone(),
2018-11-30 03:56:15 +01:00
_t: PhantomData,
}
}
}
impl<A, F, E> Service for MapErr<A, F, E>
2018-08-25 18:02:14 +02:00
where
A: Service,
2018-11-30 03:56:15 +01:00
F: Fn(A::Error) -> E + Clone,
2018-08-25 18:02:14 +02:00
{
type Request = A::Request;
2018-08-25 18:02:14 +02:00
type Response = A::Response;
type Error = E;
type Future = MapErrFuture<A, F, E>;
2018-08-25 18:02:14 +02:00
fn poll_ready(&mut self) -> Poll<(), Self::Error> {
2018-09-12 22:34:53 +02:00
self.service.poll_ready().map_err(&self.f)
2018-08-25 18:02:14 +02:00
}
fn call(&mut self, req: A::Request) -> Self::Future {
2018-09-12 22:34:53 +02:00
MapErrFuture::new(self.service.call(req), self.f.clone())
2018-08-25 18:02:14 +02:00
}
}
pub struct MapErrFuture<A, F, E>
2018-08-25 18:02:14 +02:00
where
A: Service,
2018-08-25 18:02:14 +02:00
F: Fn(A::Error) -> E,
{
f: F,
fut: A::Future,
}
impl<A, F, E> MapErrFuture<A, F, E>
2018-08-25 18:02:14 +02:00
where
A: Service,
2018-08-25 18:02:14 +02:00
F: Fn(A::Error) -> E,
{
fn new(fut: A::Future, f: F) -> Self {
MapErrFuture { f, fut }
}
}
impl<A, F, E> Future for MapErrFuture<A, F, E>
2018-08-25 18:02:14 +02:00
where
A: Service,
2018-08-25 18:02:14 +02:00
F: Fn(A::Error) -> E,
{
type Item = A::Response;
type Error = E;
fn poll(&mut self) -> Poll<Self::Item, Self::Error> {
self.fut.poll().map_err(&self.f)
}
}
2018-09-18 03:10:23 +02:00
/// NewService for the `map_err` combinator, changing the type of a new
/// service's error.
///
/// This is created by the `NewServiceExt::map_err` method.
2018-08-25 18:02:14 +02:00
pub struct MapErrNewService<A, F, E> {
a: A,
f: F,
2018-11-30 03:56:15 +01:00
e: PhantomData<E>,
2018-08-25 18:02:14 +02:00
}
2018-11-30 03:56:15 +01:00
impl<A, F, E> MapErrNewService<A, F, E> {
2018-08-25 18:02:14 +02:00
/// Create new `MapErr` new service instance
pub fn new(a: A, f: F) -> Self
2018-11-30 03:56:15 +01:00
where
A: NewService,
2018-11-30 03:56:15 +01:00
F: Fn(A::Error) -> E,
{
2018-08-25 18:02:14 +02:00
Self {
a,
f,
2018-11-30 03:56:15 +01:00
e: PhantomData,
2018-08-25 18:02:14 +02:00
}
}
}
impl<A, F, E> Clone for MapErrNewService<A, F, E>
where
2018-11-30 03:56:15 +01:00
A: Clone,
F: Clone,
2018-08-25 18:02:14 +02:00
{
fn clone(&self) -> Self {
Self {
a: self.a.clone(),
f: self.f.clone(),
2018-11-30 03:56:15 +01:00
e: PhantomData,
2018-08-25 18:02:14 +02:00
}
}
}
impl<A, F, E> NewService for MapErrNewService<A, F, E>
2018-08-25 18:02:14 +02:00
where
A: NewService,
2018-08-25 18:02:14 +02:00
F: Fn(A::Error) -> E + Clone,
{
type Request = A::Request;
2018-08-25 18:02:14 +02:00
type Response = A::Response;
type Error = E;
type Service = MapErr<A::Service, F, E>;
type InitError = A::InitError;
type Future = MapErrNewServiceFuture<A, F, E>;
2018-08-25 18:02:14 +02:00
fn new_service(&self) -> Self::Future {
MapErrNewServiceFuture::new(self.a.new_service(), self.f.clone())
}
}
pub struct MapErrNewServiceFuture<A, F, E>
2018-08-25 18:02:14 +02:00
where
A: NewService,
2018-08-25 18:02:14 +02:00
F: Fn(A::Error) -> E,
{
fut: A::Future,
f: F,
}
impl<A, F, E> MapErrNewServiceFuture<A, F, E>
2018-08-25 18:02:14 +02:00
where
A: NewService,
2018-08-25 18:02:14 +02:00
F: Fn(A::Error) -> E,
{
fn new(fut: A::Future, f: F) -> Self {
MapErrNewServiceFuture { f, fut }
}
}
impl<A, F, E> Future for MapErrNewServiceFuture<A, F, E>
2018-08-25 18:02:14 +02:00
where
A: NewService,
2018-08-25 18:02:14 +02:00
F: Fn(A::Error) -> E + Clone,
{
type Item = MapErr<A::Service, F, E>;
type Error = A::InitError;
fn poll(&mut self) -> Poll<Self::Item, Self::Error> {
if let Async::Ready(service) = self.fut.poll()? {
Ok(Async::Ready(MapErr::new(service, self.f.clone())))
} else {
Ok(Async::NotReady)
}
}
}
2018-09-12 22:34:53 +02:00
#[cfg(test)]
mod tests {
use futures::future::{err, FutureResult};
use super::*;
2018-12-13 03:56:39 +01:00
use crate::{IntoNewService, NewService, Service, ServiceExt};
2018-09-12 22:34:53 +02:00
struct Srv;
impl Service for Srv {
type Request = ();
2018-09-12 22:34:53 +02:00
type Response = ();
type Error = ();
type Future = FutureResult<(), ()>;
fn poll_ready(&mut self) -> Poll<(), Self::Error> {
Err(())
}
fn call(&mut self, _: ()) -> Self::Future {
err(())
}
}
#[test]
fn test_poll_ready() {
let mut srv = Srv.map_err(|_| "error");
let res = srv.poll_ready();
assert!(res.is_err());
assert_eq!(res.err().unwrap(), "error");
}
#[test]
fn test_call() {
let mut srv = Srv.map_err(|_| "error");
let res = srv.call(()).poll();
assert!(res.is_err());
assert_eq!(res.err().unwrap(), "error");
}
2018-09-18 04:21:24 +02:00
#[test]
fn test_new_service() {
let blank = || Ok::<_, ()>(Srv);
let new_srv = blank.into_new_service().map_err(|_| "error");
if let Async::Ready(mut srv) = new_srv.new_service().poll().unwrap() {
let res = srv.call(()).poll();
assert!(res.is_err());
assert_eq!(res.err().unwrap(), "error");
} else {
panic!()
}
}
2018-09-12 22:34:53 +02:00
}