Use internal scan state in Results implementation

async-rs · Jan 15, 2020 · ed24801 · ed24801
1 parent 38de0bf
commit ed24801
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/src/result/from_stream.rs b/src/result/from_stream.rs
@@ -21,11 +21,11 @@ where
             // if a failure occurs
             let mut found_error = None;
             let out: V = stream
-                .scan((), |(), elem| {
+                .scan(&mut found_error, |error, elem| {
                     match elem {
                         Ok(elem) => Some(elem),
                         Err(err) => {
-                            found_error = Some(err);
+                            **error = Some(err);
                             // Stop processing the stream on error
                             None
                         }

diff --git a/src/result/product.rs b/src/result/product.rs
@@ -43,11 +43,11 @@ where
             // Using `scan` here because it is able to stop the stream early
             // if a failure occurs
             let mut found_error = None;
-            let out = <T as Product<U>>::product(stream.scan((), |(), elem| {
+            let out = <T as Product<U>>::product(stream.scan(&mut found_error, |error, elem| {
                 match elem {
                     Ok(elem) => Some(elem),
                     Err(err) => {
-                        found_error = Some(err);
+                        **error = Some(err);
                         // Stop processing the stream on error
                         None
                     }

diff --git a/src/result/sum.rs b/src/result/sum.rs
@@ -43,11 +43,11 @@ where
             // Using `scan` here because it is able to stop the stream early
             // if a failure occurs
             let mut found_error = None;
-            let out = <T as Sum<U>>::sum(stream.scan((), |(), elem| {
+            let out = <T as Sum<U>>::sum(stream.scan(&mut found_error, |error, elem| {
                 match elem {
                     Ok(elem) => Some(elem),
                     Err(err) => {
-                        found_error = Some(err);
+                        **error = Some(err);
                         // Stop processing the stream on error
                         None
                     }