neurocyte/thespian
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: wait and retry stdin writes on error.WouldBlock in subprocess

Browse source
This commit is contained in:
CJ van den Berg 2024-03-12 23:04:38 +01:00
parent a740c03c11
commit 5bb05db166
1 changed files with 10 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
src/subprocess.zig
View file

@ -152,7 +152,16 @@ const Proc = struct {
} else if (try m.match(.{ "fd", "stdin", "write_ready" })) { } else if (try m.match(.{ "fd", "stdin", "write_ready" })) {
if (self.stdin_buffer.items.len > 0) { if (self.stdin_buffer.items.len > 0) {
if (self.child.stdin) |stdin| { if (self.child.stdin) |stdin| {
const written = stdin.write(self.stdin_buffer.items) catch |e| return tp.exit_error(e); const written = stdin.write(self.stdin_buffer.items) catch |e| switch (e) {
error.WouldBlock => {
if (self.fd_stdin) |fd_stdin| {
fd_stdin.wait_write() catch |e_| return tp.exit_error(e_);
self.write_pending = true;
return;
} else return tp.exit_error(error.WouldBlock);
},
else => return tp.exit_error(e),
};
self.write_pending = false; self.write_pending = false;
defer { defer {
if (self.stdin_close_pending and !self.write_pending) if (self.stdin_close_pending and !self.write_pending)