src/writer.rs


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

use snafu::ResultExt as _;

pub struct Writer<W> {
    writer: W,
    creator: crate::creator::Creator,
    to_write: std::collections::VecDeque<u8>,
}

impl<W: tokio::io::AsyncWrite> Writer<W> {
    pub fn new(writer: W) -> Self {
        Self {
            writer,
            creator: crate::creator::Creator::new(),
            to_write: std::collections::VecDeque::new(),
        }
    }

    pub fn frame(
        &mut self,
        time: std::time::Instant,
        data: &[u8],
    ) -> crate::error::Result<()> {
        let bytes = self.creator.frame(time, data)?;
        self.to_write.extend(bytes.iter());
        Ok(())
    }

    pub fn poll_write(&mut self) -> futures::Poll<(), crate::error::Error> {
        let (a, b) = self.to_write.as_slices();
        let buf = if a.is_empty() { b } else { a };
        let n = futures::try_ready!(self
            .writer
            .poll_write(buf)
            .context(crate::error::WriteFile));
        for _ in 0..n {
            self.to_write.pop_front();
        }
        Ok(futures::Async::Ready(()))
    }

    pub fn is_empty(&self) -> bool {
        self.to_write.is_empty()
    }
}