set_from: use try_fold

rust-itertools · Feb 28, 2024 · 37d47f6 · 37d47f6
1 parent 3918da2
commit 37d47f6
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/src/lib.rs b/src/lib.rs
@@ -2218,15 +2218,15 @@ pub trait Itertools: Iterator {
         Self: Iterator<Item = &'a mut A>,
         J: IntoIterator<Item = A>,
     {
-        let mut count = 0;
-        for elt in from {
-            match self.next() {
-                None => break,
-                Some(ptr) => *ptr = elt,
-            }
-            count += 1;
-        }
-        count
+        from.into_iter()
+            .try_fold(0, |count, elt| match self.next() {
+                Some(ptr) => {
+                    *ptr = elt;
+                    Ok(count + 1)
+                }
+                None => Err(count),
+            })
+            .unwrap_or_else(|count| count)
     }
 
     /// Combine all iterator elements into one String, separated by `sep`.