Zig 中的 child process

[jiacai2050]

在工作中，难免会去重复执行某些任务，直到其成功为止。比如下载依赖，之前我的做法是用 Shell

while ! do something; do echo retry; done

这样虽然可以，但是有些麻烦，于是就想能不能实现个 cli，来替我们执行，如果用 C 实现的话，可以预想到的，就是 fork + exec，貌似也不是很麻烦。

但是如果要在主进程中，等待 child 完成之前，收集 child 的 stdout，而不堵塞主进程，那就要采用异步的 API 来轮训 stdout，比如 select
或 poll， Zig 里面有个 child_process.zig，它里面把 fork + exec + poll 都封装好了，直接使用即可，一些关键函数：

/// Collect the output from the process's stdout and stderr. Will return once all output
/// has been collected. This does not mean that the process has ended. `wait` should still
/// be called to wait for and clean up the process.
///
/// The process must be started with stdout_behavior and stderr_behavior == .Pipe
pub fn collectOutput(
    child: ChildProcess,
    stdout: *std.ArrayList(u8),
    stderr: *std.ArrayList(u8),
    max_output_bytes: usize,
) !void {
    debug.assert(child.stdout_behavior == .Pipe);
    debug.assert(child.stderr_behavior == .Pipe);

    // we could make this work with multiple allocators but YAGNI
    if (stdout.allocator.ptr != stderr.allocator.ptr or
        stdout.allocator.vtable != stderr.allocator.vtable)
        @panic("ChildProcess.collectOutput only supports 1 allocator");

    var poller = std.io.poll(stdout.allocator, enum { stdout, stderr }, .{
        .stdout = child.stdout.?,
        .stderr = child.stderr.?,
    });
    defer poller.deinit();

    while (try poller.poll()) {
        if (poller.fifo(.stdout).count > max_output_bytes)
            return error.StdoutStreamTooLong;
        if (poller.fifo(.stderr).count > max_output_bytes)
            return error.StderrStreamTooLong;
    }

    stdout.* = fifoToOwnedArrayList(poller.fifo(.stdout));
    stderr.* = fifoToOwnedArrayList(poller.fifo(.stderr));
}

这是收集 stdout 的，创建进程的地方在 spawnPosix 中，感兴趣的可以自行查看，后来有机会再来介绍。

最后，下面是我基于 Zig child process 实现的 repeat 命令：

• https://github.com/jiacai2050/zigcli/blob/main/src/repeat.zig

推荐阅读

• microHOWTO: Capture the output of a child process in C
• Forking is Cool, or: A Unix Shell in Zig, or: Dave Learns How to Fork