File Coverage

bin/waitjobs

Criterion	Covered	Total	%
statement	39	64	60.9
branch	2	10	20.0
condition	1	5	20.0
subroutine	10	12	83.3
pod			n/a
total	52	91	57.1

#!/usr/bin/env perl
#ABSTRACT: Wait for selected jobs to exit the queue
#PODNAME: waitjobs

use v5.12;
use warnings;
use Getopt::Long;
use FindBin qw($RealBin);
use Data::Dumper;
use Term::ANSIColor qw(:constants);
if (-e "$RealBin/../dist.ini") {
    say STDERR "[dev mode] Using local lib"  if ($ENV{"DEBUG"});
    use lib "$RealBin/../lib";
}
use NBI::Slurm;
use NBI::Queue;
$Data::Dumper::Sortkeys = 1;

my $opt_user = $ENV{USER};
my $opt_name = '.+';
my $opt_status = '';
my $opt_verbose_bool = 0;
my $opt_debug_bool = 0;
my $refresh_time = 20;
# Check if this script is running as a job, in case save the job id
my $self_job_id = $ENV{SLURM_JOB_ID} // undef;
GetOptions(
    'u|user=s' => \$opt_user,
    'n|name=s' => \$opt_name,
    'r|refresh=i' => \$refresh_time,
    'verbose' => \$opt_verbose_bool,
    'version' => sub { say "waitjobs v", $NBI::Slurm::VERSION; exit },
    'debug' => \$opt_debug_bool,
    'help' => sub { usage() },
);

my $queue = NBI::Queue->new(
    -username => $opt_user,
    -name => $opt_name,
);
# If $self_job_id is anq integer, remove it from the queue
if (defined $self_job_id and $self_job_id =~ /^\d+$/) {
    print STDERR timelog("waitjobs"), "Ignoring current job id $self_job_id\n";
    $queue->remove($self_job_id);
}

my @starting_ids = @{$queue->ids()};
print STDERR timelog("waitjobs"),  "There are ",
  GREEN, BOLD, scalar @starting_ids, RESET  " jobs for ",
  GREEN, BOLD, $opt_user,     RESET, " with name matching ",
  GREEN, BOLD, $opt_name,     RESET, "\n";

while (scalar @starting_ids > 0) {
    # Create the new queue with the same parameters as the initial queue
    my $current_jobs = NBI::Queue->new(
        -username => $opt_user,
        -name => $opt_name,
    );
    my @current_ids = @{$current_jobs->ids()};

    # Remove from @starting_ids all the jobs that are NOT in @current_ids
    my @new_starting_ids = ();
    for my $starting_id (@starting_ids) {
        if (grep {$_ eq $starting_id} @current_ids) {
            push(@new_starting_ids, $starting_id);
        }
    }
    @starting_ids = @new_starting_ids;
    if ($opt_verbose_bool) {
        say STDERR timelog("waitjobs"), "Waiting for ", scalar @starting_ids, " jobs to finish (refresh in $refresh_time seconds)";
    }
    sleep $refresh_time;
}


say STDERR timelog("waitjobs"), "No jobs left: quitting\n";

sub usage {
    say STDERR<

  Usage: waitjobs [options] [job_name]
  ------------------------------------
    Wait for selected jobs to exit the queue

  Options:
   -u, --user       default current user
   -n, --name       default '.+' (matches any name)
   -r, --refresh    check every 20 seconds

END
    exit;
}

__END__