ruw 0.0.1

Read-Update-Write
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37

use core::{
    pin::Pin,
    task::{Context, Poll},
};

use futures_core::{FusedStream, Stream};

pub(crate) struct StreamIter<'a, 'cx, S> {
    stream: Pin<&'a mut S>,
    cx: &'a mut Context<'cx>,
}

impl<'a, 'cx, S> StreamIter<'a, 'cx, S> {
    pub(crate) fn new(stream: Pin<&'a mut S>, cx: &'a mut Context<'cx>) -> Self {
        Self { stream, cx }
    }

    pub(crate) fn extend_into(self, collection: &mut impl Extend<S::Item>)
    where
        S: FusedStream,
    {
        if !self.stream.is_terminated() {
            collection.extend(self)
        }
    }
}

impl<'a, 'cx, S: Stream> Iterator for StreamIter<'a, 'cx, S> {
    type Item = S::Item;

    fn next(&mut self) -> Option<Self::Item> {
        match self.stream.as_mut().poll_next(self.cx) {
            Poll::Ready(next) => next,
            Poll::Pending => None,
        }
    }
}