search: add mset_to_artnums method

[public-inbox.git] / lib / PublicInbox / Mbox.pm

diff --git a/lib/PublicInbox/Mbox.pm b/lib/PublicInbox/Mbox.pm
index fc83a893c60ab384f21061d3b2dff12edb6f365c..873ff7be8988d3b3ab8274efe133a48e97dcebba 100644 (file)
--- a/lib/PublicInbox/Mbox.pm
+++ b/lib/PublicInbox/Mbox.pm
@@ -205,20 +205,18 @@ sub mbox_all_ids {
 
 sub results_cb {
        my ($ctx) = @_;
-       my $mset = $ctx->{mset};
-       my $srch = $ctx->{srch};
+       my $over = $ctx->{-inbox}->over or return;
        while (1) {
-               while (my $mi = (($mset->items)[$ctx->{iter}++])) {
-                       my $smsg = PublicInbox::Smsg::from_mitem($mi,
-                                                               $srch) or next;
+               while (defined(my $num = shift(@{$ctx->{ids}}))) {
+                       my $smsg = $over->get_art($num) or next;
                        return $smsg;
                }
                # refill result set
-               $mset = $ctx->{mset} = $srch->query($ctx->{query},
-                                                       $ctx->{qopts});
+               my $srch = $ctx->{-inbox}->search(undef, $ctx) or return;
+               my $mset = $srch->query($ctx->{query}, $ctx->{qopts});
                my $size = $mset->size or return;
                $ctx->{qopts}->{offset} += $size;
-               $ctx->{iter} = 0;
+               $ctx->{ids} = $srch->mset_to_artnums($mset);
        }
 }
 
@@ -226,15 +224,15 @@ sub mbox_all {
        my ($ctx, $query) = @_;
 
        return mbox_all_ids($ctx) if $query eq '';
-       my $qopts = $ctx->{qopts} = { mset => 2 };
-       my $srch = $ctx->{srch} = $ctx->{-inbox}->search or
-               return PublicInbox::WWW::need($ctx, 'Search');;
-       my $mset = $ctx->{mset} = $srch->query($query, $qopts);
+       my $qopts = $ctx->{qopts} = { mset => 2 }; # order by docid
+       my $srch = $ctx->{-inbox}->search or
+               return PublicInbox::WWW::need($ctx, 'Search');
+       my $mset = $srch->query($query, $qopts);
        $qopts->{offset} = $mset->size or
                        return [404, [qw(Content-Type text/plain)],
                                ["No results found\n"]];
-       $ctx->{iter} = 0;
        $ctx->{query} = $query;
+       $ctx->{ids} = $srch->mset_to_artnums($mset);
        require PublicInbox::MboxGz;
        PublicInbox::MboxGz::mbox_gz($ctx, \&results_cb, 'results-'.$query);
 }