use timely::dataflow::Scope;
use differential_dataflow::{ExchangeData, VecCollection, Hashable};
use differential_dataflow::difference::{Semigroup, Monoid, Multiply};
use differential_dataflow::operators::arrange::Arranged;
use differential_dataflow::trace::TraceReader;
pub fn propose<G, Tr, K, F, P, V>(
prefixes: VecCollection<G, P, Tr::Diff>,
arrangement: Arranged<G, Tr>,
key_selector: F,
) -> VecCollection<G, (P, V), Tr::Diff>
where
G: Scope<Timestamp=Tr::Time>,
Tr: for<'a> TraceReader<
KeyOwn = K,
ValOwn = V,
Time: std::hash::Hash,
Diff: Monoid+Multiply<Output = Tr::Diff>+ExchangeData+Semigroup<Tr::DiffGat<'a>>,
>+Clone+'static,
K: Hashable + Default + Ord + 'static,
F: Fn(&P)->K+Clone+'static,
P: ExchangeData,
V: Clone + 'static,
{
crate::operators::lookup_map(
prefixes,
arrangement,
move |p: &P, k: &mut K | { *k = key_selector(p); },
move |prefix, diff, value, sum| ((prefix.clone(), Tr::owned_val(value)), diff.clone().multiply(sum)),
Default::default(),
Default::default(),
Default::default(),
)
}
pub fn propose_distinct<G, Tr, K, F, P, V>(
prefixes: VecCollection<G, P, Tr::Diff>,
arrangement: Arranged<G, Tr>,
key_selector: F,
) -> VecCollection<G, (P, V), Tr::Diff>
where
G: Scope<Timestamp=Tr::Time>,
Tr: for<'a> TraceReader<
KeyOwn = K,
ValOwn = V,
Time: std::hash::Hash,
Diff : Semigroup<Tr::DiffGat<'a>>+Monoid+Multiply<Output = Tr::Diff>+ExchangeData,
>+Clone+'static,
K: Hashable + Default + Ord + 'static,
F: Fn(&P)->K+Clone+'static,
P: ExchangeData,
V: Clone + 'static,
{
crate::operators::lookup_map(
prefixes,
arrangement,
move |p: &P, k: &mut K| { *k = key_selector(p); },
move |prefix, diff, value, _sum| ((prefix.clone(), Tr::owned_val(value)), diff.clone()),
Default::default(),
Default::default(),
Default::default(),
)
}