]> Sergey Matveev's repositories - public-inbox.git/blobdiff - lib/PublicInbox/SearchIdx.pm
searchidx: index "diff --git a/... b/..." headers
[public-inbox.git] / lib / PublicInbox / SearchIdx.pm
index f4f3ba34f5c951f3a0b3045a7babbb3cbd0f0934..6e2e614c5085c7e45fa33ad06375a9137af1ee31 100644 (file)
@@ -94,7 +94,7 @@ sub need_xapian ($) { $_[0]->{indexlevel} =~ $xapianlevels }
 sub idx_release {
        my ($self, $wake) = @_;
        if (need_xapian($self)) {
-               my $xdb = delete $self->{xdb} or croak 'not acquired';
+               my $xdb = delete $self->{xdb} or croak '{xdb} not acquired';
                $xdb->close;
        }
        $self->lock_release($wake) if $self->{creat};
@@ -103,7 +103,7 @@ sub idx_release {
 
 sub load_xapian_writable () {
        return 1 if $X->{WritableDatabase};
-       PublicInbox::Search::load_xapian() or return;
+       PublicInbox::Search::load_xapian() or die "failed to load Xapian: $@\n";
        my $xap = $PublicInbox::Search::Xap;
        for (qw(Document TermGenerator WritableDatabase)) {
                $X->{$_} = $xap.'::'.$_;
@@ -259,6 +259,7 @@ sub index_diff ($$$) {
                } elsif (m!^diff --git "?[^/]+/.+ "?[^/]+/.+\z!) {
                        # wait until "---" and "+++" to capture filenames
                        $in_diff = 1;
+                       push @xnq, $_;
                # traditional diff:
                } elsif (m/^diff -(.+) (\S+) (\S+)$/) {
                        my ($opt, $fa, $fb) = ($1, $2, $3);
@@ -451,10 +452,9 @@ sub add_xapian ($$$$) {
 sub _msgmap_init ($) {
        my ($self) = @_;
        die "BUG: _msgmap_init is only for v1\n" if $self->{ibx}->version != 1;
-       $self->{mm} //= eval {
+       $self->{mm} //= do {
                require PublicInbox::Msgmap;
-               my $rw = $self->{ibx}->{-no_fsync} ? 2 : 1;
-               PublicInbox::Msgmap->new($self->{ibx}->{inboxdir}, $rw);
+               PublicInbox::Msgmap->new_file($self->{ibx}, 1);
        };
 }
 
@@ -518,6 +518,12 @@ sub add_eidx_info {
        $self->{xdb}->replace_document($docid, $doc);
 }
 
+sub get_terms {
+       my ($self, $pfx, $docid) = @_;
+       begin_txn_lazy($self);
+       xap_terms($pfx, $self->{xdb}, $docid);
+}
+
 sub remove_eidx_info {
        my ($self, $docid, $eidx_key, $eml) = @_;
        begin_txn_lazy($self);
@@ -637,14 +643,24 @@ sub update_vmd {
 
 sub xdb_remove {
        my ($self, @docids) = @_;
-       $self->begin_txn_lazy;
-       my $xdb = $self->{xdb} or return;
+       begin_txn_lazy($self);
+       my $xdb = $self->{xdb} // die 'BUG: missing {xdb}';
        for my $docid (@docids) {
                eval { $xdb->delete_document($docid) };
                warn "E: #$docid not in in Xapian? $@\n" if $@;
        }
 }
 
+sub xdb_remove_quiet {
+       my ($self, $docid) = @_;
+       begin_txn_lazy($self);
+       my $xdb = $self->{xdb} // die 'BUG: missing {xdb}';
+       eval { $xdb->delete_document($docid) };
+       ++$self->{-quiet_rm} unless $@;
+}
+
+sub nr_quiet_rm { delete($_[0]->{-quiet_rm}) // 0 }
+
 sub index_git_blob_id {
        my ($doc, $pfx, $objid) = @_;
 
@@ -750,7 +766,8 @@ sub index_sync {
        my ($self, $opt) = @_;
        delete $self->{lock_path} if $opt->{-skip_lock};
        $self->with_umask(\&_index_sync, $self, $opt);
-       if ($opt->{reindex} && !$opt->{quit}) {
+       if ($opt->{reindex} && !$opt->{quit} &&
+                       !grep(defined, @$opt{qw(since until)})) {
                my %again = %$opt;
                delete @again{qw(rethread reindex)};
                index_sync($self, \%again);
@@ -775,18 +792,19 @@ sub v1_checkpoint ($$;$) {
        # $newest may be undef
        my $newest = $stk ? $stk->{latest_cmt} : ${$sync->{latest_cmt}};
        if (defined($newest)) {
-               my $cur = $self->{mm}->last_commit || '';
-               if (need_update($self, $cur, $newest)) {
+               my $cur = $self->{mm}->last_commit;
+               if (need_update($self, $sync, $cur, $newest)) {
                        $self->{mm}->last_commit($newest);
                }
        }
        ${$sync->{max}} = $self->{batch_bytes};
 
        $self->{mm}->{dbh}->commit;
-       my $xdb = need_xapian($self) ? $self->{xdb} : undef;
+       eval { $self->{mm}->{dbh}->do('PRAGMA optimize') };
+       my $xdb = $self->{xdb};
        if ($newest && $xdb) {
                my $cur = $xdb->get_metadata('last_commit');
-               if (need_update($self, $cur, $newest)) {
+               if (need_update($self, $sync, $cur, $newest)) {
                        $xdb->set_metadata('last_commit', $newest);
                }
        }
@@ -870,9 +888,14 @@ sub log2stack ($$$) {
 
        # Count the new files so they can be added newest to oldest
        # and still have numbers increasing from oldest to newest
-       my $fh = $git->popen(qw(log --raw -r --pretty=tformat:%at-%ct-%H
-                               --no-notes --no-color --no-renames --no-abbrev),
-                               $range);
+       my @cmd = qw(log --raw -r --pretty=tformat:%at-%ct-%H
+                       --no-notes --no-color --no-renames --no-abbrev);
+       for my $k (qw(since until)) {
+               my $v = $sync->{-opt}->{$k} // next;
+               next if !$sync->{-opt}->{reindex};
+               push @cmd, "--$k=$v";
+       }
+       my $fh = $git->popen(@cmd, $range);
        my ($at, $ct, $stk, $cmt);
        while (<$fh>) {
                return if $sync->{quit};
@@ -928,10 +951,17 @@ sub is_ancestor ($$$) {
        $? == 0;
 }
 
-sub need_update ($$$) {
-       my ($self, $cur, $new) = @_;
+sub need_update ($$$$) {
+       my ($self, $sync, $cur, $new) = @_;
        my $git = $self->{ibx}->git;
-       return 1 if $cur && !is_ancestor($git, $cur, $new);
+       $cur //= ''; # XS Search::Xapian ->get_metadata doesn't give undef
+
+       # don't rewind if --{since,until,before,after} are in use
+       return if $cur ne '' &&
+               grep(defined, @{$sync->{-opt}}{qw(since until)}) &&
+               is_ancestor($git, $new, $cur);
+
+       return 1 if $cur ne '' && !is_ancestor($git, $cur, $new);
        my $range = $cur eq '' ? $new : "$cur..$new";
        chomp(my $n = $git->qx(qw(rev-list --count), $range));
        ($n eq '' || $n > 0);
@@ -1085,25 +1115,4 @@ sub eidx_shard_new {
        $self;
 }
 
-# ensure there's no stale Xapian docs by treating $over as canonical
-sub over_check {
-       my ($self, $over) = @_;
-       begin_txn_lazy($self);
-       my $sth = $over->dbh->prepare(<<'');
-SELECT COUNT(*) FROM over WHERE num = ?
-
-       my $xdb = $self->{xdb};
-       my $cur = $xdb->postlist_begin('');
-       my $end = $xdb->postlist_end('');
-       my $xdir = $self->xdir;
-       for (; $cur != $end; $cur++) {
-               my $docid = $cur->get_docid;
-               $sth->execute($docid);
-               my $x = $sth->fetchrow_array;
-               next if $x > 0;
-               warn "I: removing $xdir #$docid, not in `over'\n";
-               $xdb->delete_document($docid);
-       }
-}
-
 1;