lei: commit store on interrupted partial imports

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

diff --git a/lib/PublicInbox/LeiPmdir.pm b/lib/PublicInbox/LeiPmdir.pm
index b71efe701e76289c928e92f4f2828ab8f1934040..d4aa0212bea7dfb4ce3766cf13eb20c040eb0ecc 100644 (file)
--- a/lib/PublicInbox/LeiPmdir.pm
+++ b/lib/PublicInbox/LeiPmdir.pm
@@ -1,4 +1,4 @@
-# Copyright (C) 2021 all contributors <meta@public-inbox.org>
+# Copyright (C) all contributors <meta@public-inbox.org>
 # License: AGPL-3.0+ <https://www.gnu.org/licenses/agpl-3.0.txt>
 
 # WQ worker for dealing with parallel Maildir reads;
@@ -25,43 +25,30 @@ sub new {
        my ($op_c, $ops) = $lei->workers_start($self, $nproc,
                undef, { ipt => $ipt }); # LeiInput subclass
        $op_c->{ops} = $ops; # for PktOp->event_step
+       $self->{lei_sock} = $lei->{sock}; # keep client for pmd_done_wait
        $lei->{pmd} = $self;
 }
 
 sub ipc_atfork_child {
        my ($self) = @_;
-       my $lei = $self->{lei};
-       $lei->_lei_atfork_child;
        my $ipt = $self->{ipt} // die 'BUG: no self->{ipt}';
-       $ipt->{lei} = $lei;
-       $ipt->{sto} = $lei->{sto} // die 'BUG: no lei->{sto}';
-       $ipt->{lse} = $ipt->{sto}->search;
-       $ipt->{over} = $ipt->{lse}->over;
-       $ipt->{-lms_ro} //= $ipt->{lse}->lms; # may be undef or '0'
-       $self->SUPER::ipc_atfork_child;
+       my $lei = $ipt->{lei} = $self->{lei};
+       delete @$lei{qw(auth net)}; # no network access in this worker
+       $ipt->ipc_atfork_child; # calls _lei_atfork_child;
 }
 
 sub each_mdir_fn { # maildir_each_file callback
-       my ($f, $self, @args) = @_;
-       $self->wq_io_do('mdir_iter', [], $f, @args);
+       my ($f, $fl, $self, @args) = @_;
+       $self->wq_io_do('mdir_iter', [], $f, $fl, @args);
 }
 
 sub mdir_iter { # via wq_io_do
-       my ($self, $f, @args) = @_;
-       $self->{ipt}->pmdir_cb($f, @args);
-}
-
-sub pmd_done_wait {
-       my ($arg, $pid) = @_;
-       my ($self, $lei) = @$arg;
-       my $wait = $lei->{sto}->ipc_do('done');
-       $lei->can('wq_done_wait')->($arg, $pid);
+       my ($self, $f, $fl, @args) = @_;
+       $self->{ipt}->pmdir_cb($f, $fl, @args);
 }
 
 sub _lei_wq_eof { # EOF callback for main lei daemon
-       my ($lei) = @_;
-       my $pmd = delete $lei->{pmd} or return $lei->fail;
-       $pmd->wq_wait_old(\&pmd_done_wait, $lei);
+       $_[0]->wq_eof('pmd');
 }
 
 1;