* Use the Store API bindings in nix-copy-closure.

This commit is contained in:
Eelco Dolstra 2011-10-11 15:41:13 +00:00
parent d43a148204
commit 6761757428
4 changed files with 36 additions and 39 deletions

View File

@ -5,6 +5,7 @@ perllibdir = $(libdir)/perl5/site_perl/$(perlversion)/$(perlarchname)
PERL_MODULES = lib/Nix/Store.pm lib/Nix/Manifest.pm lib/Nix/GeneratePatches.pm lib/Nix/SSH.pm lib/Nix/Config.pm.in PERL_MODULES = lib/Nix/Store.pm lib/Nix/Manifest.pm lib/Nix/GeneratePatches.pm lib/Nix/SSH.pm lib/Nix/Config.pm.in
all: $(PERL_MODULES:.in=) all: $(PERL_MODULES:.in=)
ln -sfn $(abs_builddir)/.libs/libNixStore.so lib/Store.so
install-exec-local: $(PERL_MODULES:.in=) install-exec-local: $(PERL_MODULES:.in=)
$(INSTALL) -d $(DESTDIR)$(perllibdir)/Nix $(INSTALL) -d $(DESTDIR)$(perllibdir)/Nix

View File

@ -12,7 +12,7 @@ our %EXPORT_TAGS = ( 'all' => [ qw( ) ] );
our @EXPORT_OK = ( @{ $EXPORT_TAGS{'all'} } ); our @EXPORT_OK = ( @{ $EXPORT_TAGS{'all'} } );
our @EXPORT = qw( ); our @EXPORT = qw(isValidPath topoSortPaths computeFSClosure followLinksToStorePath);
our $VERSION = '0.15'; our $VERSION = '0.15';

View File

@ -39,8 +39,7 @@ void init()
doInit(); doInit();
int isValidPath(path) int isValidPath(char * path)
char * path
CODE: CODE:
try { try {
doInit(); doInit();
@ -52,8 +51,7 @@ int isValidPath(path)
RETVAL RETVAL
SV * queryReferences(path) SV * queryReferences(char * path)
char * path
PPCODE: PPCODE:
try { try {
doInit(); doInit();
@ -66,8 +64,7 @@ SV * queryReferences(path)
} }
SV * queryPathHash(path) SV * queryPathHash(char * path)
char * path
PPCODE: PPCODE:
try { try {
doInit(); doInit();
@ -79,8 +76,7 @@ SV * queryPathHash(path)
} }
SV * queryDeriver(path) SV * queryDeriver(char * path)
char * path
PPCODE: PPCODE:
try { try {
doInit(); doInit();
@ -92,8 +88,7 @@ SV * queryDeriver(path)
} }
SV * queryPathInfo(path) SV * queryPathInfo(char * path)
char * path
PPCODE: PPCODE:
try { try {
doInit(); doInit();
@ -127,3 +122,29 @@ SV * computeFSClosure(int flipDirection, int includeOutputs, ...)
} catch (Error & e) { } catch (Error & e) {
croak(e.what()); croak(e.what());
} }
SV * topoSortPaths(...)
PPCODE:
try {
doInit();
PathSet paths;
for (int n = 0; n < items; ++n) paths.insert(SvPV_nolen(ST(n)));
Paths sorted = topoSortPaths(*store, paths);
for (Paths::iterator i = sorted.begin(); i != sorted.end(); ++i)
XPUSHs(sv_2mortal(newSVpv(i->c_str(), 0)));
} catch (Error & e) {
croak(e.what());
}
SV * followLinksToStorePath(char * path)
CODE:
try {
doInit();
RETVAL = newSVpv(followLinksToStorePath(path).c_str(), 0);
} catch (Error & e) {
croak(e.what());
}
OUTPUT:
RETVAL

View File

@ -2,6 +2,7 @@
use Nix::SSH; use Nix::SSH;
use Nix::Config; use Nix::Config;
use Nix::Store;
if (scalar @ARGV < 1) { if (scalar @ARGV < 1) {
@ -57,19 +58,8 @@ openSSHConnection $sshHost or die "$0: unable to start SSH\n";
if ($toMode) { # Copy TO the remote machine. if ($toMode) { # Copy TO the remote machine.
my @allStorePaths;
# Get the closure of this path. # Get the closure of this path.
my $pid = open(READ, "set -f; $Nix::Config::binDir/nix-store --query --requisites @storePaths|") or die; my @allStorePaths = reverse(topoSortPaths(computeFSClosure(0, 0, map { followLinksToStorePath $_ } @storePaths)));
while (<READ>) {
chomp;
die "bad: $_" unless /^\//;
push @allStorePaths, $_;
}
close READ or die "nix-store failed: $?";
# Ask the remote host which paths are invalid. # Ask the remote host which paths are invalid.
open(READ, "set -f; ssh $sshHost @sshOpts nix-store --check-validity --print-invalid @allStorePaths|"); open(READ, "set -f; ssh $sshHost @sshOpts nix-store --check-validity --print-invalid @allStorePaths|");
@ -80,7 +70,6 @@ if ($toMode) { # Copy TO the remote machine.
} }
close READ or die; close READ or die;
# Export the store paths and import them on the remote machine. # Export the store paths and import them on the remote machine.
if (scalar @missing > 0) { if (scalar @missing > 0) {
print STDERR "copying these missing paths:\n"; print STDERR "copying these missing paths:\n";
@ -93,7 +82,6 @@ if ($toMode) { # Copy TO the remote machine.
} }
else { # Copy FROM the remote machine. else { # Copy FROM the remote machine.
# Query the closure of the given store paths on the remote # Query the closure of the given store paths on the remote
@ -102,27 +90,14 @@ else { # Copy FROM the remote machine.
my $pid = open(READ, my $pid = open(READ,
"set -f; ssh @sshOpts $sshHost nix-store --query --requisites @storePaths|") or die; "set -f; ssh @sshOpts $sshHost nix-store --query --requisites @storePaths|") or die;
my @allStorePaths;
while (<READ>) { while (<READ>) {
chomp; chomp;
die "bad: $_" unless /^\//; die "bad: $_" unless /^\//;
push @allStorePaths, $_; push @missing, $_ unless isValidPath($_);
} }
close READ or die "nix-store on remote machine `$sshHost' failed: $?"; close READ or die "nix-store on remote machine `$sshHost' failed: $?";
# What paths are already valid locally?
open(READ, "set -f; @bindir@/nix-store --check-validity --print-invalid @allStorePaths|");
my @missing = ();
while (<READ>) {
chomp;
push @missing, $_;
}
close READ or die;
# Export the store paths on the remote machine and import them on locally. # Export the store paths on the remote machine and import them on locally.
if (scalar @missing > 0) { if (scalar @missing > 0) {
print STDERR "copying these missing paths:\n"; print STDERR "copying these missing paths:\n";