http_body_util/combinators/
map_err.rs

1use http_body::{Body, Frame, SizeHint};
2use pin_project_lite::pin_project;
3use std::{
4    any::type_name,
5    fmt,
6    pin::Pin,
7    task::{Context, Poll},
8};
9
10pin_project! {
11    /// Body returned by the [`map_err`] combinator.
12    ///
13    /// [`map_err`]: crate::BodyExt::map_err
14    #[derive(Clone, Copy)]
15    pub struct MapErr<B, F> {
16        #[pin]
17        inner: B,
18        f: F
19    }
20}
21
22impl<B, F> MapErr<B, F> {
23    #[inline]
24    pub(crate) fn new(body: B, f: F) -> Self {
25        Self { inner: body, f }
26    }
27
28    /// Get a reference to the inner body
29    pub fn get_ref(&self) -> &B {
30        &self.inner
31    }
32
33    /// Get a mutable reference to the inner body
34    pub fn get_mut(&mut self) -> &mut B {
35        &mut self.inner
36    }
37
38    /// Get a pinned mutable reference to the inner body
39    pub fn get_pin_mut(self: Pin<&mut Self>) -> Pin<&mut B> {
40        self.project().inner
41    }
42
43    /// Consume `self`, returning the inner body
44    pub fn into_inner(self) -> B {
45        self.inner
46    }
47}
48
49impl<B, F, E> Body for MapErr<B, F>
50where
51    B: Body,
52    F: FnMut(B::Error) -> E,
53{
54    type Data = B::Data;
55    type Error = E;
56
57    fn poll_frame(
58        self: Pin<&mut Self>,
59        cx: &mut Context<'_>,
60    ) -> Poll<Option<Result<Frame<Self::Data>, Self::Error>>> {
61        let this = self.project();
62        match this.inner.poll_frame(cx) {
63            Poll::Pending => Poll::Pending,
64            Poll::Ready(None) => Poll::Ready(None),
65            Poll::Ready(Some(Ok(frame))) => Poll::Ready(Some(Ok(frame))),
66            Poll::Ready(Some(Err(err))) => Poll::Ready(Some(Err((this.f)(err)))),
67        }
68    }
69
70    fn is_end_stream(&self) -> bool {
71        self.inner.is_end_stream()
72    }
73
74    fn size_hint(&self) -> SizeHint {
75        self.inner.size_hint()
76    }
77}
78
79impl<B, F> fmt::Debug for MapErr<B, F>
80where
81    B: fmt::Debug,
82{
83    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
84        f.debug_struct("MapErr")
85            .field("inner", &self.inner)
86            .field("f", &type_name::<F>())
87            .finish()
88    }
89}
http_body_util/combinators/map_err.rs

http_body_util/combinators/
map_err.rs
