]> Sergey Matveev's repositories - public-inbox.git/blobdiff - lib/PublicInbox/Admin.pm
Merge branch 'v2-idx-progress'
[public-inbox.git] / lib / PublicInbox / Admin.pm
index 278482090a8bbc7d09d26a61b6b576f2caed0978..07d8b572c3d3075e9f7ef94620ca19d3096bc2a7 100644 (file)
@@ -135,4 +135,54 @@ invalid indexlevel=$indexlevel (must be `basic', `medium', or `full')
        die missing_mod_msg($err) ." required for indexlevel=$indexlevel\n";
 }
 
+sub index_inbox {
+       my ($ibx, $opt) = @_;
+       my $jobs = delete $opt->{jobs} if $opt;
+       if (ref($ibx) && ($ibx->{version} || 1) == 2) {
+               eval { require PublicInbox::V2Writable };
+               die "v2 requirements not met: $@\n" if $@;
+               my $v2w = eval { $ibx->importer(0) } || eval {
+                       PublicInbox::V2Writable->new($ibx, {nproc=>$jobs});
+               };
+               if (defined $jobs) {
+                       if ($jobs == 0) {
+                               $v2w->{parallel} = 0;
+                       } else {
+                               my $n = $v2w->{partitions};
+                               if ($jobs != ($n + 1)) {
+                                       warn
+"Unable to respect --jobs=$jobs, inbox was created with $n partitions\n";
+                               }
+                       }
+               }
+               my $warn_cb = $SIG{__WARN__} || sub { print STDERR @_ };
+               local $SIG{__WARN__} = sub {
+                       $warn_cb->($v2w->{current_info}, ': ', @_);
+               };
+               $v2w->index_sync($opt);
+       } else {
+               require PublicInbox::SearchIdx;
+               my $s = PublicInbox::SearchIdx->new($ibx, 1);
+               $s->index_sync($opt);
+       }
+}
+
+sub progress_prepare ($) {
+       my ($opt) = @_;
+
+       # public-inbox-index defaults to quiet, -xcpdb and -compact do not
+       if (defined($opt->{quiet}) && $opt->{quiet} < 0) {
+               $opt->{quiet} = !$opt->{verbose};
+       }
+       if ($opt->{quiet}) {
+               open my $null, '>', '/dev/null' or
+                       die "failed to open /dev/null: $!\n";
+               $opt->{1} = fileno($null); # suitable for spawn() redirect
+               $opt->{-dev_null} = $null;
+       } else {
+               $opt->{verbose} ||= 1;
+               $opt->{-progress} = sub { print STDERR @_ };
+       }
+}
+
 1;