lei + ipc: simplify process reaping

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

diff --git a/lib/PublicInbox/LeiP2q.pm b/lib/PublicInbox/LeiP2q.pm
index e7ddc852e492cfe500aa8b9ce31215f2bf498acb..08ec81c5295ee3c43831dad38e3ef313c1e51dd6 100644 (file)
--- a/lib/PublicInbox/LeiP2q.pm
+++ b/lib/PublicInbox/LeiP2q.pm
@@ -105,17 +105,23 @@ sub do_p2q { # via wq_do
        }
        my $smsg = bless {}, 'PublicInbox::Smsg';
        my $in = $self->{0};
+       my @cmd;
        unless ($in) {
                my $input = $self->{input};
-               if (-e $input) {
-                       $in = $lei->fopen('<', $input) or
+               my $devfd = $lei->path_to_fd($input) // return;
+               if ($devfd >= 0) {
+                       $in = $lei->{$devfd};
+               } elsif (-e $input) {
+                       open($in, '<', $input) or
                                return $lei->fail("open < $input: $!");
                } else {
-                       my @cmd = (qw(git format-patch --stdout -1), $input);
+                       @cmd = (qw(git format-patch --stdout -1), $input);
                        $in = popen_rd(\@cmd, undef, { 2 => $lei->{2} });
                }
        };
-       my $eml = PublicInbox::Eml->new(\(do { local $/; <$in> }));
+       my $str = do { local $/; <$in> };
+       @cmd && !close($in) and return $lei->fail("E: @cmd failed: $?");
+       my $eml = PublicInbox::Eml->new(\$str);
        $lei->{diff_want} = +{ map { $_ => 1 } @want };
        $smsg->populate($eml);
        while (my ($pfx, $fields) = each %pfx2smsg) {
@@ -129,7 +135,7 @@ sub do_p2q { # via wq_do
        if ($lei->{opt}->{debug}) {
                my $json = ref(PublicInbox::Config->json)->new;
                $json->utf8->canonical->pretty;
-               $lei->err($json->encode($lei->{qterms}));
+               print { $lei->{2} } $json->encode($lei->{qterms});
        }
        my (@q, %seen);
        for my $pfx (@want) {
@@ -144,7 +150,7 @@ sub do_p2q { # via wq_do
                        my $end = ($pfx =~ s/([0-9\*]+)\z//) ? $1 : '';
                        my $x = delete($lei->{qterms}->{$pfx}) or next;
                        my $star = $end =~ tr/*//d ? '*' : '';
-                       my $min_len = ($end // 0) + 0;
+                       my $min_len = ($end || 0) + 0;
 
                        # no wildcards for bool_pfx_external
                        $star = '' if $pfx =~ /\A(dfpre|dfpost|mid)\z/;
@@ -174,27 +180,24 @@ sub do_p2q { # via wq_do
        $lei->out(@q, "\n");
 }
 
-sub call { # the "lei patch-to-query" entry point
-       my ($cls, $lei, $input) = @_;
-       my $self = $lei->{p2q} = bless {}, $cls;
+sub lei_p2q { # the "lei patch-to-query" entry point
+       my ($lei, $input) = @_;
+       my $self = bless {}, __PACKAGE__;
        if ($lei->{opt}->{stdin}) {
-               $self->{0} = delete $lei->{0}; # guard from lei_atfork_child
+               $self->{0} = delete $lei->{0}; # guard from _lei_atfork_child
        } else {
                $self->{input} = $input;
        }
-       my $op = $lei->workers_start($self, 'lei patch2query', 1, {
-               '' => [ $lei->{p2q_done} // $lei->can('dclose'), $lei ]
-       });
+       my ($op_c, $ops) = $lei->workers_start($self, 1);
+       $lei->{wq1} = $self;
        $self->wq_io_do('do_p2q', []);
-       $self->wq_close(1);
-       while ($op && $op->{sock}) { $op->event_step }
+       $self->wq_close;
+       $lei->wait_wq_events($op_c, $ops);
 }
 
 sub ipc_atfork_child {
        my ($self) = @_;
-       my $lei = $self->{lei};
-       $lei->lei_atfork_child;
-       $SIG{__WARN__} = PublicInbox::Eml::warn_ignore_cb();
+       $self->{lei}->_lei_atfork_child;
        $self->SUPER::ipc_atfork_child;
 }