#!perl -w
# Copyright (C) 2019-2021 all contributors
# License: AGPL-3.0+
use strict;
use v5.10.1;
use PublicInbox::TestCommon;
require_mods(qw(DBD::SQLite Search::Xapian));
require_git('2.6');
use PublicInbox::Eml;
require PublicInbox::Search;
my ($tmpdir, $for_destroy) = tmpdir();
my $nproc = 8;
my $ndoc = 13;
my $ibx = create_inbox 'test', version => 2, indexlevel => 'medium',
tmpdir => "$tmpdir/testbox", nproc => $nproc, sub {
my ($im, $ibx) = @_;
my $eml = PublicInbox::Eml->new(<<'EOF');
From: a@example.com
To: test@example.com
Subject: this is a subject
Date: Fri, 02 Oct 1993 00:00:00 +0000
EOF
for my $i (1..$ndoc) {
$eml->header_set('Message-ID', "");
ok($im->add($eml), "message $i added");
}
open my $fh, '>', "$ibx->{inboxdir}/empty" or BAIL_OUT "open $!";
};
my $env = { PI_CONFIG => "$ibx->{inboxdir}/empty" };
my @shards = grep(m!/\d+\z!, glob("$ibx->{inboxdir}/xap*/*"));
is(scalar(@shards), $nproc - 1, 'got expected shards');
my $orig = $ibx->over->query_xover(1, $ndoc);
my %nums = map {; "$_->{num}" => 1 } @$orig;
my @xcpdb = qw(-xcpdb -q);
my $XapianDatabase = do {
no warnings 'once';
$PublicInbox::Search::X{Database};
};
# ensure we can go up or down in shards, or stay the same:
for my $R (qw(2 4 1 3 3)) {
delete $ibx->{search}; # release old handles
my $cmd = [@xcpdb, "-R$R", $ibx->{inboxdir}];
push @$cmd, '--compact' if $R == 1 && have_xapian_compact;
ok(run_script($cmd, $env), "xcpdb -R$R");
my @new_shards = grep(m!/\d+\z!, glob("$ibx->{inboxdir}/xap*/*"));
is(scalar(@new_shards), $R, 'resharded to two shards');
my $mset = $ibx->search->mset('s:this');
my $msgs = $ibx->search->mset_to_smsg($ibx, $mset);
is(scalar(@$msgs), $ndoc, 'got expected docs after resharding');
my %by_mid = map {; "$_->{mid}" => $_ } @$msgs;
ok($by_mid{"m$_\@example.com"}, "$_ exists") for (1..$ndoc);
delete $ibx->{search}; # release old handles
# ensure docids in Xapian match NNTP article numbers
my $tot = 0;
my %tmp = %nums;
foreach my $d (@new_shards) {
my $xdb = $XapianDatabase->new($d);
$tot += $xdb->get_doccount;
my $it = $xdb->postlist_begin('');
my $end = $xdb->postlist_end('');
for (; $it != $end; $it++) {
my $docid = $it->get_docid;
if ($xdb->get_document($docid)) {
ok(delete($tmp{$docid}), "saw #$docid");
}
}
}
is(scalar keys %tmp, 0, 'all docids seen');
}
done_testing;