From f555d1618917c59ef3e8f3a217fdea8015f08554 Mon Sep 17 00:00:00 2001 From: Xenira <1288524+Xenira@users.noreply.github.com> Date: Mon, 27 May 2024 20:00:38 +0200 Subject: [PATCH] feat(`FilterOk`): implement `DoubleEndedIterator` Refs: #947 --- benches/specializations.rs | 1 + src/adaptors/mod.rs | 14 ++++++++++++++ tests/specializations.rs | 4 +++- 3 files changed, 18 insertions(+), 1 deletion(-) diff --git a/benches/specializations.rs b/benches/specializations.rs index 18039fc4e..958bcc124 100644 --- a/benches/specializations.rs +++ b/benches/specializations.rs @@ -639,6 +639,7 @@ bench_specializations! { v.iter().copied().map_ok(|x| x + 1) } filter_ok { + DoubleEndedIterator { let v = black_box((0_u32..1024) .map(|x| if x % 2 == 1 { Err(x) } else { Ok(x) }) diff --git a/src/adaptors/mod.rs b/src/adaptors/mod.rs index 6f5e0260e..cbef7ef3d 100644 --- a/src/adaptors/mod.rs +++ b/src/adaptors/mod.rs @@ -936,6 +936,20 @@ where } } +impl DoubleEndedIterator for FilterOk +where + I: DoubleEndedIterator>, + F: FnMut(&T) -> bool, +{ + fn next_back(&mut self) -> Option { + let f = &mut self.f; + self.iter.rfind(|res| match res { + Ok(t) => f(t), + _ => true, + }) + } +} + impl FusedIterator for FilterOk where I: FusedIterator>, diff --git a/tests/specializations.rs b/tests/specializations.rs index 712311472..cdc51fb23 100644 --- a/tests/specializations.rs +++ b/tests/specializations.rs @@ -447,7 +447,9 @@ quickcheck! { } fn filter_ok(v: Vec>) -> () { - test_specializations(&v.into_iter().filter_ok(|&i| i < 20)); + let it = v.into_iter().filter_ok(|&i| i < 20); + test_specializations(&it); + test_double_ended_specializations(&it); } fn filter_map_ok(v: Vec>) -> () {