]> Sergey Matveev's repositories - public-inbox.git/blobdiff - lib/PublicInbox/LeiMirror.pm
lei_mirror: properly pack-refs in non-forkgroup repos
[public-inbox.git] / lib / PublicInbox / LeiMirror.pm
index 829740bc456ae7403ed70f857d71a0852a46eba0..4464b6b109b9bc1884c6a3d1a8cd73d29980927f 100644 (file)
@@ -21,7 +21,6 @@ use PublicInbox::OnDestroy;
 use Digest::SHA qw(sha256_hex);
 
 our $LIVE; # pid => callback
-my $update_ref_stdin = $ENV{GIT_CAN_UPDATE_REF_STDIN} // 1;
 
 sub _wq_done_wait { # dwaitpid callback (via wq_eof)
        my ($arg, $pid) = @_;
@@ -279,18 +278,9 @@ sub fetch_args ($$) {
        @cmd;
 }
 
-sub fgrp_update_old ($) { # for git <1.8.5
-       my ($fgrp) = @_;
-       my $cmd = [ @{$fgrp->{-torsocks}}, 'git', "--git-dir=$fgrp->{cur_dst}",
-               fetch_args($fgrp->{lei}, my $opt = {}) ];
-       $fgrp->{lei}->qerr("# @$cmd");
-       do_reap($fgrp);
-       $LIVE->{spawn($cmd, undef, $opt)} = [ \&reap_cmd, $fgrp, $cmd ];
-}
-
 sub upr { # feed `git update-ref --stdin -z' verbosely
-       my ($fgrp, $w, $op, $ref, $oid) = @_;
-       $fgrp->{lei}->qerr("# $op $ref $oid");
+       my ($lei, $w, $op, $ref, $oid) = @_;
+       $lei->qerr("# $op $ref $oid") if $lei->{opt}->{verbose};
        print $w "$op $ref\0$oid\0" or die "print(w): $!";
 }
 
@@ -307,31 +297,31 @@ sub fgrp_update {
        pipe(my ($r, $w)) or die "pipe: $!";
        my $cmd = [ 'git', "--git-dir=$fgrp->{cur_dst}",
                qw(update-ref --stdin -z) ];
-       $fgrp->{lei}->qerr("# @$cmd");
-       my $opt = { 0 => $r, 1 => $fgrp->{lei}->{1}, 2 => $fgrp->{lei}->{2} };
+       my $lei = $fgrp->{lei};
+       $lei->qerr("# @$cmd");
+       my $opt = { 0 => $r, 1 => $lei->{1}, 2 => $lei->{2} };
        my $pid = spawn($cmd, undef, $opt);
        close $r or die "close(r): $!";
        for my $ref (keys %dst) {
                my $new = delete $src{$ref};
                my $old = $dst{$ref};
                if (defined $new) {
-                       upr($fgrp, $w, 'update', $ref, $new) if $new ne $old;
+                       upr($lei, $w, 'update', $ref, $new) if $new ne $old;
                } else {
-                       upr($fgrp, $w, 'delete', $ref, $old);
+                       upr($lei, $w, 'delete', $ref, $old);
                }
        }
        while (my ($ref, $oid) = each %src) {
-               upr($fgrp, $w, 'create', $ref, $oid);
-       }
-       if (close($w)) { # git >= 1.8.5
-               $LIVE->{$pid} = [ \&reap_cmd, $fgrp, $cmd ];
-               do_reap($fgrp);
-       } else { # git <1.8.5 w/o update-ref --stdin
-               warn "E: close(update-ref --stdin): $!\n";
-               $update_ref_stdin = 0;
-               waitpid($pid, 0) // die "waitpid(update-ref --stdin): $!";
-               fgrp_update_old($fgrp);
+               upr($lei, $w, 'create', $ref, $oid);
        }
+       close($w) or warn "E: close(update-ref --stdin): $! (need git 1.8.5+)\n";
+       my $pack = PublicInbox::OnDestroy->new($$, \&pack_dst, $fgrp);
+       $LIVE->{$pid} = [ \&reap_cmd, $fgrp, $cmd, $pack ];
+}
+
+sub pack_dst { # packs lightweight satellite repos
+       my ($fgrp) = @_;
+       pack_refs($fgrp, $fgrp->{cur_dst});
 }
 
 sub pack_refs {
@@ -339,51 +329,101 @@ sub pack_refs {
        do_reap($self);
        my $cmd = [ 'git', "--git-dir=$git_dir", qw(pack-refs --all --prune) ];
        $self->{lei}->qerr("# @$cmd");
+       return if $self->{dry_run};
        my $opt = { 1 => $self->{lei}->{1}, 2 => $self->{lei}->{2} };
        $LIVE->{spawn($cmd, undef, $opt)} = [ \&reap_cmd, $self, $cmd ];
 }
 
-sub fgrp_fetched {
-       my ($fgrp) = @_;
-       return if $fgrp->{dry_run} || !$LIVE;
-       my $rn = $fgrp->{-remote};
-       my %opt = map { $_ => $fgrp->{lei}->{$_} } (0..2);
-       pack_refs($fgrp, $fgrp->{-osdir}); # objstore refs always packed
-
-       $update_ref_stdin or return fgrp_update_old($fgrp);
-
-       my $update_ref = PublicInbox::OnDestroy->new($$, \&fgrp_update, $fgrp);
-
-       my $src = [ 'git', "--git-dir=$fgrp->{-osdir}", 'for-each-ref',
-               "--format=refs/%(refname:lstrip=3)%00%(objectname)",
-               "refs/remotes/$rn/" ];
-       do_reap($fgrp);
-       open($fgrp->{srcfh}, '+>', undef) or die "open(src): $!";
-       $fgrp->{lei}->qerr("# @$src >SRC");
-       my $pid = spawn($src, undef, { %opt, 1 => $fgrp->{srcfh} });
-       $LIVE->{$pid} = [ \&reap_cmd, $fgrp, $src, $update_ref ];
-
-       my $dst = [ 'git', "--git-dir=$fgrp->{cur_dst}", 'for-each-ref',
-               '--format=%(refname)%00%(objectname)' ];
-       do_reap($fgrp);
-       open($fgrp->{dstfh}, '+>', undef) or die "open(dst): $!";
-       $fgrp->{lei}->qerr("# @$dst >DST");
-       $pid = spawn($dst, undef, { %opt, 1 => $fgrp->{dstfh} });
-       $LIVE->{$pid} = [ \&reap_cmd, $fgrp, $dst, $update_ref ];
-}
-
-sub fgrp_fetch {
-       my ($fgrp, $fini) = @_;
-       my $cmd = [ @{$fgrp->{-torsocks}}, 'git', "--git-dir=$fgrp->{-osdir}",
-                       fetch_args($fgrp->{lei}, my $opt = {}), '--no-tags',
-                       $fgrp->{-remote} ];
-       $fgrp->{-fini} = $fini;
-       do_reap($fgrp);
-       $fgrp->{lei}->qerr("# @$cmd");
-       return if $fgrp->{dry_run};
-       my $fgrp_fini = PublicInbox::OnDestroy->new($$, \&fgrp_fetched, $fgrp);
-       my $pid = spawn($cmd, undef, $opt);
-       $LIVE->{$pid} = [ \&reap_cmd, $fgrp, $cmd, $fgrp_fini ];
+sub fgrpv_done {
+       my ($fgrpv) = @_;
+       return if !$LIVE;
+       my $pid;
+       my $first = $fgrpv->[0] // die 'BUG: no fgrpv->[0]';
+       pack_refs($first, $first->{-osdir}); # objstore refs always packed
+       for my $fgrp (@$fgrpv) {
+               my $rn = $fgrp->{-remote};
+               my %opt = map { $_ => $fgrp->{lei}->{$_} } (0..2);
+
+               my $update_ref = $fgrp->{dry_run} ? undef :
+                       PublicInbox::OnDestroy->new($$, \&fgrp_update, $fgrp);
+
+               my $src = [ 'git', "--git-dir=$fgrp->{-osdir}", 'for-each-ref',
+                       "--format=refs/%(refname:lstrip=3)%00%(objectname)",
+                       "refs/remotes/$rn/" ];
+               do_reap($fgrp);
+               $fgrp->{lei}->qerr("# @$src >SRC");
+               if ($update_ref) {
+                       open(my $fh, '+>', undef) or die "open(src): $!";
+                       $pid = spawn($src, undef, { %opt, 1 => $fh });
+                       $fgrp->{srcfh} = $fh;
+                       $LIVE->{$pid} = [ \&reap_cmd, $fgrp, $src, $update_ref ]
+               }
+               my $dst = [ 'git', "--git-dir=$fgrp->{cur_dst}", 'for-each-ref',
+                       '--format=%(refname)%00%(objectname)' ];
+               do_reap($fgrp);
+               $fgrp->{lei}->qerr("# @$dst >DST");
+               if ($update_ref) {
+                       open(my $fh, '+>', undef) or die "open(dst): $!";
+                       $pid = spawn($dst, undef, { %opt, 1 => $fh });
+                       $fgrp->{dstfh} = $fh;
+                       $LIVE->{$pid} = [ \&reap_cmd, $fgrp, $dst, $update_ref ]
+               }
+       }
+}
+
+sub fgrp_fetch_all {
+       my ($self) = @_;
+       my $todo = delete $self->{fgrp_todo} or return;
+       keys(%$todo) or return;
+
+       # Rely on the fgrptmp remote groups in the config file rather
+       # than listing all remotes since the remote name list may exceed
+       # system argv limits:
+       my $grp = 'fgrptmp';
+
+       my @git = (@{$self->{-torsocks}}, 'git');
+       my $j = $self->{lei}->{opt}->{jobs};
+       my $opt = {};
+       my @fetch = do {
+               local $self->{lei}->{opt}->{jobs} = 1;
+               (fetch_args($self->{lei}, $opt),
+                       qw(--no-tags --multiple));
+       };
+       push(@fetch, "-j$j") if $j;
+       my $pid;
+       while (my ($osdir, $fgrpv) = each %$todo) {
+               my $f = "$osdir/config";
+
+               # clobber group from previous run atomically
+               my $cmd = ['git', "--git-dir=$osdir", qw(config -f),
+                               $f, '--unset-all', "remotes.$grp"];
+               $self->{lei}->qerr("# @$cmd");
+               if (!$self->{dry_run}) {
+                       $pid = spawn($cmd);
+                       waitpid($pid, 0) // die "waitpid: $!";
+                       die "E: @$cmd: \$?=$?" if ($? && ($? >> 8) != 5);
+
+                       # update the config atomically via O_APPEND while
+                       # respecting git-config locking
+                       sysopen(my $lk, "$f.lock", O_CREAT|O_EXCL|O_WRONLY)
+                               or die "open($f.lock): $!";
+                       open my $fh, '>>', $f or die "open(>>$f): $!";
+                       $fh->autoflush(1);
+                       my $buf = join('', "[remotes]\n",
+                               map { "\t$grp = $_->{-remote}\n" } @$fgrpv);
+                       print $fh $buf or die "print($f): $!";
+                       close $fh or die "close($f): $!";
+                       unlink("$f.lock") or die "unlink($f.lock): $!";
+               }
+
+               $cmd = [ @git, "--git-dir=$osdir", @fetch, $grp ];
+               do_reap($self);
+               $self->{lei}->qerr("# @$cmd");
+               my $end = PublicInbox::OnDestroy->new($$, \&fgrpv_done, $fgrpv);
+               return if $self->{dry_run};
+               $pid = spawn($cmd, undef, $opt);
+               $LIVE->{$pid} = [ \&reap_cmd, $self, $cmd, $end ];
+       }
 }
 
 # keep this idempotent for future use by public-inbox-fetch
@@ -395,7 +435,7 @@ sub forkgroup_prep {
        my $dir = "$os/$fg.git";
        my @cmd = ('git', "--git-dir=$dir", 'config');
        my $opt = +{ map { $_ => $self->{lei}->{$_} } (0..2) };
-       if (!-d $dir) {
+       if (!-d $dir && !$self->{dry_run}) {
                PublicInbox::Import::init_bare($dir);
                for ('repack.useDeltaIslands=true',
                                'pack.island=refs/remotes/([^/]+)/') {
@@ -403,17 +443,17 @@ sub forkgroup_prep {
                }
        }
        my $key = $self->{-key} // die 'BUG: no -key';
-       my ($bn) = ($key =~ m{/([a-z0-9_,;=!\+\{\}\|][^/]*)(?:\.git)?\z}i);
-       my $rn = "$bn-".substr(sha256_hex($key), 0, 16);
-       # --no-tags is required to avoid conflicts
-       for ("url=$uri", "fetch=+refs/*:refs/remotes/$rn/*",
-                       'tagopt=--no-tags') {
-               my @kv = split(/=/, $_, 2);
-               $kv[0] = "remote.$rn.$kv[0]";
-               run_die([@cmd, @kv], undef, $opt);
-       }
-       $self->{-do_pack_refs} = 1; # likely coderepo
-       if (!-d $self->{cur_dst}) {
+       my $rn = substr(sha256_hex($key), 0, 16);
+       unless ($self->{dry_run}) {
+               # --no-tags is required to avoid conflicts
+               for ("url=$uri", "fetch=+refs/*:refs/remotes/$rn/*",
+                               'tagopt=--no-tags') {
+                       my @kv = split(/=/, $_, 2);
+                       $kv[0] = "remote.$rn.$kv[0]";
+                       run_die([@cmd, @kv], undef, $opt);
+               }
+       }
+       if (!-d $self->{cur_dst} && !$self->{dry_run}) {
                my $alt = File::Spec->rel2abs("$dir/objects");
                PublicInbox::Import::init_bare($self->{cur_dst});
                my $o = "$self->{cur_dst}/objects";
@@ -426,11 +466,18 @@ sub forkgroup_prep {
                        say $fh $l or die "say($f): $!";
                }
                close $fh or die "close($f): $!";
-               @cmd = ('git', "--git-dir=$self->{cur_dst}",
-                       qw(remote add --mirror=fetch origin), "$uri");
-               my $pid = spawn(\@cmd, undef, $opt);
-               waitpid($pid, 0) // die "waitpid(@cmd): $!";
-               die "E: @cmd: \$?=$?" if ($? && ($? >> 8) != 3);
+               $f = "$self->{cur_dst}/config";
+               open $fh, '+>>', $f or die "open:($f): $!";
+               print $fh <<EOM or die "print($f): $!";
+; rely on the "$rn" remote in the
+; $fg fork group for fetches
+; only uncomment the following iff you detach from fork groups
+; [remote "origin"]
+;      url = $uri
+;      fetch = +refs/*:refs/*
+;      mirror = true
+EOM
+               close $fh or die "close($f): $!";
        }
        bless { %$self, -osdir => $dir, -remote => $rn }, __PACKAGE__;
 }
@@ -445,18 +492,21 @@ sub clone_v1 {
        $self->{-torsocks} //= $curl->torsocks($lei, $uri) or return;
        my $dst = $self->{cur_dst} // $self->{dst};
        my $fini = PublicInbox::OnDestroy->new($$, \&v1_done, $self);
-       my $cmd = [ @{$self->{-torsocks}}, clone_cmd($lei, my $opt = {}),
-               "$uri", $dst ];
-       my $fgrp = forkgroup_prep($self, $uri);
-       if (!defined($fgrp) && defined($self->{-ent})) {
-               if (defined(my $ref = $self->{-ent}->{reference})) {
-                       -e "$self->{dst}$ref" and
-                               push @$cmd, '--reference', "$self->{dst}$ref";
+       if (my $fgrp = forkgroup_prep($self, $uri)) {
+               $fgrp->{-fini} = $fini;
+               push @{$self->{fgrp_todo}->{$fgrp->{-osdir}}}, $fgrp;
+       } else { # normal fetch
+               my $cmd = [ @{$self->{-torsocks}},
+                               clone_cmd($lei, my $opt = {}), "$uri", $dst ];
+               if (defined($self->{-ent})) {
+                       if (defined(my $ref = $self->{-ent}->{reference})) {
+                               -e "$self->{dst}$ref" and
+                                       push @$cmd, '--reference',
+                                               "$self->{dst}$ref";
+                       }
                }
-       }
-       $fgrp ? fgrp_fetch($fgrp, $fini) :
                start_clone($self, $cmd, $opt, $fini);
-
+       }
        if (!$self->{-is_epoch} && $lei->{opt}->{'inbox-config'} =~
                                /\A(?:always|v1)\z/s) {
                _get_txt_start($self, '_/text/config/raw', $fini);
@@ -581,7 +631,6 @@ sub v1_done { # called via OnDestroy
                        ft_rename($ft, $fn, 0666, $fh);
                }
        }
-       pack_refs($self, $dst) if delete $self->{-do_pack_refs};
        eval { set_description($self) };
        warn $@ if $@;
        return if ($self->{-is_epoch} ||
@@ -775,6 +824,7 @@ EOM
                        last; # restart %$todo iteration
                }
        }
+       fgrp_fetch_all($self);
        do_reap($self, 1);
 }
 
@@ -798,6 +848,7 @@ sub try_manifest {
        my $uri = URI->new($self->{src});
        my $lei = $self->{lei};
        my $curl = $self->{curl} //= PublicInbox::LeiCurl->new($lei) or return;
+       $self->{-torsocks} //= $curl->torsocks($lei, $uri) or return;
        my $path = $uri->path;
        chop($path) eq '/' or die "BUG: $uri not canonicalized";
        $uri->path($path . '/manifest.js.gz');
@@ -819,6 +870,7 @@ sub try_manifest {
        return $lei->child_error(1, $multi) if !ref($multi);
        my $v2 = delete $multi->{v2};
        local $self->{todo} = {};
+       local $self->{fgrp_todo} = {}; # { objstore_dir => [fgrp, ...] }
        if ($v2) {
                for my $name (sort keys %$v2) {
                        my $epochs = delete $v2->{$name};
@@ -887,6 +939,7 @@ sub start_clone_url {
 sub do_mirror { # via wq_io_do or public-inbox-clone
        my ($self) = @_;
        my $lei = $self->{lei};
+       $self->{dry_run} = 1 if $lei->{opt}->{'dry-run'};
        umask($lei->{client_umask}) if defined $lei->{client_umask};
        eval {
                my $ic = $lei->{opt}->{'inbox-config'} //= 'always';