mirror/async-std
2
0
Fork 1
mirror of https://github.com/async-rs/async-std.git synced 2025-05-15 19:41:31 +00:00
Code Issues Releases Wiki Activity

Apply suggestions from code review

Browse source 
Co-authored-by: nasa <htilcs1115@gmail.com>
This commit is contained in:
Oleg Nosov 2020-06-12 18:03:07 +03:00 committed by GitHub
parent 68063adddf
commit 2323ac9a8e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 13 additions and 25 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
src/stream/stream/flat_map.rs
View file

@ -51,22 +51,16 @@ where
let mut this = self.project();
loop {
if let Some(inner) = this.inner_stream.as_mut().as_pin_mut() {
let next_item = futures_core::ready!(inner.poll_next(cx));
if next_item.is_some() {
return Poll::Ready(next_item);
} else {
this.inner_stream.set(None);
match futures_core::ready!(inner.poll_next(cx)) {
item @ Some(_) => return Poll::Ready(item),
None => this.inner_stream.set(None),
}
}
let inner = futures_core::ready!(this.stream.as_mut().poll_next(cx));
if inner.is_some() {
this.inner_stream.set(inner.map(IntoStream::into_stream));
} else {
return Poll::Ready(None);
match futures_core::ready!(this.stream.as_mut().poll_next(cx)) {
inner @ Some(_) => this.inner_stream.set(inner.map(IntoStream::into_stream)),
None => return Poll::Ready(None),
}
}
}
}
}

18
src/stream/stream/flatten.rs
View file

@ -52,21 +52,15 @@ where
let mut this = self.project();
loop {
if let Some(inner) = this.inner_stream.as_mut().as_pin_mut() {
let next_item = futures_core::ready!(inner.poll_next(cx));
if next_item.is_some() {
return Poll::Ready(next_item);
} else {
this.inner_stream.set(None);
match futures_core::ready!(inner.poll_next(cx)) {
item @ Some(_) => return Poll::Ready(next_item),
None => this.inner_stream.set(None),
}
}
let inner = futures_core::ready!(this.stream.as_mut().poll_next(cx));
if inner.is_some() {
this.inner_stream.set(inner.map(IntoStream::into_stream));
} else {
return Poll::Ready(None);
match futures_core::ready!(this.stream.as_mut().poll_next(cx)) {
inner @ Some(_) => this.inner_stream.set(inner.map(IntoStream::into_stream)),
None => Poll::Ready(None),
}
}
}