search: add mset_to_artnums method

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

diff --git a/lib/PublicInbox/SearchView.pm b/lib/PublicInbox/SearchView.pm
index 61534c2509a4888efc3858c8919e4c51209327fb..aa8fa037843741337850645dc25c5488fc730f01 100644 (file)
--- a/lib/PublicInbox/SearchView.pm
+++ b/lib/PublicInbox/SearchView.pm
@@ -91,22 +91,21 @@ sub mset_summary {
        my $pfx = ' ' x $pad;
        my $res = \($ctx->{-html_tip});
        my $ibx = $ctx->{-inbox};
-       my $over = $ibx->over;
-       my $nshard = $ibx->search->{nshard} // 1;
        my $obfs_ibx = $ibx->{obfuscate} ? $ibx : undef;
+       my @nums = @{$ibx->search->mset_to_artnums($mset)};
+       my %num2msg = map { $_->{num} => $_ } @{$ibx->over->get_all(@nums)};
+
        foreach my $m ($mset->items) {
                my $rank = sprintf("%${pad}d", $m->get_rank + 1);
                my $pct = get_pct($m);
-               my $num = mdocid($nshard, $m);
-               my $smsg = $over->get_art($num, 1);
-               unless ($smsg) {
+               my $num = shift @nums;
+               my $smsg = delete($num2msg{$num}) or do {
                        eval {
                                $m = "$m $num expired\n";
                                $ctx->{env}->{'psgi.errors'}->print($m);
                        };
                        next;
-               }
-               PublicInbox::Smsg::psgi_cull($smsg);
+               };
                my $s = ascii_html($smsg->{subject});
                my $f = ascii_html($smsg->{from_name});
                if ($obfs_ibx) {
@@ -255,20 +254,13 @@ sub get_pct ($) {
        $n > 99 ? 99 : $n;
 }
 
-sub load_msgs {
-       my ($mset) = @_;
-       [ map {
-               my $mi = $_;
-               my $smsg = PublicInbox::Smsg::from_mitem($mi);
-               $smsg->{pct} = get_pct($mi);
-               $smsg;
-       } ($mset->items) ]
-}
-
 sub mset_thread {
        my ($ctx, $mset, $q) = @_;
        my $ibx = $ctx->{-inbox};
-       my $msgs = $ibx->search->retry_reopen(\&load_msgs, $mset);
+       my $nshard = $ibx->search->{nshard} // 1;
+       my %pct = map { mdocid($nshard, $_) => get_pct($_) } $mset->items;
+       my $msgs = $ibx->over->get_all(keys %pct);
+       $_->{pct} = $pct{$_->{num}} for @$msgs;
        my $r = $q->{r};
        my $rootset = PublicInbox::SearchThread::thread($msgs,
                $r ? \&sort_relevance : \&PublicInbox::View::sort_ds,
@@ -323,7 +315,7 @@ sub ctx_prepare {
 
 sub adump {
        my ($cb, $mset, $q, $ctx) = @_;
-       $ctx->{items} = [ $mset->items ];
+       $ctx->{ids} = $ctx->{-inbox}->search->mset_to_artnums($mset);
        $ctx->{search_query} = $q; # used by WwwAtomStream::atom_header
        PublicInbox::WwwAtomStream->response($ctx, 200, \&adump_i);
 }
@@ -331,11 +323,8 @@ sub adump {
 # callback for PublicInbox::WwwAtomStream::getline
 sub adump_i {
        my ($ctx) = @_;
-       while (my $mi = shift @{$ctx->{items}}) {
-               my $srch = $ctx->{-inbox}->search(undef, $ctx) or return;
-               my $smsg = eval {
-                       PublicInbox::Smsg::from_mitem($mi, $srch);
-               } or next;
+       while (my $num = shift @{$ctx->{ids}}) {
+               my $smsg = eval { $ctx->{-inbox}->over->get_art($num) } or next;
                return $smsg;
        }
 }