tange/parallel
1
0
Fork 0
mirror of https://git.savannah.gnu.org/git/parallel.git synced 2024-11-26 07:57:58 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

niceload: --sensor implemented.

Browse source
This commit is contained in:
Ole Tange 2012-11-21 22:30:29 +01:00
parent 6748cf82c8
commit 7648a3c0b2
1 changed files with 67 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

74
src/niceload
View file

@ -24,7 +24,7 @@
use strict; use strict;
use Getopt::Long; use Getopt::Long;
$Global::progname="niceload"; $Global::progname="niceload";
$Global::version = 20121022; $Global::version = 20121122;
Getopt::Long::Configure("bundling","require_order"); Getopt::Long::Configure("bundling","require_order");
get_options_from_array(\@ARGV) || die_usage(); get_options_from_array(\@ARGV) || die_usage();
if($::opt_version) { if($::opt_version) {
@ -109,6 +109,7 @@ sub get_options_from_array {
"factor|f=s" => \$::opt_factor, "factor|f=s" => \$::opt_factor,
"hard|H" => \$::opt_hard, "hard|H" => \$::opt_hard,
"soft|S" => \$::opt_soft, "soft|S" => \$::opt_soft,
"sensor=s" => \$::opt_sensor,
"si|sio|startio|start-io=s" => \$::opt_start_io, "si|sio|startio|start-io=s" => \$::opt_start_io,
"ri|rio|runio|run-io=s" => \$::opt_run_io, "ri|rio|runio|run-io=s" => \$::opt_run_io,
@ -541,20 +542,79 @@ sub sleep_while_running {
} }
sub nonblockGetLines {
# An non-blocking filehandle read that returns an array of lines read
# Returns: ($eof,@lines)
my ($fh,$timeout) = @_;
$timeout = 0 unless defined $timeout;
my $rfd = '';
$::nonblockGetLines_last{$fh} = ''
unless defined $::nonblockGetLines_last{$fh};
vec($rfd,fileno($fh),1) = 1;
return unless select($rfd, undef, undef, $timeout)>=0;
# I'm not sure the following is necessary?
return unless vec($rfd,fileno($fh),1);
my $buf = '';
my $n = sysread($fh,$buf,1024*1024);
# If we're done, make sure to send the last unfinished line
return (1,$::nonblockGetLines_last{$fh}) unless $n;
# Prepend the last unfinished line
$buf = $::nonblockGetLines_last{$fh}.$buf;
# And save any newly unfinished lines
$::nonblockGetLines_last{$fh} =
(substr($buf,-1) !~ /[\r\n]/ && $buf =~ s/([^\r\n]*)$//)
? $1 : '';
$buf ? (0,split(/\n/,$buf)) : (0);
}
# --sensor 'vmstat 1 | perl -ane '\''$|=1; 4..0 and print $F[8],"\n"'\'
sub load_status { sub load_status {
# Returns: # Returns:
# loadavg # loadavg or sensor measurement
my $self = shift; my $self = shift;
# Cache for some seconds if($::opt_sensor) {
if(not defined $self->{'load_status'} or my $fh = $self->{'sensor_fh'};
$self->{'load_status_cache_time'}+$self->{'recheck'} < time) { if(not defined $fh) {
$self->{'load_status'} = load_status_linux(); # Start the sensor
$self->{'load_status_cache_time'} = time; ::debug("start: $::opt_sensor\n");
open($fh, "-|", $::opt_sensor) || ::die_bug("Cannot open: $::opt_sensor");
$self->{'sensor_fh'} = $fh;
}
::debug("sensor: $::opt_sensor");
# Read as much as we can (block for up to 0.02 seconds)
my ($eof,@lines) = nonblockGetLines($fh,0.2);
my $newval = pop @lines;
::debug(" = $newval\n");
if(defined $newval) {
$newval =~ s/\D*([0-9.]*)(\D.*)?$/$1/;
$self->{'load_status'} = undef_as_zero($newval);
}
if($eof) {
close($fh);
undef $self->{'sensor_fh'};
}
} else {
# Normal load avg
# Cache for some seconds
if(not defined $self->{'load_status'} or
$self->{'load_status_cache_time'}+$self->{'recheck'} < time) {
$self->{'load_status'} = load_status_linux();
$self->{'load_status_cache_time'} = time;
}
} }
::debug("load_status: $self->{'load_status'}\n"); ::debug("load_status: $self->{'load_status'}\n");
return $self->{'load_status'}; return $self->{'load_status'};
} }
sub undef_as_zero {
my $a = shift;
return $a ? $a : 0;
}
sub load_status_linux { sub load_status_linux {
my ($loadavg); my ($loadavg);