]> Sergey Matveev's repositories - public-inbox.git/blobdiff - lib/PublicInbox/LeiNoteEvent.pm
lei: commit store on interrupted partial imports
[public-inbox.git] / lib / PublicInbox / LeiNoteEvent.pm
index d6511cf6cd1bb750d9bfe33d874c243510dae68b..db387633e877c9e3b8bbfd3c4458a12b8695188a 100644 (file)
@@ -2,27 +2,26 @@
 # License: AGPL-3.0+ <https://www.gnu.org/licenses/agpl-3.0.txt>
 
 # internal command for dealing with inotify, kqueue vnodes, etc
+# it is a semi-persistent worker
 package PublicInbox::LeiNoteEvent;
 use strict;
 use v5.10.1;
 use parent qw(PublicInbox::IPC);
+use PublicInbox::DS;
+use Errno qw(ENOENT);
 
-my $flush_timer;
 our $to_flush; # { cfgpath => $lei }
 
-sub flush_lei ($) {
-       my ($lei) = @_;
-       if (my $lne = delete $lei->{cfg}->{-lei_note_event}) {
-               $lne->wq_close(1, undef, $lei); # runs _lei_wq_eof;
-       } elsif ($lei->{sto}) { # lms_clear_src calls only:
-               my $wait = $lei->{sto}->ipc_do('done');
-       }
+sub flush_lei ($;$) {
+       my ($lei, $manual) = @_;
+       my $lne = delete $lei->{cfg}->{-lei_note_event} // return;
+       $lne->{lei_sock} = $lei->{sock} if $manual;
+       $lne->wq_close; # runs _lei_wq_eof;
 }
 
 # we batch up writes and flush every 5s (matching Linux default
 # writeback behavior) since MUAs can trigger a storm of inotify events
 sub flush_task { # PublicInbox::DS timer callback
-       undef $flush_timer;
        my $todo = $to_flush // return;
        $to_flush = undef;
        for my $lei (values %$todo) { flush_lei($lei) }
@@ -31,37 +30,41 @@ sub flush_task { # PublicInbox::DS timer callback
 # sets a timer to flush
 sub note_event_arm_done ($) {
        my ($lei) = @_;
-       $flush_timer //= PublicInbox::DS::add_timer(5, \&flush_task);
+       PublicInbox::DS::add_uniq_timer('flush_timer', 5, \&flush_task);
        $to_flush->{$lei->{cfg}->{'-f'}} //= $lei;
 }
 
 sub eml_event ($$$$) {
-       my ($self, $eml, $kw, $state) = @_;
+       my ($self, $eml, $vmd, $state) = @_;
        my $sto = $self->{lei}->{sto};
-       my $lse = $self->{lse} //= $sto->search;
-       my $vmd = { kw => $kw };
        if ($state =~ /\Aimport-(?:rw|ro)\z/) {
-               $sto->ipc_do('set_eml', $eml, $vmd);
+               $sto->wq_do('set_eml', $eml, $vmd);
        } elsif ($state =~ /\Aindex-(?:rw|ro)\z/) {
                my $xoids = $self->{lei}->ale->xoids_for($eml);
-               $sto->ipc_do('index_eml_only', $eml, $vmd, $xoids);
+               $sto->wq_do('index_eml_only', $eml, $vmd, $xoids);
        } elsif ($state =~ /\Atag-(?:rw|ro)\z/) {
-               my $c = $lse->kw_changed($eml, $kw, my $docids = []);
+               my $docids = [];
+               my $c = eval {
+                       $self->{lse}->kw_changed($eml, $vmd->{kw}, $docids);
+               } // 1; # too new, assume changed since still to-be-committed.
                if (scalar @$docids) { # already in lei/store
-                       $sto->ipc_do('set_eml_vmd', undef, $vmd, $docids) if $c;
+                       $sto->wq_do('set_eml_vmd', undef, $vmd, $docids) if $c;
                } elsif (my $xoids = $self->{lei}->ale->xoids_for($eml)) {
                        # it's in an external, only set kw, here
-                       $sto->ipc_do('set_xvmd', $xoids, $eml, $vmd);
-               } # else { totally unknown
+                       $sto->wq_do('set_xvmd', $xoids, $eml, $vmd);
+               } # else { totally unknown: ignore
        } else {
                warn "unknown state: $state (in $self->{lei}->{cfg}->{'-f'})\n";
        }
 }
 
 sub maildir_event { # via wq_io_do
-       my ($self, $fn, $kw, $state) = @_;
-       my $eml = PublicInbox::InboxWritable::eml_from_path($fn) // return;
-       eml_event($self, $eml, $kw, $state);
+       my ($self, $fn, $vmd, $state) = @_;
+       if (my $eml = PublicInbox::InboxWritable::eml_from_path($fn)) {
+               eml_event($self, $eml, $vmd, $state);
+       } elsif ($! == ENOENT) {
+               $self->{lms}->clear_src(@{$vmd->{sync_info}});
+       } # else: eml_from_path already warns
 }
 
 sub lei_note_event {
@@ -69,25 +72,24 @@ sub lei_note_event {
        die "BUG: unexpected: @rest" if @rest;
        my $cfg = $lei->_lei_cfg or return; # gone (race)
        my $sto = $lei->_lei_store or return; # gone
-       return flush_lei($lei) if $folder eq 'done'; # special case
-       my $lms = $sto->search->lms or return;
-       my $err = $lms->arg2folder($lei, [ $folder ]);
-       return if $err->{fail};
-       undef $lms;
-       my $state = $cfg->get_1("watch.$folder", 'state') // 'pause';
+       return flush_lei($lei, 1) if $folder eq 'done'; # special case
+       my $lms = $lei->lms or return;
+       $lms->lms_write_prepare if $new_cur eq ''; # for ->clear_src below
+       $lei->{opt}->{quiet} = 1;
+       $lms->arg2folder($lei, [ $folder ]);
+       my $state = $cfg->get_1("watch.$folder.state") // 'tag-rw';
        return if $state eq 'pause';
+       return $lms->clear_src($folder, \$bn) if $new_cur eq '';
+       $lms->lms_pause;
        $lei->ale; # prepare
        $sto->write_prepare($lei);
-       if ($new_cur eq '') {
-               $sto->ipc_do('lms_clear_src', $folder, \$bn);
-               return note_event_arm_done($lei);
-       }
        require PublicInbox::MdirReader;
        my $self = $cfg->{-lei_note_event} //= do {
-               my $wq = bless {}, __PACKAGE__;
+               my $wq = bless { lms => $lms }, __PACKAGE__;
                # MUAs such as mutt can trigger massive rename() storms so
-               # use all CPU power available:
+               # use some CPU, but don't overwhelm slower storage, either
                my $jobs = $wq->detect_nproc // 1;
+               $jobs = 4 if $jobs > 4; # same default as V2Writable
                my ($op_c, $ops) = $lei->workers_start($wq, $jobs);
                $lei->wait_wq_events($op_c, $ops);
                note_event_arm_done($lei);
@@ -98,27 +100,21 @@ sub lei_note_event {
                        // return;
                return if index($fl, 'T') >= 0;
                my $kw = PublicInbox::MdirReader::flags2kw($fl);
-               $self->wq_io_do('maildir_event', [], $fn, $kw, $state);
+               my $vmd = { kw => $kw, sync_info => [ $folder, \$bn ] };
+               $self->wq_do('maildir_event', $fn, $vmd, $state);
        } # else: TODO: imap
 }
 
 sub ipc_atfork_child {
        my ($self) = @_;
        $self->{lei}->_lei_atfork_child(1); # persistent, for a while
+       $self->{lms}->lms_write_prepare;
+       $self->{lse} = $self->{lei}->{sto}->search;
        $self->SUPER::ipc_atfork_child;
 }
 
-sub lne_done_wait {
-       my ($arg, $pid) = @_;
-       my ($self, $lei) = @$arg;
-       $lei->can('wq_done_wait')->($arg, $pid);
-}
-
 sub _lei_wq_eof { # EOF callback for main lei daemon
-       my ($lei) = @_;
-       my $lne = delete $lei->{lne} or return $lei->fail;
-       my $wait = $lei->{sto}->ipc_do('done');
-       $lne->wq_wait_old(\&lne_done_wait, $lei);
+       $_[0]->wq_eof('lne');
 }
 
 1;