magmaus3/continuwuity
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

add ready_try_fold to utils

Browse source 
Signed-off-by: Jason Volk <jason@zemos.net>
This commit is contained in:
Jason Volk 2024-11-05 04:37:08 +00:00
parent 79c6b51860
commit 27966221f1
3 changed files with 52 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

36
src/core/utils/stream/try_ready.rs
View file

@ -3,7 +3,7 @@
use futures::{
future::{ready, Ready},
stream::{AndThen, TryForEach, TryStream, TryStreamExt},
stream::{AndThen, TryFold, TryForEach, TryStream, TryStreamExt},
};
use crate::Result;
@ -25,6 +25,19 @@ where
) -> TryForEach<Self, Ready<Result<(), E>>, impl FnMut(S::Ok) -> Ready<Result<(), E>>>
where
F: FnMut(S::Ok) -> Result<(), E>;
fn ready_try_fold<U, F>(
self, init: U, f: F,
) -> TryFold<Self, Ready<Result<U, E>>, U, impl FnMut(U, S::Ok) -> Ready<Result<U, E>>>
where
F: Fn(U, S::Ok) -> Result<U, E>;
fn ready_try_fold_default<U, F>(
self, f: F,
) -> TryFold<Self, Ready<Result<U, E>>, U, impl FnMut(U, S::Ok) -> Ready<Result<U, E>>>
where
F: Fn(U, S::Ok) -> Result<U, E>,
U: Default;
}
impl<T, E, S> TryReadyExt<T, E, S> for S
@ -49,4 +62,25 @@ where
{
self.try_for_each(move |t| ready(f(t)))
}
#[inline]
fn ready_try_fold<U, F>(
self, init: U, f: F,
) -> TryFold<Self, Ready<Result<U, E>>, U, impl FnMut(U, S::Ok) -> Ready<Result<U, E>>>
where
F: Fn(U, S::Ok) -> Result<U, E>,
{
self.try_fold(init, move |a, t| ready(f(a, t)))
}
#[inline]
fn ready_try_fold_default<U, F>(
self, f: F,
) -> TryFold<Self, Ready<Result<U, E>>, U, impl FnMut(U, S::Ok) -> Ready<Result<U, E>>>
where
F: Fn(U, S::Ok) -> Result<U, E>,
U: Default,
{
self.ready_try_fold(U::default(), f)
}
}