Subversion Repositories havirt

                  &makeCommand &forceScan

                  &executeAndWait

# scans a node to determine which domains are running on it

sub getDomainsOnNode {

   my $node = shift;

   my @nodeList = grep { /^\s*\d/ } `ssh $node 'virsh list'`;

   for ( my $i = 0; $i < @nodeList; $i++ ) {

      if ( $nodeList[$i] =~ m/\s*\d+\s*([^ ]+)/ ) {

         $nodeList[$i] = $1;

      }

   }

   my %hash = map{ $_ => time } @nodeList;

   return \%hash;

}

# check one or more nodes and determine which domains are running on them.

# defaults to everything in the node database, but the -t can have it run on only one

# this is the function that should be run every few minutes on one of the servers

sub scan {

   if ( -f $main::lastScanFileName && ! $main::force ) {

      my $lastScan = time - ( stat( $main::lastScanFileName ) ) [9];

      return "Scan was run $lastScan seconds ago\n" unless $lastScan > $main::minScanTimes;

   }

   `touch $main::lastScanFileName`;

   &main::readDB(1);

   print Dumper( $main::statusDB->{'nodePopulation'} ) if $main::DEBUG > 2;

   my @targets;

   if ( $main::targetNode ) {

      push @targets, $main::targetNode;

   } else {

      @targets = keys %{$main::statusDB->{'node'}};

   }

   print "Scanning " . join( "\n", @targets ) . "\n" if $main::DEBUG;

   foreach my $node (@targets) {

      $main::statusDB->{'nodePopulation'}->{$node}->{'running'} = &getDomainsOnNode( $node );

      $main::statusDB->{'nodePopulation'}->{$node}->{'lastchecked'} = time;

      foreach my $domain ( keys %{$main::statusDB->{'nodePopulation'}->{$node}->{'running'}} ) {

         # make sure there is an entry for all of these domains

         $main::statusDB->{'virt'}->{$domain} = {} unless exists( $main::statusDB->{'virt'}->{$domain} );

      }

      print Dumper( $main::statusDB->{'nodePopulation'}->{$node} ) if $main::DEBUG > 2;

   }

   &main::writeDB();

   return "Node(s) updated\n";

}

sub makeCommand {

   my ( $node, $command ) = @_;

   return "ssh $node '$command'";

}

sub forceScan {

   my $save = $main::force;

   $main::force = 1;

   &main::scan();

   $main::force = $save;

}

# executes command $command, then repeatedly runs virsh list

# on $scanNode, grep'ing for $scanDomain

# $condition is 1 (true) or 0 (false)

sub executeAndWait {

   my ( $command, $scanNode, $scanDomain, $condition ) = @_;

   my $waitSeconds = 5; # number of seconds to wait before checking again

   my $maxIterations = 60 / $waitSeconds; # maximum number of tries

   print "Running [$command], then waiting $waitSeconds to check if complete\n" if $main::DEBUG;

   `$command`;

   my $waitCommand = &makeCommand( $scanNode, "virsh list | grep $scanDomain" );

   my $output = '';

   do {

      return 0 unless ( $maxIterations-- ); # we've waited too long, so probably not working

      print '. ';

      sleep 5;

      $output = `$waitCommand`;

      print "[$waitCommand] returned [$output]\n" if $main::DEBUG > 1;

   } until ( $condition ? $output : !$output );

   return 1; # made it successful

}