add Iterator::contains

Author: Qelxiros

library/core/src/iter/traits/iterator.rs

@@ -2826,6 +2826,43 @@ pub trait Iterator {
         self.try_fold((), check(f)) == ControlFlow::Break(())
     }
 
+    /// Tests whether a value is contained in the iterator.
+    ///
+    /// `contains()` is short-circuiting; in other words, it will stop processing
+    /// as soon as the function finds the item in the `Iterator`.
+    ///
+    /// This method checks the whole iterator, which is O(n). If the iterator is a sorted
+    /// slice, [`binary_search`](slice::binary_search) may be faster. If this is an iterator
+    /// on collections that have a `.contains()` or `.contains_key()` method (such as
+    /// `HashMap` or `BtreeSet`), using those methods directly will be faster.
+    ///
+    /// # Examples
+    ///
+    /// Basic usage:
+    ///
+    /// ```
+    /// #![feature(iter_contains)]
+    /// assert!([1, 2, 3].iter().contains(2));
+    /// assert!(![1, 2, 3].iter().contains(5));
+    /// ```
+    ///
+    /// [`Iterator::contains`] can be used where [`slice::contains`] cannot be used:
+    ///
+    /// ```
+    /// #![feature(iter_contains)]
+    /// let s = [String::from("a"), String::from("b"), String::from("c")];
+    /// assert!(s.iter().contains("b"));
+    /// ```
+    #[inline]
+    #[unstable(feature = "iter_contains", issue = "127494")]
+    fn contains<Q>(&mut self, item: Q) -> bool
+    where
+        Q: PartialEq<Self::Item> + ?Sized,
+        Self: Sized,
+    {
+        self.any(|elem| item == elem)
+    }
+
     /// Searches for an element of an iterator that satisfies a predicate.
     ///
     /// `find()` takes a closure that returns `true` or `false`. It applies

library/core/src/lib.rs

@@ -110,6 +110,7 @@
 #![feature(internal_impls_macro)]
 #![feature(ip)]
 #![feature(is_ascii_octdigit)]
+#![feature(iter_contains)]
 #![feature(lazy_get)]
 #![feature(link_cfg)]
 #![feature(offset_of_enum)]

library/core/src/str/iter.rs

@@ -331,6 +331,14 @@ impl Iterator for Bytes<'_> {
         self.0.any(f)
     }
 
+    #[inline]
+    fn contains<Q>(&mut self, item: Q) -> bool
+    where
+        Q: PartialEq<Self::Item> + ?Sized,
+    {
+        self.0.contains(item)
+    }
+
     #[inline]
     fn find<P>(&mut self, predicate: P) -> Option<Self::Item>
     where

library/coretests/tests/iter/traits/iterator.rs

@@ -272,6 +272,17 @@ fn test_any() {
     assert!(!v[..0].iter().any(|_| panic!()));
 }
 
+#[test]
+fn test_iterator_contains() {
+    let v: Box<[isize]> = Box::new([1, 2, 3, 4, 5]);
+    assert_eq!(true, v.iter().contains(&3));
+    assert_eq!(v.iter().contains(&3), v.iter().any(|&x| x == 3));
+    assert_eq!(false, v.iter().contains(&10));
+    assert_eq!(v.iter().contains(&10), v.iter().any(|&x| x == 10));
+    assert_eq!(true, Iterator::contains(&mut (1isize..=5), 3));
+    assert_eq!(false, Iterator::contains(&mut (1isize..=5), 10));
+}
+
 #[test]
 fn test_find() {
     let v: &[isize] = &[1, 3, 9, 27, 103, 14, 11];

library/coretests/tests/lib.rs

@@ -55,6 +55,7 @@
 #![feature(iter_array_chunks)]
 #![feature(iter_chain)]
 #![feature(iter_collect_into)]
+#![feature(iter_contains)]
 #![feature(iter_intersperse)]
 #![feature(iter_is_partitioned)]
 #![feature(iter_map_windows)]