mirror of
https://github.com/tokio-rs/tokio.git
synced 2026-08-27 00:00:12 +02:00
Ivan Petkov
338b37884a
This avoids having consumers import libc for common signals, and it improves discoverability since users need not be aware that libc contains all supported constants.
32 lines
806 B
Rust
32 lines
806 B
Rust
#![cfg(unix)]
|
|
#![warn(rust_2018_idioms)]
|
|
|
|
use futures_util::future::FutureExt;
|
|
use libc::{c_int, getpid, kill};
|
|
use std::future::Future;
|
|
use std::time::Duration;
|
|
use tokio_timer::Timeout;
|
|
|
|
pub use futures_util::future;
|
|
pub use futures_util::stream::StreamExt;
|
|
pub use tokio::runtime::current_thread::{self, Runtime as CurrentThreadRuntime};
|
|
pub use tokio_signal::unix::{Signal, SignalKind};
|
|
|
|
pub fn with_timeout<F: Future>(future: F) -> impl Future<Output = F::Output> {
|
|
Timeout::new(future, Duration::from_secs(1)).map(Result::unwrap)
|
|
}
|
|
|
|
pub fn run_with_timeout<F>(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);
|
|
}
|
|
}
|