set_from: use zip

Co-Authored-By: Jack Wrenn <jack@wrenn.fyi>
Philippe-Cholet · Feb 28, 2024 · 01e2062 · 01e2062
1 parent 1131899
commit 01e2062
Showing 1 changed file with 4 additions and 9 deletions.
diff --git a/src/lib.rs b/src/lib.rs
@@ -2222,15 +2222,10 @@ 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()
+            .zip(self)
+            .map(|(new, old)| *old = new)
+            .count()
     }
 
     /// Combine all iterator elements into one String, separated by `sep`.