stream_lib 0.5.2

Tool to download differnt types of 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

use bytes::Bytes;
use futures_core::stream::Stream;
use tokio::sync::mpsc::{unbounded_channel, UnboundedReceiver, UnboundedSender};

/// This struct implments a stream that is used to
/// received data from chunked and hls streams.
#[derive(Debug)]
pub struct DownloadStream {
    rx: UnboundedReceiver<Event>,
}

impl DownloadStream {
    pub(crate) fn new() -> (Self, UnboundedSender<Event>) {
        let (tx, rx) = unbounded_channel();
        (DownloadStream { rx }, tx)
    }
}

impl Stream for DownloadStream {
    type Item = Event;

    fn poll_next(
        mut self: std::pin::Pin<&mut Self>,
        cx: &mut std::task::Context<'_>,
    ) -> std::task::Poll<Option<Self::Item>> {
        self.rx.poll_recv(cx)
    }
}

#[derive(Debug)]
pub enum Event {
    /// Bytes to be written to
    Bytes {
        bytes: Bytes,
    },
    End,
    Error {
        error: crate::Error,
    },
}