use futures_util::{stream, Stream};
use crate::Intervalable;
pub fn intervalable_iter_stream<I, INTVL>(iter: I, interval: INTVL) -> impl Stream<Item = I::Item>
where
I: IntoIterator,
INTVL: Intervalable,
{
stream::unfold(
(iter.into_iter(), interval),
|(mut iter, mut interval)| async move {
if let Some(item) = iter.next() {
interval.wait().await;
Some((item, (iter, interval)))
} else {
None
}
},
)
}
pub fn intervalable_repeat_stream<T, INTVL>(item: T, interval: INTVL) -> impl Stream<Item = T>
where
T: Clone,
INTVL: Intervalable,
{
stream::unfold((item, interval), |(item, mut interval)| async move {
interval.wait().await;
Some((item.clone(), (item, interval)))
})
}
pub fn intervalable_repeat_with_stream<A, F, INTVL>(
repeater: F,
interval: INTVL,
) -> impl Stream<Item = A>
where
F: FnMut() -> A,
INTVL: Intervalable,
{
stream::unfold(
(repeater, interval),
|(mut repeater, mut interval)| async move {
interval.wait().await;
Some((repeater(), (repeater, interval)))
},
)
}