oss/continuwuity
oss/continuwuity
1
0
Fork 0
mirror of https://forgejo.ellis.link/continuwuation/continuwuity.git synced 2025-07-28 10:48:30 +00:00
Code Wiki Activity

additional stream tools

Browse source 
Signed-off-by: Jason Volk <jason@zemos.net>
This commit is contained in:
Jason Volk 2024-09-30 06:46:54 +00:00 • committed by strawberry
parent a8d5cf9651
commit 6b80361c31
10 changed files with 242 additions and 46 deletions
Download patch file Download diff file Expand all files Collapse all files

80
src/core/utils/stream/tools.rs Normal file
View file

@ -0,0 +1,80 @@
//! StreamTools for futures::Stream
use std::{collections::HashMap, hash::Hash};
use futures::{Future, Stream, StreamExt};
use super::ReadyExt;
use crate::expected;
/// StreamTools
///
/// This interface is not necessarily complete; feel free to add as-needed.
pub trait Tools<Item>
where
Self: Stream<Item = Item> + Send + Sized,
<Self as Stream>::Item: Send,
{
fn counts(self) -> impl Future<Output = HashMap<Item, usize>> + Send
where
<Self as Stream>::Item: Eq + Hash;
fn counts_by<K, F>(self, f: F) -> impl Future<Output = HashMap<K, usize>> + Send
where
F: Fn(Item) -> K + Send,
K: Eq + Hash + Send;
fn counts_by_with_cap<const CAP: usize, K, F>(self, f: F) -> impl Future<Output = HashMap<K, usize>> + Send
where
F: Fn(Item) -> K + Send,
K: Eq + Hash + Send;
fn counts_with_cap<const CAP: usize>(self) -> impl Future<Output = HashMap<Item, usize>> + Send
where
<Self as Stream>::Item: Eq + Hash;
}
impl<Item, S> Tools<Item> for S
where
S: Stream<Item = Item> + Send + Sized,
<Self as Stream>::Item: Send,
{
#[inline]
fn counts(self) -> impl Future<Output = HashMap<Item, usize>> + Send
where
<Self as Stream>::Item: Eq + Hash,
{
self.counts_with_cap::<0>()
}
#[inline]
fn counts_by<K, F>(self, f: F) -> impl Future<Output = HashMap<K, usize>> + Send
where
F: Fn(Item) -> K + Send,
K: Eq + Hash + Send,
{
self.counts_by_with_cap::<0, K, F>(f)
}
#[inline]
fn counts_by_with_cap<const CAP: usize, K, F>(self, f: F) -> impl Future<Output = HashMap<K, usize>> + Send
where
F: Fn(Item) -> K + Send,
K: Eq + Hash + Send,
{
self.map(f).counts_with_cap::<CAP>()
}
#[inline]
fn counts_with_cap<const CAP: usize>(self) -> impl Future<Output = HashMap<Item, usize>> + Send
where
<Self as Stream>::Item: Eq + Hash,
{
self.ready_fold(HashMap::with_capacity(CAP), |mut counts, item| {
let entry = counts.entry(item).or_default();
let value = *entry;
*entry = expected!(value + 1);
counts
})
}
}