]> Sergey Matveev's repositories - public-inbox.git/blobdiff - lib/PublicInbox/V2Writable.pm
searchidx: log2stack: simplify callers
[public-inbox.git] / lib / PublicInbox / V2Writable.pm
index aa812a6b3b3b4284329bcac6d447e8a5adbe6192..a403f22feb2220eaca68b3ea6b58016a7e6b9e0b 100644 (file)
@@ -65,11 +65,13 @@ sub nproc_shards ($) {
 
 sub count_shards ($) {
        my ($self) = @_;
-       # always load existing shards in case core count changes:
-       # Also, shard count may change while -watch is running
-       my $srch = $self->{ibx}->search or return 0;
-       delete $self->{ibx}->{search};
-       $srch->{nshard} // 0
+       $self->{ibx} ? do {
+               # always load existing shards in case core count changes:
+               # Also, shard count may change while -watch is running
+               my $srch = $self->{ibx}->search or return 0;
+               delete $self->{ibx}->{search};
+               $srch->{nshard} // 0
+       } : $self->{nshard}; # self->{nshard} is for ExtSearchIdx
 }
 
 sub new {
@@ -118,7 +120,7 @@ sub init_inbox {
        $self->idx_init;
        $self->{mm}->skip_artnum($skip_artnum) if defined $skip_artnum;
        my $epoch_max = -1;
-       git_dir_latest($self, \$epoch_max);
+       $self->{ibx}->git_dir_latest(\$epoch_max);
        if (defined $skip_epoch && $epoch_max == -1) {
                $epoch_max = $skip_epoch;
        }
@@ -133,12 +135,17 @@ sub add {
        $self->{ibx}->with_umask(\&_add, $self, $eml, $check_cb);
 }
 
+sub idx_shard ($$) {
+       my ($self, $num) = @_;
+       $self->{idx_shards}->[$num % scalar(@{$self->{idx_shards}})];
+}
+
 # indexes a message, returns true if checkpointing is needed
 sub do_idx ($$$$) {
        my ($self, $msgref, $mime, $smsg) = @_;
        $smsg->{bytes} = $smsg->{raw_bytes} + crlf_adjust($$msgref);
        $self->{oidx}->add_overview($mime, $smsg);
-       my $idx = idx_shard($self, $smsg->{num} % $self->{shards});
+       my $idx = idx_shard($self, $smsg->{num});
        $idx->index_raw($msgref, $mime, $smsg);
        my $n = $self->{transact_bytes} += $smsg->{raw_bytes};
        $n >= $self->{batch_bytes};
@@ -249,11 +256,6 @@ sub v2_num_for_harder {
        ($num, $mid0);
 }
 
-sub idx_shard {
-       my ($self, $shard_i) = @_;
-       $self->{idx_shards}->[$shard_i];
-}
-
 sub _idx_init { # with_umask callback
        my ($self, $opt) = @_;
        $self->lock_acquire unless $opt && $opt->{-skip_lock};
@@ -318,12 +320,13 @@ sub idx_init {
 sub _replace_oids ($$$) {
        my ($self, $mime, $replace_map) = @_;
        $self->done;
-       my $pfx = "$self->{ibx}->{inboxdir}/git";
+       my $ibx = $self->{ibx};
+       my $pfx = "$ibx->{inboxdir}/git";
        my $rewrites = []; # epoch => commit
        my $max = $self->{epoch_max};
 
        unless (defined($max)) {
-               defined(my $latest = git_dir_latest($self, \$max)) or return;
+               defined(my $latest = $ibx->git_dir_latest(\$max)) or return;
                $self->{epoch_max} = $max;
        }
 
@@ -420,7 +423,7 @@ sub rewrite_internal ($$;$$$) {
                        } else { # ->purge or ->remove
                                $self->{mm}->num_delete($num);
                        }
-                       unindex_oid_remote($self, $oid, $mid);
+                       unindex_oid_aux($self, $oid, $mid);
                }
        }
 
@@ -606,10 +609,11 @@ sub checkpoint ($;$) {
        }
        my $shards = $self->{idx_shards};
        if ($shards) {
-               my $dbh = $self->{mm}->{dbh};
+               my $mm = $self->{mm};
+               my $dbh = $mm->{dbh} if $mm;
 
                # SQLite msgmap data is second in importance
-               $dbh->commit;
+               $dbh->commit if $dbh;
 
                # SQLite overview is third
                $self->{oidx}->commit_lazy;
@@ -628,12 +632,13 @@ sub checkpoint ($;$) {
                }
 
                # last_commit is special, don't commit these until
-               # remote shards are done:
-               $dbh->begin_work;
+               # Xapian shards are done:
+               $dbh->begin_work if $dbh;
                set_last_commits($self);
-               $dbh->commit;
-
-               $dbh->begin_work;
+               if ($dbh) {
+                       $dbh->commit;
+                       $dbh->begin_work;
+               }
        }
        $self->{total_bytes} += $self->{transact_bytes};
        $self->{transact_bytes} = 0;
@@ -746,23 +751,6 @@ sub git_init {
        $git_dir
 }
 
-sub git_dir_latest {
-       my ($self, $max) = @_;
-       $$max = -1;
-       my $pfx = "$self->{ibx}->{inboxdir}/git";
-       return unless -d $pfx;
-       my $latest;
-       opendir my $dh, $pfx or die "opendir $pfx: $!\n";
-       while (defined(my $git_dir = readdir($dh))) {
-               $git_dir =~ m!\A([0-9]+)\.git\z! or next;
-               if ($1 > $$max) {
-                       $$max = $1;
-                       $latest = "$pfx/$git_dir";
-               }
-       }
-       $latest;
-}
-
 sub importer {
        my ($self) = @_;
        my $im = $self->{im};
@@ -781,7 +769,7 @@ sub importer {
        }
        my $epoch = 0;
        my $max;
-       my $latest = git_dir_latest($self, \$max);
+       my $latest = $self->{ibx}->git_dir_latest(\$max);
        if (defined $latest) {
                my $git = PublicInbox::Git->new($latest);
                my $packed_bytes = $git->packed_bytes;
@@ -975,8 +963,6 @@ sub update_last_commit {
        last_epoch_commit($self, $i, $cmt);
 }
 
-sub git_dir_n ($$) { "$_[0]->{ibx}->{inboxdir}/git/$_[1].git" }
-
 sub last_commits ($$) {
        my ($self, $epoch_max) = @_;
        my $heads = [];
@@ -987,8 +973,8 @@ sub last_commits ($$) {
 }
 
 # returns a revision range for git-log(1)
-sub log_range ($$$$$) {
-       my ($self, $sync, $git, $i, $tip) = @_;
+sub log_range ($$$$) {
+       my ($sync, $git, $i, $tip) = @_;
        my $opt = $sync->{-opt};
        my $pr = $opt->{-progress} if (($opt->{verbose} || 0) > 1);
        my $cur = $sync->{ranges}->[$i] or do {
@@ -1041,14 +1027,14 @@ sub sync_prepare ($$$) {
        my ($self, $sync, $epoch_max) = @_;
        my $pr = $sync->{-opt}->{-progress};
        my $regen_max = 0;
-       my $head = $self->{ibx}->{ref_head} || 'HEAD';
+       my $head = $sync->{ibx}->{ref_head} || 'HEAD';
 
        # reindex stops at the current heads and we later rerun index_sync
        # without {reindex}
        my $reindex_heads = $self->last_commits($epoch_max) if $sync->{reindex};
 
        for (my $i = $epoch_max; $i >= 0; $i--) {
-               my $git_dir = git_dir_n($self, $i);
+               my $git_dir = $sync->{ibx}->git_dir_n($i);
                -d $git_dir or next; # missing epochs are fine
                my $git = PublicInbox::Git->new($git_dir);
                if ($reindex_heads) {
@@ -1057,7 +1043,7 @@ sub sync_prepare ($$$) {
                chomp(my $tip = $git->qx(qw(rev-parse -q --verify), $head));
 
                next if $?; # new repo
-               my $range = log_range($self, $sync, $git, $i, $tip) or next;
+               my $range = log_range($sync, $git, $i, $tip) or next;
                # can't use 'rev-list --count' if we use --diff-filter
                $pr->("$i.git counting $range ... ") if $pr;
                # Don't bump num_highwater on --reindex by using {D}.
@@ -1065,7 +1051,7 @@ sub sync_prepare ($$$) {
                # because we want NNTP article number gaps from unindexed
                # messages to show up in mirrors, too.
                $sync->{D} //= $sync->{reindex} ? {} : undef; # OID_BIN => NR
-               my $stk = log2stack($sync, $git, $range, $self->{ibx});
+               my $stk = log2stack($sync, $git, $range);
                my $nr = $stk ? $stk->num_records : 0;
                $pr->("$nr\n") if $pr;
                $sync->{stacks}->[$i] = $stk if $stk;
@@ -1098,11 +1084,11 @@ sub sync_prepare ($$$) {
        $regen_max + $self->{mm}->num_highwater() || 0;
 }
 
-sub unindex_oid_remote ($$$) {
+sub unindex_oid_aux ($$$) {
        my ($self, $oid, $mid) = @_;
        my @removed = $self->{oidx}->remove_oid($oid, $mid);
        for my $num (@removed) {
-               my $idx = idx_shard($self, $num % $self->{shards});
+               my $idx = idx_shard($self, $num);
                $idx->shard_remove($oid, $num);
        }
 }
@@ -1133,7 +1119,7 @@ sub unindex_oid ($$;$) { # git->cat_async callback
                        }
                        $mm->num_delete($num);
                }
-               unindex_oid_remote($self, $oid, $mid);
+               unindex_oid_aux($self, $oid, $mid);
        }
 }
 
@@ -1141,7 +1127,7 @@ sub git { $_[0]->{ibx}->git }
 
 # this is rare, it only happens when we get discontiguous history in
 # a mirror because the source used -purge or -edit
-sub unindex ($$$$) {
+sub unindex_epoch ($$$$) {
        my ($self, $sync, $git, $unindex_range) = @_;
        my $unindexed = $sync->{unindexed} //= {}; # $mid0 => $num
        my $before = scalar keys %$unindexed;
@@ -1183,7 +1169,7 @@ sub sync_ranges ($$$) {
 sub index_xap_only { # git->cat_async callback
        my ($bref, $oid, $type, $size, $smsg) = @_;
        my $self = $smsg->{v2w};
-       my $idx = idx_shard($self, $smsg->{num} % $self->{shards});
+       my $idx = idx_shard($self, $smsg->{num});
        $smsg->{raw_bytes} = $size;
        $idx->index_raw($bref, undef, $smsg);
        $self->{transact_bytes} += $size;
@@ -1214,11 +1200,11 @@ sub index_xap_step ($$$;$) {
 sub index_epoch ($$$) {
        my ($self, $sync, $i) = @_;
 
-       my $git_dir = git_dir_n($self, $i);
+       my $git_dir = $sync->{ibx}->git_dir_n($i);
        -d $git_dir or return; # missing epochs are fine
        my $git = PublicInbox::Git->new($git_dir);
        if (my $unindex_range = delete $sync->{unindex_range}->{$i}) { # rare
-               unindex($self, $sync, $git, $unindex_range);
+               unindex_epoch($self, $sync, $git, $unindex_range);
        }
        defined(my $stk = $sync->{stacks}->[$i]) or return;
        $sync->{stacks}->[$i] = undef;
@@ -1282,7 +1268,7 @@ sub index_sync {
 
        my $pr = $opt->{-progress};
        my $epoch_max;
-       my $latest = git_dir_latest($self, \$epoch_max);
+       my $latest = $self->{ibx}->git_dir_latest(\$epoch_max);
        return unless defined $latest;
 
        my $seq = $opt->{sequential_shard};
@@ -1299,6 +1285,7 @@ sub index_sync {
                reindex => $opt->{reindex},
                -opt => $opt,
                v2w => $self,
+               ibx => $self->{ibx},
        };
        $sync->{ranges} = sync_ranges($self, $sync, $epoch_max);
        if (sync_prepare($self, $sync, $epoch_max)) {