diff options
Diffstat (limited to 'urpm/parallel_ka_run.pm')
| -rw-r--r-- | urpm/parallel_ka_run.pm | 109 |
1 files changed, 109 insertions, 0 deletions
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; |