From 248c8b96a0a96190df2cefc784d0586a11b4b5a4 Mon Sep 17 00:00:00 2001
From: GrayJack <gr41.j4ck@gmail.com>
Date: Mon, 25 Nov 2024 21:06:01 -0300
Subject: [PATCH] feat: Specialize `Iterator::count` when for `Drain`,
 `IntoIter` and `Iter`

---
 src/lib.rs | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/src/lib.rs b/src/lib.rs
index 511f272..06cd1f6 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -1073,6 +1073,11 @@ impl<T: PrimInt, const N: usize> Iterator for Drain<'_, T, N> {
         let len = self.inner.len();
         (len, Some(len))
     }
+
+    fn count(self) -> usize
+    where Self: Sized {
+        self.len()
+    }
 }
 
 impl<T: PrimInt, const N: usize> ExactSizeIterator for Drain<'_, T, N> {
@@ -1136,6 +1141,11 @@ impl<T: PrimInt, const N: usize> Iterator for IntoIter<T, N> {
         let len = self.0.len();
         (len, Some(len))
     }
+
+    fn count(self) -> usize
+    where Self: Sized {
+        self.len()
+    }
 }
 
 impl<T: PrimInt, const N: usize> DoubleEndedIterator for IntoIter<T, N> {
@@ -1158,6 +1168,7 @@ impl<T: PrimInt, const N: usize> DoubleEndedIterator for IntoIter<T, N> {
 
 impl<T: PrimInt, const N: usize> FusedIterator for IntoIter<T, N> {}
 impl<T: PrimInt, const N: usize> ExactSizeIterator for IntoIter<T, N> {
+    #[inline]
     fn len(&self) -> usize {
         self.0.len()
     }
@@ -1224,6 +1235,11 @@ impl<T: PrimInt, const N: usize> Iterator for Iter<'_, T, N> {
         let len = self.borrow.len() - self.passed_count;
         (len, Some(len))
     }
+
+    fn count(self) -> usize
+    where Self: Sized {
+        self.len()
+    }
 }
 
 impl<T: PrimInt, const N: usize> DoubleEndedIterator for Iter<'_, T, N> {
@@ -1244,6 +1260,7 @@ impl<T: PrimInt, const N: usize> DoubleEndedIterator for Iter<'_, T, N> {
 
 impl<T: PrimInt, const N: usize> FusedIterator for Iter<'_, T, N> {}
 impl<T: PrimInt, const N: usize> ExactSizeIterator for Iter<'_, T, N> {
+    #[inline]
     fn len(&self) -> usize {
         self.borrow.len() - self.passed_count
     }