procpilot 0.7.0

Production-grade subprocess runner with typed errors, retry, and timeout
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

//! Test helper: write `N` bytes of `A` characters (0x41) to stdout.
//!
//! Usage: `pp_spam <byte-count>`
//!
//! Used to exercise pipe-buffer handling in timeout tests.
//! Not part of procpilot's public API.

use std::io::Write;

fn main() {
    let count: usize = std::env::args()
        .nth(1)
        .and_then(|a| a.parse().ok())
        .unwrap_or(0);

    let chunk = vec![b'A'; 4096];
    let mut remaining = count;
    let mut stdout = std::io::stdout().lock();

    while remaining > 0 {
        let n = remaining.min(chunk.len());
        if stdout.write_all(&chunk[..n]).is_err() {
            return;
        }
        remaining -= n;
    }
    let _ = stdout.flush();
}