]> Sergey Matveev's repositories - public-inbox.git/blobdiff - lib/PublicInbox/SearchIdx.pm
searchidx: index eidx_key as a boolean term
[public-inbox.git] / lib / PublicInbox / SearchIdx.pm
index eb620f44fe990e4c38514d9375fa6328ae30e80d..5171c610da36a28bd2063baf0f87b06520c53a3a 100644 (file)
@@ -17,6 +17,7 @@ use PublicInbox::MsgIter;
 use PublicInbox::IdxStack;
 use Carp qw(croak);
 use POSIX qw(strftime);
+use Time::Local qw(timegm);
 use PublicInbox::OverIdx;
 use PublicInbox::Spawn qw(spawn nodatacow_dir);
 use PublicInbox::Git qw(git_unquote);
@@ -31,11 +32,11 @@ use constant DEBUG => !!$ENV{DEBUG};
 my $xapianlevels = qr/\A(?:full|medium)\z/;
 my $hex = '[a-f0-9]';
 my $OID = $hex .'{40,}';
+our $INDEXLEVELS = qr/\A(?:full|medium|basic)\z/;
 
 sub new {
        my ($class, $ibx, $creat, $shard) = @_;
        ref $ibx or die "BUG: expected PublicInbox::Inbox object: $ibx";
-       my $levels = qr/\A(?:full|medium|basic)\z/;
        my $inboxdir = $ibx->{inboxdir};
        my $version = $ibx->version;
        my $indexlevel = 'full';
@@ -45,7 +46,7 @@ sub new {
                $altid = [ map { PublicInbox::AltId->new($ibx, $_); } @$altid ];
        }
        if ($ibx->{indexlevel}) {
-               if ($ibx->{indexlevel} =~ $levels) {
+               if ($ibx->{indexlevel} =~ $INDEXLEVELS) {
                        $indexlevel = $ibx->{indexlevel};
                } else {
                        die("Invalid indexlevel $ibx->{indexlevel}\n");
@@ -104,6 +105,7 @@ sub load_xapian_writable () {
        }
        eval 'require '.$X->{WritableDatabase} or die;
        *sortable_serialise = $xap.'::sortable_serialise';
+       *sortable_unserialise = $xap.'::sortable_unserialise';
        $DB_CREATE_OR_OPEN = eval($xap.'::DB_CREATE_OR_OPEN()');
        $DB_OPEN = eval($xap.'::DB_OPEN()');
        my $ver = (eval($xap.'::major_version()') << 16) |
@@ -323,6 +325,16 @@ sub index_xapian { # msg_iter callback
        }
 }
 
+sub index_list_id ($$$) {
+       my ($self, $doc, $hdr) = @_;
+       for my $l ($hdr->header_raw('List-Id')) {
+               $l =~ /<([^>]+)>/ or next;
+               my $lid = lc $1;
+               $doc->add_boolean_term('G' . $lid);
+               index_text($self, $lid, 1, 'XL'); # probabilistic
+       }
+}
+
 sub index_ids ($$$$) {
        my ($self, $doc, $hdr, $mids) = @_;
        for my $mid (@$mids) {
@@ -336,12 +348,7 @@ sub index_ids ($$$$) {
                }
        }
        $doc->add_boolean_term('Q' . $_) for @$mids;
-       for my $l ($hdr->header_raw('List-Id')) {
-               $l =~ /<([^>]+)>/ or next;
-               my $lid = $1;
-               $doc->add_boolean_term('G' . $lid);
-               index_text($self, $lid, 1, 'XL'); # probabilistic
-       }
+       index_list_id($self, $doc, $hdr);
 }
 
 sub add_xapian ($$$$) {
@@ -361,6 +368,11 @@ sub add_xapian ($$$$) {
        $tg->set_document($doc);
        index_headers($self, $smsg);
 
+       if (defined(my $eidx_key = $smsg->{eidx_key})) {
+               $doc->add_boolean_term('O'.$eidx_key);
+               $doc->add_boolean_term('P'.
+                               "$eidx_key:$smsg->{num}:$smsg->{blob}");
+       }
        msg_iter($eml, \&index_xapian, [ $self, $doc ]);
        index_ids($self, $doc, $eml, $mids);
 
@@ -434,20 +446,82 @@ sub add_message {
        $smsg->{num};
 }
 
+sub _get_doc ($$$) {
+       my ($self, $docid, $oid) = @_;
+       my $doc = eval { $self->{xdb}->get_document($docid) };
+       $doc // do {
+               warn "E: $@\n" if $@;
+               warn "E: #$docid $oid missing in Xapian\n";
+               undef;
+       }
+}
+
+sub add_xref3 {
+       my ($self, $docid, $xnum, $oid, $eidx_key, $eml) = @_;
+       begin_txn_lazy($self);
+       my $doc = _get_doc($self, $docid, $oid) or return;
+       term_generator($self)->set_document($doc);
+       $doc->add_boolean_term('O'.$eidx_key);
+       $doc->add_boolean_term('P'."$eidx_key:$xnum:$oid");
+       index_list_id($self, $doc, $eml);
+       $self->{xdb}->replace_document($docid, $doc);
+}
+
+sub remove_xref3 {
+       my ($self, $docid, $oid, $eidx_key, $eml) = @_;
+       begin_txn_lazy($self);
+       my $doc = _get_doc($self, $docid, $oid) or return;
+       my $xref3 = PublicInbox::Smsg::xref3(undef, $doc);
+       for (grep(/\A\Q$eidx_key\E:[0-9]+:\Q$oid\E\z/, @$xref3)) {
+               $doc->remove_term('P' . $_);
+       }
+       for my $l ($eml->header_raw('List-Id')) {
+               $l =~ /<([^>]+)>/ or next;
+               my $lid = lc $1;
+               $doc->remove_term('G' . $lid);
+
+               # nb: we don't remove the XL probabilistic terms
+               # since terms may overlap if cross-posted.
+               #
+               # IOW, a message which has both <foo.example.com>
+               # and <bar.example.com> would have overlapping
+               # "XLexample" and "XLcom" as terms and which we
+               # wouldn't know if they're safe to remove if we just
+               # unindex <foo.example.com> while preserving
+               # <bar.example.com>.
+               #
+               # In any case, this entire sub is will likely never
+               # be needed and users using the "l:" prefix are probably
+               # rarer.
+       }
+       $self->{xdb}->replace_document($docid, $doc);
+}
+
+sub get_val ($$) {
+       my ($doc, $col) = @_;
+       sortable_unserialise($doc->get_value($col));
+}
+
+sub smsg_from_doc ($) {
+       my ($doc) = @_;
+       my $data = $doc->get_data or return;
+       my $smsg = bless {}, 'PublicInbox::Smsg';
+       $smsg->{ts} = get_val($doc, PublicInbox::Search::TS());
+       my $dt = get_val($doc, PublicInbox::Search::DT());
+       my ($yyyy, $mon, $dd, $hh, $mm, $ss) = unpack('A4A2A2A2A2A2', $dt);
+       $smsg->{ds} = timegm($ss, $mm, $hh, $dd, $mon - 1, $yyyy);
+       $smsg->load_from_data($data);
+       $smsg;
+}
+
 sub xdb_remove {
        my ($self, $oid, @removed) = @_;
        my $xdb = $self->{xdb} or return;
        for my $num (@removed) {
-               my $doc = eval { $xdb->get_document($num) };
-               unless ($doc) {
-                       warn "E: $@\n" if $@;
-                       warn "E: #$num $oid missing in Xapian\n";
-                       next;
-               }
-               my $smsg = bless {}, 'PublicInbox::Smsg';
-               $smsg->load_expand($doc);
-               my $blob = $smsg->{blob} // '(unset)';
-               if ($blob eq $oid) {
+               my $doc = _get_doc($self, $num, $oid) or next;
+               my $smsg = smsg_from_doc($doc);
+               my $blob = $smsg->{blob}; # may be undef if --skip-docdata
+               if (!defined($blob) || $blob eq $oid) {
                        $xdb->delete_document($num);
                } else {
                        warn "E: #$num $oid != $blob in Xapian\n";
@@ -569,8 +643,7 @@ sub check_size { # check_async cb for -index --max-size=...
 
 sub v1_checkpoint ($$;$) {
        my ($self, $sync, $stk) = @_;
-       $self->{ibx}->git->check_async_wait;
-       $self->{ibx}->git->cat_async_wait;
+       $self->{ibx}->git->async_wait_all;
 
        # latest_cmt may be undef
        my $newest = $stk ? $stk->{latest_cmt} : undef;