#![warn(rust_2018_idioms)] #![allow(dead_code)] pub use tokio::runtime::current_thread::{self, Runtime as CurrentThreadRuntime}; use tokio::timer::Timeout; #[cfg(all(unix, feature = "signal"))] pub use tokio_net::signal::unix::{signal, SignalKind}; pub use futures_util::future; use futures_util::future::FutureExt; pub use futures_util::stream::StreamExt; #[cfg(unix)] use libc::{c_int, getpid, kill}; use std::future::Future; use std::time::Duration; pub fn with_timeout(future: F) -> impl Future { Timeout::new(future, Duration::from_secs(3)).map(Result::unwrap) } pub fn run_with_timeout(rt: &mut CurrentThreadRuntime, future: F) -> F::Output where F: Future, { rt.block_on(with_timeout(future)) } #[cfg(unix)] pub fn send_signal(signal: c_int) { unsafe { assert_eq!(kill(getpid(), signal), 0); } }