Takfile

   1 package Tak::MyScript;
   2
   3 use Tak::Takfile;
   4
   5 sub every_exec (stream|s) {
   6   my ($self, $remotes, $options, @command) = @_;
   7
   8   my @requests;
   9
  10   $_->ensure(command_service => 'Tak::CommandService') for @$remotes;
  11
  12   foreach my $remote (@$remotes) {
  13     if ($options->{stream}) {
  14       my $stdout = $self->stdout;
  15       my $host = $remote->host;
  16       push @requests, $remote->start(
  17         {
  18           on_result => sub { $self->print_exec_result($remote, @_) },
  19           on_progress => sub {
  20             $stdout->print($host.' '.$_[0].': '.$_[1]);
  21             $stdout->print("\n") unless $_[1] =~ /\n\Z/;
  22           }
  23         },
  24         command_service => stream_exec => \@command
  25       );
  26     } else {
  27       push @requests, $remote->start(
  28         { on_result => sub { $self->print_exec_result($remote, @_) } },
  29         command_service => exec => \@command
  30       );
  31     }
  32   }
  33   Tak->await_all(@requests);
  34 }
  35
  36 sub print_exec_result {
  37   my ($self, $remote, $result) = @_;
  38
  39   my $res = eval { $result->get }
  40     or do {
  41       $self->stderr->print("Host ${\$remote->host}: Error: $@\n");
  42       return;
  43     };
  44
  45   my $code = $res->{exit_code};
  46   $self->stdout->print(
  47     "Host ${\$remote->host}: ".($code ? "NOT OK ${code}" : "OK")."\n"
  48   );
  49   if ($res->{stderr}) {
  50     $self->stdout->print("Stderr:\n${\$res->{stderr}}\n");
  51   }
  52   if ($res->{stdout}) {
  53     $self->stdout->print("Stdout:\n${\$res->{stdout}}\n");
  54   }
  55 }
  56
  57 1;