process_control 5.2.0

Ergonomically run processes with limits
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
36
37
38
39
40
41
42
43
44
45
46
47

use std::sync::mpsc;
use std::thread;
use std::time::Duration;

use crate::WaitResult;

macro_rules! check_result {
    ( $result:expr ) => {{
        use libc::EINTR;

        // https://github.com/rust-lang/rust/blob/49c68bd53f90e375bfb3cbba8c1c67a9e0adb9c0/src/libstd/sys/unix/process/process_unix.rs#L432-L441
        match $result {
            Ok(result) => result,
            Err(error) => {
                if error.raw_os_error() != Some(EINTR) {
                    break Err(error);
                }
                continue;
            }
        }
    }};
}

attr_alias::eval_block! {
    #[attr_alias(unix_waitid, cfg_attr(*, path = "waitid.rs"))]
    #[attr_alias(unix_waitid, cfg_attr(not(*), path = "common.rs"))]
    mod imp;
    pub(super) use imp::wait;
}

fn run_with_time_limit<F, R>(
    run_fn: F,
    time_limit: Option<Duration>,
) -> WaitResult<R>
where
    F: 'static + FnOnce() -> R + Send,
    R: 'static + Send,
{
    let Some(time_limit) = time_limit else {
        return Ok(Some(run_fn()));
    };

    let (result_sender, result_receiver) = mpsc::channel();
    thread::Builder::new()
        .spawn(move || result_sender.send(run_fn()))
        .map(|_| result_receiver.recv_timeout(time_limit).ok())
}