ds: awaitpid: do not clobber entries for reaped processes

[public-inbox.git] / lib / PublicInbox / DS.pm

diff --git a/lib/PublicInbox/DS.pm b/lib/PublicInbox/DS.pm
index 523d47e4255ecf9fea2d7dab9aec756364febc24..0a763d0ef643ad0d5c5deb8e027b9df67d4438ae 100644 (file)
--- a/lib/PublicInbox/DS.pm
+++ b/lib/PublicInbox/DS.pm
@@ -699,8 +699,8 @@ sub long_response ($$;@) {
 }
 
 sub awaitpid {
-       my ($pid, @cb_args) = @_;
-       $AWAIT_PIDS->{$pid} //= @cb_args ? \@cb_args : 0;
+       my ($pid, @cb_args) = @_; # @cb_args = ($cb, @args), $cb may be undef
+       $AWAIT_PIDS->{$pid} = \@cb_args if @cb_args;
        # provide synchronous API
        if (defined(wantarray) || (!$in_loop && !@cb_args)) {
                my $ret;
@@ -716,9 +716,9 @@ again:
                        delete $AWAIT_PIDS->{$pid};
                }
                return $ret;
+       } elsif ($in_loop) { # We could've just missed our SIGCHLD, cover it, here:
+               enqueue_reap();
        }
-       # We could've just missed our SIGCHLD, cover it, here:
-       enqueue_reap() if $in_loop;
 }
 
 1;