moirai_iter/parallel/
split.rs

1use super::ParallelIterator;
2
3/// Sum type used by `ParallelIterator::partition_map`.
4#[derive(Clone, Copy, Debug, Eq, Hash, PartialEq)]
5pub enum Either<L, R> {
6    /// Route the value into the left output collection.
7    Left(L),
8    /// Route the value into the right output collection.
9    Right(R),
10}
11
12pub(in crate::parallel) fn partition_map<I, A, B, P, L, R>(iterator: I, predicate: P) -> (A, B)
13where
14    I: ParallelIterator,
15    A: Default + Extend<L> + Send,
16    B: Default + Extend<R> + Send,
17    P: Fn(I::Item) -> Either<L, R> + Send + Sync + Clone,
18    L: Send,
19    R: Send,
20{
21    let mut left = A::default();
22    let mut right = B::default();
23
24    for item in iterator.seq_items() {
25        match predicate(item) {
26            Either::Left(value) => left.extend(std::iter::once(value)),
27            Either::Right(value) => right.extend(std::iter::once(value)),
28        }
29    }
30
31    (left, right)
32}
moirai_iter/parallel/split.rs

moirai_iter/parallel/
split.rs
