mirror/async-std
2
0
Fork 1
mirror of https://github.com/async-rs/async-std.git synced 2025-04-25 01:36:50 +00:00
Code Issues Releases Wiki Activity

Fixes review issues

Browse source
This commit is contained in:
Fedor Sakharov 2019-09-17 12:25:02 +03:00
parent 0080a0da8c
commit efe351659f
No known key found for this signature in database
GPG key ID: 93D436E666BF0FEE
1 changed files with 10 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
src/stream/stream/fold.rs
View file

@ -31,23 +31,23 @@ impl<S, F, T, B> FoldFuture<S, F, T, B> {
impl<S, F, B> Future for FoldFuture<S, F, S::Item, B> impl<S, F, B> Future for FoldFuture<S, F, S::Item, B>
where where
S: Stream + Unpin + Sized, S: Stream + Sized,
F: FnMut(B, S::Item) -> B, F: FnMut(B, S::Item) -> B,
{ {
type Output = B; type Output = B;
fn poll(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> { fn poll(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
loop {
let next = futures_core::ready!(self.as_mut().stream().poll_next(cx)); let next = futures_core::ready!(self.as_mut().stream().poll_next(cx));
match next { match next {
Some(v) => { Some(v) => {
cx.waker().wake_by_ref();
let old = self.as_mut().acc().take().unwrap(); let old = self.as_mut().acc().take().unwrap();
let new = (self.as_mut().f())(old, v); let new = (self.as_mut().f())(old, v);
*self.as_mut().acc() = Some(new); *self.as_mut().acc() = Some(new);
Poll::Pending
} }
None => Poll::Ready(self.as_mut().acc().take().unwrap()), None => return Poll::Ready(self.as_mut().acc().take().unwrap()),
}
} }
} }
} }