]> Sergey Matveev's repositories - public-inbox.git/blobdiff - lib/PublicInbox/Search.pm
inbox: add ->version method
[public-inbox.git] / lib / PublicInbox / Search.pm
index 88c8dc70e17c3bedf3883f07555894b6c54eb7c3..5e820594c3777b322b2a36997cb08d50c95e9403 100644 (file)
@@ -13,24 +13,44 @@ use constant YYYYMMDD => 1; # Date: header for searching in the WWW UI
 use constant DT => 2; # Date: YYYYMMDDHHMMSS
 
 use PublicInbox::SearchMsg;
-use PublicInbox::MIME;
-use PublicInbox::MID qw/id_compress/;
 use PublicInbox::Over;
 my $QP_FLAGS;
+our %X = map { $_ => 0 } qw(BoolWeight Database Enquire
+                       NumberValueRangeProcessor QueryParser Stem);
+our $Xap; # 'Search::Xapian' or 'Xapian'
+my $ENQ_ASCENDING;
+
 sub load_xapian () {
-       $QP_FLAGS ||= eval {
-               require Search::Xapian;
-               Search::Xapian->import(qw(:standard));
+       return 1 if defined $Xap;
+       for my $x (qw(Search::Xapian Xapian)) {
+               eval "require $x";
+               next if $@;
+
+               $x->import(qw(:standard));
+               $Xap = $x;
+               $X{$_} = $Xap.'::'.$_ for (keys %X);
+
+               # ENQ_ASCENDING doesn't seem exported by SWIG Xapian.pm,
+               # so lets hope this part of the ABI is stable because it's
+               # just an integer:
+               $ENQ_ASCENDING = $x eq 'Xapian' ?
+                               1 : Search::Xapian::ENQ_ASCENDING();
 
+               # for SearchMsg:
+               *PublicInbox::SearchMsg::sortable_unserialise =
+                                               $Xap.'::sortable_unserialise';
                # n.b. FLAG_PURE_NOT is expensive not suitable for a public
                # website as it could become a denial-of-service vector
                # FLAG_PHRASE also seems to cause performance problems chert
                # (and probably earlier Xapian DBs).  glass seems fine...
                # TODO: make this an option, maybe?
                # or make indexlevel=medium as default
-               FLAG_PHRASE()|FLAG_BOOLEAN()|FLAG_LOVEHATE()|FLAG_WILDCARD();
-       };
-};
+               $QP_FLAGS = FLAG_PHRASE() | FLAG_BOOLEAN() | FLAG_LOVEHATE() |
+                               FLAG_WILDCARD();
+               return 1;
+       }
+       undef;
+}
 
 # This is English-only, everything else is non-standard and may be confused as
 # a prefix common in patch emails
@@ -148,7 +168,7 @@ sub _xdb ($) {
        if ($self->{version} >= 2) {
                foreach my $shard (<$dir/*>) {
                        -d $shard && $shard =~ m!/[0-9]+\z! or next;
-                       my $sub = Search::Xapian::Database->new($shard);
+                       my $sub = $X{Database}->new($shard);
                        if ($xdb) {
                                $xdb->add_database($sub);
                        } else {
@@ -158,7 +178,7 @@ sub _xdb ($) {
                }
        } else {
                $slow_phrase = -f "$dir/iamchert";
-               $xdb = Search::Xapian::Database->new($dir);
+               $xdb = $X{Database}->new($dir);
        }
        $$qpf |= FLAG_PHRASE() unless $slow_phrase;
        $xdb;
@@ -178,7 +198,7 @@ sub new {
        my $self = bless {
                inboxdir => $ibx->{inboxdir},
                altid => $ibx->{altid},
-               version => $ibx->{version} // 1,
+               version => $ibx->version,
        }, $class;
        my $dir = xdir($self, 1);
        $self->{over_ro} = PublicInbox::Over->new("$dir/over.sqlite3");
@@ -209,20 +229,20 @@ sub query {
 }
 
 sub retry_reopen {
-       my ($self, $cb) = @_;
+       my ($self, $cb, $arg) = @_;
        for my $i (1..10) {
                if (wantarray) {
                        my @ret;
-                       eval { @ret = $cb->() };
+                       eval { @ret = $cb->($arg) };
                        return @ret unless $@;
                } else {
                        my $ret;
-                       eval { $ret = $cb->() };
+                       eval { $ret = $cb->($arg) };
                        return $ret unless $@;
                }
                # Exception: The revision being read has been discarded -
                # you should call Xapian::Database::reopen()
-               if (ref($@) eq 'Search::Xapian::DatabaseModifiedError') {
+               if (ref($@) =~ /\bDatabaseModifiedError\b/) {
                        warn "reopen try #$i on $@\n";
                        reopen($self);
                } else {
@@ -237,19 +257,19 @@ sub retry_reopen {
 
 sub _do_enquire {
        my ($self, $query, $opts) = @_;
-       retry_reopen($self, sub { _enquire_once($self, $query, $opts) });
+       retry_reopen($self, \&_enquire_once, [ $self, $query, $opts ]);
 }
 
-sub _enquire_once {
-       my ($self, $query, $opts) = @_;
+sub _enquire_once { # retry_reopen callback
+       my ($self, $query, $opts) = @{$_[0]};
        my $xdb = xdb($self);
-       my $enquire = Search::Xapian::Enquire->new($xdb);
+       my $enquire = $X{Enquire}->new($xdb);
        $enquire->set_query($query);
        $opts ||= {};
         my $desc = !$opts->{asc};
        if (($opts->{mset} || 0) == 2) {
-               $enquire->set_docid_order(Search::Xapian::ENQ_ASCENDING());
-               $enquire->set_weighting_scheme(Search::Xapian::BoolWeight->new);
+               $enquire->set_docid_order($ENQ_ASCENDING);
+               $enquire->set_weighting_scheme($X{BoolWeight}->new);
        } elsif ($opts->{relevance}) {
                $enquire->set_sort_by_relevance_then_value(TS, $desc);
        } else {
@@ -259,16 +279,14 @@ sub _enquire_once {
        my $limit = $opts->{limit} || 50;
        my $mset = $enquire->get_mset($offset, $limit);
        return $mset if $opts->{mset};
-       my @msgs = map {
-               PublicInbox::SearchMsg->load_doc($_->get_document);
-       } $mset->items;
+       my @msgs = map { PublicInbox::SearchMsg::from_mitem($_) } $mset->items;
        return \@msgs unless wantarray;
 
        ($mset->get_matches_estimated, \@msgs)
 }
 
 # read-write
-sub stemmer { Search::Xapian::Stem->new($LANG) }
+sub stemmer { $X{Stem}->new($LANG) }
 
 # read-only
 sub qp {
@@ -278,16 +296,15 @@ sub qp {
        return $qp if $qp;
        my $xdb = xdb($self);
        # new parser
-       $qp = Search::Xapian::QueryParser->new;
+       $qp = $X{QueryParser}->new;
        $qp->set_default_op(OP_AND());
        $qp->set_database($xdb);
        $qp->set_stemmer($self->stemmer);
        $qp->set_stemming_strategy(STEM_SOME());
        $qp->set_max_wildcard_expansion(100);
-       $qp->add_valuerangeprocessor(
-               Search::Xapian::NumberValueRangeProcessor->new(YYYYMMDD, 'd:'));
-       $qp->add_valuerangeprocessor(
-               Search::Xapian::NumberValueRangeProcessor->new(DT, 'dt:'));
+       my $nvrp = $X{NumberValueRangeProcessor};
+       $qp->add_valuerangeprocessor($nvrp->new(YYYYMMDD, 'd:'));
+       $qp->add_valuerangeprocessor($nvrp->new(DT, 'dt:'));
 
        while (my ($name, $prefix) = each %bool_pfx_external) {
                $qp->add_boolean_prefix($name, $_) foreach split(/ /, $prefix);
@@ -317,11 +334,6 @@ EOF
        $self->{query_parser} = $qp;
 }
 
-sub lookup_article {
-       my ($self, $num) = @_;
-       $self->{over_ro}->get_art($num);
-}
-
 sub help {
        my ($self) = @_;
        $self->qp; # parse altids