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

//! Read lines from STDIN and signal when the STDIN has been consumed

use crate::common::Result;
use crate::events::Event;
use async_std::channel::Sender;
use async_std::io;
use async_std::prelude::*;
use async_std::stream;

/// Run the data input task
pub async fn task<R>(stdin: R, sender: Sender<Event>) -> Result<()>
where
    R: io::Read + Unpin + Send + 'static,
{
    log::trace!("starting to read input data");

    let reader = io::BufReader::new(stdin);
    let mut stream = reader
        .lines()
        .map(|res| res.expect("Error reading from STDIN"))
        .filter(|line| !line.is_empty())
        .map(Event::NewLine)
        .chain(stream::once(Event::EOF));

    while let Some(event) = stream.next().await {
        sender.send(event).await?;
    }

    log::trace!("input data done");

    Ok(())
}