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

Fix review nits

Browse source
This commit is contained in:
Fedor Sakharov 2019-10-06 08:32:44 +03:00
parent 2384df11ed
commit 49d123c7f9
No known key found for this signature in database
GPG key ID: 93D436E666BF0FEE
1 changed files with 8 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
src/stream/repeat_with.rs
View file

@ -57,7 +57,11 @@ pub struct RepeatWith<F, Fut, A> {
/// assert_eq!(s.next().await, None); /// assert_eq!(s.next().await, None);
/// # }) } /// # }) }
/// ``` /// ```
pub fn repeat_with<F, Fut, A>(repeater: F) -> RepeatWith<F, Fut, A> { pub fn repeat_with<F, Fut, A>(repeater: F) -> RepeatWith<F, Fut, A>
where
F: FnMut() -> Fut,
Fut: Future<Output = A>,
{
RepeatWith { RepeatWith {
f: repeater, f: repeater,
future: None, future: None,
@ -79,8 +83,8 @@ where
fn poll_next(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Option<Self::Item>> { fn poll_next(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Option<Self::Item>> {
loop { loop {
match self.future.is_some() { match &self.future {
true => { Some(_) => {
let res = let res =
futures_core::ready!(self.as_mut().future().as_pin_mut().unwrap().poll(cx)); futures_core::ready!(self.as_mut().future().as_pin_mut().unwrap().poll(cx));
@ -88,7 +92,7 @@ where
return Poll::Ready(Some(res)); return Poll::Ready(Some(res));
} }
false => { None => {
let fut = (self.as_mut().f())(); let fut = (self.as_mut().f())();
self.as_mut().future().set(Some(fut)); self.as_mut().future().set(Some(fut));