corsa 0.7.0

Production-oriented Rust bindings, orchestration layers, and Node integration for typescript-go
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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99

use std::{
    process::{Child, Command, ExitStatus, Stdio},
    thread,
    time::{Duration, Instant},
};

use corsa::{Result, TsgoError, fast::CompactString};
use corsa_core::terminate_child_process;

pub fn run_command(
    command: &mut Command,
    timeout: Duration,
    expected_exit_codes: &[i32],
    label: &str,
) -> Result<()> {
    let started = Instant::now();
    let mut child = ManagedChild::spawn(command)?;
    loop {
        if let Some(status) = child.try_wait()? {
            child.disarm();
            return validate_exit_status(status, expected_exit_codes, label);
        }
        if started.elapsed() >= timeout {
            child.terminate()?;
            return Err(TsgoError::Protocol(CompactString::from(format!(
                "{label} timed out after {} ms",
                timeout.as_millis()
            ))));
        }
        thread::sleep(Duration::from_millis(10));
    }
}

struct ManagedChild {
    child: Option<Child>,
}

impl ManagedChild {
    fn spawn(command: &mut Command) -> Result<Self> {
        command
            .stdin(Stdio::null())
            .stdout(Stdio::null())
            .stderr(Stdio::null());
        Ok(Self {
            child: Some(command.spawn()?),
        })
    }

    fn try_wait(&mut self) -> std::io::Result<Option<ExitStatus>> {
        self.child
            .as_mut()
            .ok_or_else(|| std::io::Error::other("child already reaped"))?
            .try_wait()
    }

    fn disarm(&mut self) {
        self.child.take();
    }

    fn terminate(&mut self) -> Result<()> {
        if let Some(child) = self.child.as_mut() {
            terminate_child_process(child)?;
        }
        self.disarm();
        Ok(())
    }
}

impl Drop for ManagedChild {
    fn drop(&mut self) {
        if let Some(child) = self.child.as_mut() {
            let _ = terminate_child_process(child);
        }
    }
}

fn validate_exit_status(
    status: ExitStatus,
    expected_exit_codes: &[i32],
    label: &str,
) -> Result<()> {
    if matches_expected_exit(status, expected_exit_codes) {
        return Ok(());
    }
    let rendered = status
        .code()
        .map(|code| code.to_string())
        .unwrap_or_else(|| "terminated by signal".to_string());
    Err(TsgoError::Protocol(CompactString::from(format!(
        "{label} exited with unexpected status {rendered}"
    ))))
}

fn matches_expected_exit(status: ExitStatus, expected_exit_codes: &[i32]) -> bool {
    if let Some(code) = status.code() {
        return expected_exit_codes.contains(&code);
    }
    status.success() && expected_exit_codes.contains(&0)
}