|
|
|
@ -40,24 +40,35 @@ where
|
|
|
|
|
S: Stream<Item = Result<U, E>> + 'a,
|
|
|
|
|
{
|
|
|
|
|
Box::pin(async move {
|
|
|
|
|
// Using `scan` here because it is able to stop the stream early
|
|
|
|
|
// Using `take_while` here because it is able to stop the stream early
|
|
|
|
|
// if a failure occurs
|
|
|
|
|
let mut is_error = false;
|
|
|
|
|
let mut found_error = None;
|
|
|
|
|
let out = <T as Product<U>>::product(stream.scan(&mut found_error, |error, elem| {
|
|
|
|
|
match elem {
|
|
|
|
|
Ok(elem) => Some(elem),
|
|
|
|
|
Err(err) => {
|
|
|
|
|
**error = Some(err);
|
|
|
|
|
// Stop processing the stream on error
|
|
|
|
|
None
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}))
|
|
|
|
|
let out = <T as Product<U>>::product(
|
|
|
|
|
stream
|
|
|
|
|
.take_while(|elem| {
|
|
|
|
|
// Stop processing the stream on `Err`
|
|
|
|
|
!is_error
|
|
|
|
|
&& (elem.is_ok() || {
|
|
|
|
|
is_error = true;
|
|
|
|
|
// Capture first `Err`
|
|
|
|
|
true
|
|
|
|
|
})
|
|
|
|
|
})
|
|
|
|
|
.filter_map(|elem| match elem {
|
|
|
|
|
Ok(value) => Some(value),
|
|
|
|
|
Err(err) => {
|
|
|
|
|
found_error = Some(err);
|
|
|
|
|
None
|
|
|
|
|
}
|
|
|
|
|
}),
|
|
|
|
|
)
|
|
|
|
|
.await;
|
|
|
|
|
|
|
|
|
|
match found_error {
|
|
|
|
|
Some(err) => Err(err),
|
|
|
|
|
None => Ok(out),
|
|
|
|
|
if is_error {
|
|
|
|
|
Err(found_error.unwrap())
|
|
|
|
|
} else {
|
|
|
|
|
Ok(out)
|
|
|
|
|
}
|
|
|
|
|
})
|
|
|
|
|
}
|
|
|
|
|