jo

1 files changed, 115 insertions, 14 deletions

diff --git a/cpupload.pl b/cpupload.pl
index a490a89..74c1bed 100755
--- a/cpupload.pl
+++ b/cpupload.pl
@@ -4,30 +4,50 @@
 use strict;
 use warnings;
 
+use IPC::Open2;
+use Data::Dumper;
+
+use threads;
+use threads::shared;
+
+$| = 1;
+
+my %GLOBAL_STATS :shared;
+my %GLOBAL_CONF  :shared;
+
+%GLOBAL_CONF = (
+	'sleep' => 1,
+	size => 10,
+	events => 20,
+);
+
 sub say (@) {
    	print "$_\n" for @_;
-	return scalar @_;
+	return undef;
 }
 
 sub reduce (&@) {
-   	
-}
+	my ($func, @params) = @_;
 
-sub parse_stat_line ($) {
-   	my %load;
-	my ($name, $user, $nice, $system, $idle, @rest) = split / +/, shift;
-	my $total = $user + $nice + $system + $idle;
-	print $nice, "\n";
+	my $sub;
+	$sub = sub { 
+		my ($elem, @rest) = @_;
+		$func->($elem, @rest == 1 ? $rest[0] : $sub->(@rest));
+	};
 
-	return undef;
+	return $sub->($func, @params);
 }
 
-sub parse_stat (@) {
-	my ($total,@rest) = @_;
+sub sum (@) { reduce { $_[0] + $_[1] } @_ }
 
-	parse_stat_line $total;
+sub loop (&) { $_[0]->() while 1 }
 
-	return undef;
+sub parse_cpu_line ($) {
+   	my %load;
+	@load{qw(name user nice system iowait irq softirq)} = split ' ', shift;
+	$load{TOTAL} = sum @load{qw(user)};
+
+	return ($load{name}, \%load);
 }
 
 sub get_local_stat () {
@@ -38,9 +58,90 @@ sub get_local_stat () {
 	return @stat;
 }
 
+sub get_remote_stat ($) {
+   	my $host = shift;
+
+	loop {
+		my $sleep = $GLOBAL_CONF{sleep};
+		my $events = $GLOBAL_CONF{events};
+
+		my $pid = open2 my $out, my $in, qq{ 
+		   	ssh $host 'for i in \$(seq $events); do cat /proc/stat; sleep $sleep; done'
+		} or die "Error: $!\n";
+
+		$SIG{STOP} = sub {
+			say "Shutting down get_remote_stat($host) & $pid";
+			kill 1, $pid;
+			threads->exit();
+		};
+
+		while (<$out>) {
+		   	/^cpu/ && do {
+				my ($name, $val) = parse_cpu_line $_;
+				$GLOBAL_STATS{"$host;$name"} = $val->{TOTAL};
+			}
+		}
+	}
+}
+
+sub display_stats () {
+	$SIG{STOP} = sub {
+		say "Shutting down display_stats";
+		threads->exit();
+	};
+
+   	my $size = $GLOBAL_CONF{size};
+
+	my $print_header = sub {
+		sleep $GLOBAL_CONF{sleep} until %GLOBAL_STATS;
+
+	   	my (@header) = ('', '');
+		for my $key (sort keys %GLOBAL_STATS) {
+	      		my ($host, $name) = split ';', $key;
+			$header[0] .= sprintf "%${size}s ", $host;
+			$header[1] .= sprintf "%${size}s ", $name;
+		}
+
+		say @header;
+	};
+
+	$print_header->();
+	my $header_counter = 0;
+
+   	loop { 
+		unless (++$header_counter % 10) {
+			$header_counter = 0;
+			$print_header->();
+		}
+
+		my $line = '';
+
+		$line .= sprintf "%${size}d", $GLOBAL_STATS{$_} for sort keys %GLOBAL_STATS;
+
+		say $line;
+
+	   	sleep $GLOBAL_CONF{sleep};
+	}
+}
+
 sub main () {
-	parse_stat get_local_stat;
+   	my @threads;
+	push @threads, threads->create('get_remote_stat', 'localhost');
+	push @threads, threads->create('display_stats');
+
+	while (<STDIN>) {
+		/^q/ && last;
+		/^s/ && do {
+		   $GLOBAL_CONF{sleep} = <STDIN>;
+		};
+	}
+
+	for (@threads) {
+		$_->kill('STOP');
+		$_->join();
+	}
 
+	say "Good bye";
 	exit 0;
 }