dfir_pipes/push/
inspect.rs

1//! [`Inspect`] push combinator.
2use core::pin::Pin;
3
4use pin_project_lite::pin_project;
5
6use crate::push::{Push, PushStep};
7
8pin_project! {
9    /// Push combinator that calls a closure on each item for side effects before pushing downstream.
10    #[must_use = "`Push`es do nothing unless items are pushed into them"]
11    #[derive(Clone, Debug)]
12    pub struct Inspect<Next, Func> {
13        #[pin]
14        next: Next,
15        func: Func,
16    }
17}
18
19impl<Next, Func> Inspect<Next, Func> {
20    /// Creates with inspecting `func` and next `push`.
21    pub(crate) const fn new<Item>(func: Func, next: Next) -> Self
22    where
23        Func: FnMut(&Item),
24    {
25        Self { next, func }
26    }
27}
28
29impl<Next, Func, Item, Meta> Push<Item, Meta> for Inspect<Next, Func>
30where
31    Next: Push<Item, Meta>,
32    Func: FnMut(&Item),
33    Meta: Copy,
34{
35    type Ctx<'ctx> = Next::Ctx<'ctx>;
36
37    type CanPend = Next::CanPend;
38
39    fn poll_ready(self: Pin<&mut Self>, ctx: &mut Self::Ctx<'_>) -> PushStep<Self::CanPend> {
40        self.project().next.poll_ready(ctx)
41    }
42
43    fn start_send(self: Pin<&mut Self>, item: Item, meta: Meta) {
44        let this = self.project();
45        (this.func)(&item);
46        this.next.start_send(item, meta)
47    }
48
49    fn poll_flush(self: Pin<&mut Self>, ctx: &mut Self::Ctx<'_>) -> PushStep<Self::CanPend> {
50        self.project().next.poll_flush(ctx)
51    }
52
53    fn size_hint(self: Pin<&mut Self>, hint: (usize, Option<usize>)) {
54        self.project().next.size_hint(hint);
55    }
56}
dfir_pipes/push/inspect.rs

dfir_pipes/push/
inspect.rs
