tangent128/webmetro
1
0
Fork 0
Code Issues 1 Pull requests Releases Packages Wiki Activity

Run filter under Tokio runtime so that Delay works

Browse source
This commit is contained in:
Tangent 128 2018-04-14 18:44:53 -04:00
parent ab55d951a0
commit 0075e52f7b
1 changed files with 8 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
src/commands/filter.rs
View file

@ -6,6 +6,7 @@ use std::{
use clap::{App, Arg, ArgMatches, SubCommand}; use clap::{App, Arg, ArgMatches, SubCommand};
use futures::prelude::*; use futures::prelude::*;
use tokio;
use super::stdin_stream; use super::stdin_stream;
use webmetro::{ use webmetro::{
@ -28,7 +29,7 @@ pub fn options() -> App<'static, 'static> {
pub fn run(args: &ArgMatches) -> Result<(), Box<Error>> { pub fn run(args: &ArgMatches) -> Result<(), Box<Error>> {
let mut chunk_stream: Box<Stream<Item = Chunk, Error = WebmetroError>> = Box::new( let mut chunk_stream: Box<Stream<Item = Chunk, Error = WebmetroError> + Send> = Box::new(
stdin_stream() stdin_stream()
.parse_ebml() .parse_ebml()
.chunk_webm() .chunk_webm()
@ -39,9 +40,12 @@ pub fn run(args: &ArgMatches) -> Result<(), Box<Error>> {
chunk_stream = Box::new(chunk_stream.throttle()); chunk_stream = Box::new(chunk_stream.throttle());
} }
chunk_stream.fold((), |_, chunk| { tokio::run(chunk_stream.for_each(|chunk| {
io::stdout().write_all(chunk.as_ref()) io::stdout().write_all(chunk.as_ref()).map_err(WebmetroError::IoError)
}).wait()?; }).map_err(|err| {
println!("Error: {}", err);
::std::process::exit(1);
}));
Ok(()) Ok(())
} }