Skip to content

Commit 338d167

Browse files
committed
Adds Stream::ge
1 parent 4953297 commit 338d167

File tree

2 files changed

+88
-0
lines changed

2 files changed

+88
-0
lines changed

src/stream/stream/ge.rs

+47
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,47 @@
1+
use std::cmp::Ordering;
2+
use std::pin::Pin;
3+
4+
use super::partial_cmp::PartialCmpFuture;
5+
use crate::future::Future;
6+
use crate::prelude::*;
7+
use crate::stream::Stream;
8+
use crate::task::{Context, Poll};
9+
10+
// Determines if the elements of this `Stream` are lexicographically
11+
// greater than or equal to those of another.
12+
#[doc(hidden)]
13+
#[allow(missing_debug_implementations)]
14+
pub struct GeFuture<L: Stream, R: Stream> {
15+
partial_cmp: PartialCmpFuture<L, R>,
16+
}
17+
18+
impl<L: Stream, R: Stream> GeFuture<L, R>
19+
where
20+
L::Item: PartialOrd<R::Item>,
21+
{
22+
pin_utils::unsafe_pinned!(partial_cmp: PartialCmpFuture<L, R>);
23+
24+
pub(super) fn new(l: L, r: R) -> Self {
25+
GeFuture {
26+
partial_cmp: l.partial_cmp(r),
27+
}
28+
}
29+
}
30+
31+
impl<L: Stream, R: Stream> Future for GeFuture<L, R>
32+
where
33+
L: Stream + Sized,
34+
R: Stream + Sized,
35+
L::Item: PartialOrd<R::Item>,
36+
{
37+
type Output = bool;
38+
39+
fn poll(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
40+
let result = futures_core::ready!(self.as_mut().partial_cmp().poll(cx));
41+
42+
match result {
43+
Some(Ordering::Greater) | Some(Ordering::Equal) => Poll::Ready(true),
44+
_ => Poll::Ready(false),
45+
}
46+
}
47+
}

src/stream/stream/mod.rs

+41
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@ mod find_map;
3232
mod fold;
3333
mod for_each;
3434
mod fuse;
35+
mod ge;
3536
mod inspect;
3637
mod map;
3738
mod min_by;
@@ -55,6 +56,7 @@ use find::FindFuture;
5556
use find_map::FindMapFuture;
5657
use fold::FoldFuture;
5758
use for_each::ForEachFuture;
59+
use ge::GeFuture;
5860
use min_by::MinByFuture;
5961
use next::NextFuture;
6062
use nth::NthFuture;
@@ -1257,6 +1259,45 @@ extension_trait! {
12571259
{
12581260
PartialCmpFuture::new(self, other)
12591261
}
1262+
1263+
#[doc = r#"
1264+
Determines if the elements of this `Stream` are lexicographically
1265+
greater than or equal to those of another.
1266+
1267+
# Examples
1268+
```
1269+
# fn main() { async_std::task::block_on(async {
1270+
#
1271+
use async_std::prelude::*;
1272+
use std::collections::VecDeque;
1273+
1274+
let single: VecDeque<isize> = vec![1].into_iter().collect();
1275+
let single_gt: VecDeque<isize> = vec![10].into_iter().collect();
1276+
let multi: VecDeque<isize> = vec![1,2].into_iter().collect();
1277+
let multi_gt: VecDeque<isize> = vec![1,5].into_iter().collect();
1278+
1279+
assert_eq!(single.clone().ge(single.clone()).await, true);
1280+
assert_eq!(single_gt.clone().ge(single.clone()).await, true);
1281+
1282+
assert_eq!(multi.clone().ge(single_gt.clone()).await, false);
1283+
assert_eq!(multi_gt.clone().ge(multi.clone()).await, true);
1284+
1285+
1286+
#
1287+
# }) }
1288+
```
1289+
"#]
1290+
fn ge<S>(
1291+
self,
1292+
other: S
1293+
) -> impl Future<Output = bool> + '_ [GeFuture<Self, S>]
1294+
where
1295+
Self: Sized + Stream,
1296+
S: Stream,
1297+
Self::Item: PartialOrd<S::Item>,
1298+
{
1299+
GeFuture::new(self, other)
1300+
}
12601301
}
12611302

12621303
impl<S: Stream + Unpin + ?Sized> Stream for Box<S> {

0 commit comments

Comments
 (0)