ruchei 0.1.2

Utilities for working with many streams
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50

use std::convert::Infallible;

use futures_util::{
    Stream, StreamExt, TryStream,
    future::{Ready, ready},
    stream::FilterMap,
};

#[derive(Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord, Hash)]
#[must_use]
pub enum ConnectionItem<S, T = <S as TryStream>::Ok, E = <S as TryStream>::Error> {
    Item(T),
    Closed(S, Option<E>),
}

pub type MultiRouteItem<K, S, T = <S as TryStream>::Ok, E = <S as TryStream>::Error> =
    ConnectionItem<(K, S), (K, T), E>;

type Fut<R> = Ready<Option<Result<<R as ConnectionItemExt>::T, Infallible>>>;

type Mi<R> = ConnectionItem<
    <R as ConnectionItemExt>::S,
    <R as ConnectionItemExt>::T,
    <R as ConnectionItemExt>::E,
>;

pub trait ConnectionItemExt:
    Sized + Stream<Item = ConnectionItem<Self::S, Self::T, Self::E>>
{
    type T;
    type S;
    type E;

    fn connection_item_ignore(
        self,
    ) -> FilterMap<Self, Fut<Self>, impl FnMut(Mi<Self>) -> Fut<Self>> {
        self.filter_map(|item| {
            ready(match item {
                ConnectionItem::Item(item) => Some(Ok(item)),
                ConnectionItem::Closed(_, _) => None,
            })
        })
    }
}

impl<T, S, E, R: Stream<Item = ConnectionItem<S, T, E>>> ConnectionItemExt for R {
    type T = T;
    type S = S;
    type E = E;
}