From 295d031d89e99fd41f519bbea819eb83df07a480 Mon Sep 17 00:00:00 2001
From: Francois Pons <fpons@mandriva.com>
Date: Fri, 23 Aug 2002 14:23:05 +0000
Subject: 4.0-2mdk (was not added in CVS for 1mdk).

---
 urpm/parallel_ka_run.pm | 109 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 109 insertions(+)
 create mode 100644 urpm/parallel_ka_run.pm

(limited to 'urpm/parallel_ka_run.pm')

diff --git a/urpm/parallel_ka_run.pm b/urpm/parallel_ka_run.pm
new file mode 100644
index 00000000..dacc1056
--- /dev/null
+++ b/urpm/parallel_ka_run.pm
@@ -0,0 +1,109 @@
+package urpm::parallel_ka_run;
+
+#- parallel resolve_dependencies
+sub parallel_resolve_dependencies {
+    my ($parallel, $synthesis, $urpm, $state, $requested, %options) = @_;
+
+    #- first propagate the synthesis file to all machine.
+    $urpm->{log}("parallel_ka_run: mput $parallel->{options} -- '$synthesis' '$synthesis'");
+    system "mput $parallel->{options} -- '$synthesis' '$synthesis'";
+    $parallel->{synthesis} = $synthesis;
+
+    #- compute command line of urpm? tools.
+    my $line = $options{auto_select} ? ' --auto-select' : '';
+    foreach (keys %$requested) {
+	if (/\|/) {
+	    #- simplified choices resolution.
+	    my $choice = $options{callback_choices}->($urpm, undef, $state, [ map { /^\d+$/ ?
+										      $urpm->{depslist}[$_] :
+											$urpm->search($_) } split '\|', $_ ]);
+	    $line .= ' '.$choice->fullname;
+	} else {
+	    my $pkg = $urpm->{depslist}[$_] or next;
+	    $line .= ' '.$pkg->fullname;
+	}
+    }
+
+    #- execute urpmq to determine packages to install.
+    my ($node, $cont, %chosen);
+    local (*F, $_);
+    do {
+	$cont = 0; #- prepare to stop iteration.
+	#- the following state should be cleaned for each iteration.
+	delete $state->{selected};
+	#- now try an iteration of urpmq.
+	$urpm->{log}("parallel_ka_run: rshp -v $parallel->{options} -- urpmq --synthesis $synthesis -f $line ".join(' ', keys %chosen));
+	open F, "rshp -v $parallel->{options} -- urpmq --synthesis $synthesis -fdu $line ".join(' ', keys %chosen)." |";
+	while ($_ = <F>) {
+	    chomp;
+	    s/<([^>]*)>.*:->:(.*)/$2/ and $node = $1;
+	    if (/\|/) {
+		#- distant urpmq returned a choices, check if it has already been chosen
+		#- or continue iteration to make sure no more choices are left.
+		$cont ||= 1; #- invalid transitory state (still choices is strange here if next sentence is not executed).
+		unless (grep { exists $chosen{$_} } split '\|', $_) {
+		    #- it has not yet been chosen so need to ask user.
+		    $cont = 2;
+		    my $choice = $options{callback_choices}->($urpm, undef, $state, [ map { $urpm->search($_) } split '\|', $_ ]);
+		    $chosen{scalar $choice->fullname} = $choice;
+		}
+	    } else {
+		my $pkg = $urpm->search($_) or next; #TODO
+		$state->{selected}{$pkg->id}{$node} = $_;
+	    }
+	}
+	close F or $urpm->{fatal}(1, _("host %s does not have a good version of urpmi", $node));
+	#- check for internal error of resolution.
+	$cont == 1 and die "internal distant urpmq error on choice not taken";
+    } while ($cont);
+
+    #- keep trace of what has been chosen finally (if any).
+    $parallel->{line} = "$line ".join(' ', keys %chosen);
+
+    #- update ask_remove, ask_unselect too along with provided value.
+    #TODO
+}
+
+#- parallel install.
+sub parallel_install {
+    my ($parallel, $urpm, $remove, $install, $upgrade) = @_;
+
+    foreach (values %$install, values %$upgrade) {
+	my ($basename) = /([^\/]*)$/;
+	$urpm->{log}("parallel_ka_run: mput $parallel->{options} -- '$_' $urpm->{cachedir}/rpms/$basename");
+	system "mput $parallel->{options} -- '$_' $urpm->{cachedir}/rpms/$basename";
+    }
+
+    $urpm->{log}("parallel_ka_run: rshp -v $parallel->{options} -- urpmi --synthesis $parallel->{synthesis} $parallel->{line}");
+    system "rshp -v $parallel->{options} -- urpmi --auto --synthesis $parallel->{synthesis} $parallel->{line}";
+}
+
+
+#- allow bootstrap from urpmi code directly (namespace is urpm).
+package urpm;
+sub handle_parallel_options {
+    my ($urpm, $options) = @_;
+    my ($ka_run_options) = $options =~ /ka-run:(.*)/;
+
+    if ($ka_run_options) {
+	my ($flush_nodes, %nodes);
+
+	foreach (split ' ', $ka_run_options) {
+	    if ($_ eq '-m') {
+		$flush_nodes = 1;
+	    } else {
+		$flush_nodes and $nodes{/host=([^,]*)/ ? $1 : $_} = undef;
+		undef $flush_nodes;
+	    }
+	}
+
+	return bless {
+		      options => $ka_run_options,
+		      nodes   => \%nodes,
+		     }, "urpm::parallel_ka_run";
+    }
+
+    return undef;
+}
+
+1;
-- 
cgit v1.2.1