datapipe 0.1.4

Stream data from here to there
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

use crate::datapipe_types::InputReader;
/// Reader for files
use bytes::{Bytes, BytesMut};
use log::trace;
use std::io::{Error, SeekFrom};
use std::path::Path;
use tokio::io::{AsyncReadExt, AsyncSeekExt};

#[derive(Debug)]
pub struct FileReader {
    file: tokio::fs::File,
}

impl FileReader {
    pub async fn new(pathref: &Path) -> Result<Self, Error> {
        Ok(Self {
            file: tokio::fs::OpenOptions::new()
                .read(true)
                .open(pathref)
                .await?,
        })
    }
}

impl InputReader for FileReader {
    async fn read(&mut self) -> Result<Bytes, Error> {
        let position = self.file.seek(SeekFrom::Current(0)).await?;
        trace!("Reading input file at position {}", position);
        let mut bytes = BytesMut::new();
        match self.file.read_buf(&mut bytes).await {
            Ok(_bytes_read) => Ok(bytes.freeze()),
            Err(error) => Err(error),
        }
    }
}