flowly-service 0.4.13

Flowly is a library of modular and reusable components for building robust pipelines processing audio, video and other.
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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71

use std::{marker::PhantomData, pin::pin};

use futures::StreamExt;

use crate::Service;

#[derive(Clone)]
pub struct SpawnEach<M, S> {
    service: S,
    buffer: usize,
    _m: PhantomData<M>,
}

impl<M, S> SpawnEach<M, S>
where
    M: Send,
    S: Service<M> + Send + Clone + 'static,
{
    pub fn new(service: S, buffer: usize) -> Self
    where
        S::Out: Send,
    {
        Self {
            service,
            buffer,
            _m: PhantomData,
        }
    }
}

impl<M, R, E, S> Service<M> for SpawnEach<M, S>
where
    M: Send + 'static,
    R: Send + 'static,
    E: Send + 'static,
    S: Clone + Service<M, Out = Result<R, E>> + Send + 'static,
{
    type Out = Result<flowly_spsc::Receiver<Result<R, E>>, E>;

    fn handle(
        &mut self,
        input: M,
        cx: &crate::Context,
    ) -> impl futures::Stream<Item = Self::Out> + Send {
        let mut service = self.service.clone();
        let (mut tx, rx) = flowly_spsc::channel(self.buffer);
        let cx = cx.clone();

        tokio::spawn(async move {
            let mut stream = pin!(service.handle(input, &cx));

            while let Some(Some(res)) = cx.fuse_abort(stream.next()).await {
                if tx.send(res).await.is_err() {
                    log::warn!("mpsc::send failed");
                    break;
                }
            }
        });

        futures::stream::iter([Ok(rx)])
    }
}

pub fn spawn_each<M, S>(service: S) -> SpawnEach<M, S>
where
    M: Send,
    S: Send + Service<M> + Clone + 'static,
    S::Out: Send,
{
    SpawnEach::new(service, 2)
}