init: add --skip-artnum parameter

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

diff --git a/lib/PublicInbox/InboxWritable.pm b/lib/PublicInbox/InboxWritable.pm
index 3558403bca655dec35a9aba29bcf427a3c578fc8..f9e28502001499b72d7f6e7766f4e824920862db 100644 (file)
--- a/lib/PublicInbox/InboxWritable.pm
+++ b/lib/PublicInbox/InboxWritable.pm
@@ -39,10 +39,21 @@ sub assert_usable_dir {
 
 sub init_inbox {
        my ($self, $shards, $skip_epoch, $skip_artnum) = @_;
-       # TODO: honor skip_artnum
        if ($self->version == 1) {
                my $dir = assert_usable_dir($self);
                PublicInbox::Import::init_bare($dir);
+               if (defined($self->{indexlevel}) || defined($skip_artnum)) {
+                       require PublicInbox::SearchIdx;
+                       my $sidx = PublicInbox::SearchIdx->new($self, 1); # just create
+                       $sidx->begin_txn_lazy;
+                       $self->with_umask(sub {
+                               my $mm = PublicInbox::Msgmap->new($dir, 1);
+                               $mm->{dbh}->begin_work;
+                               $mm->skip_artnum($skip_artnum);
+                               $mm->{dbh}->commit;
+                       }) if defined($skip_artnum);
+                       $sidx->commit_txn_lazy;
+               }
        } else {
                my $v2w = importer($self);
                $v2w->init_inbox($shards, $skip_epoch, $skip_artnum);
@@ -57,7 +68,7 @@ sub importer {
                die "v2 not supported: $@\n" if $@;
                my $opt = $self->{-creat_opt};
                my $v2w = PublicInbox::V2Writable->new($self, $opt);
-               $v2w->{parallel} = $parallel;
+               $v2w->{parallel} = $parallel if defined $parallel;
                $v2w;
        } elsif ($v == 1) {
                my @arg = (undef, undef, undef, $self);