Implement Unique::fold

rust-itertools · Apr 18, 2024 · db0c386 · db0c386
1 parent ed695af
commit db0c386
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/src/unique_impl.rs b/src/unique_impl.rs
@@ -116,6 +116,23 @@ where
         })
     }
 
+    fn fold<B, F>(mut self, init: B, mut f: F) -> B
+    where
+        Self: Sized,
+        F: FnMut(B, Self::Item) -> B,
+    {
+        let UniqueBy { iter, used, .. } = &mut self.iter;
+        iter.fold(init, |mut acc, v| {
+            if let Entry::Vacant(e) = used.entry(v) {
+                let elt = e.key().clone();
+                e.insert(());
+                acc = f(acc, elt);
+            }
+
+            acc
+        })
+    }
+
     #[inline]
     fn size_hint(&self) -> (usize, Option<usize>) {
         let (low, hi) = self.iter.iter.size_hint();