Topic: Short monitoring script for pool (Read 1014 times)

#!/usr/bin/perl
# This program is free software. It comes without any warranty, to
# the extent permitted by applicable law. You can redistribute it
# and/or modify it under the terms of the Do What The Fuck You Want
# To Public License, Version 2, as published by Sam Hocevar. See
# http://sam.zoy.org/wtfpl/COPYING for more details.

# sudo aptitude install libjson-perl sendemail

use strict;
use Data::Dumper;
use JSON;

my $request = 'http://www.btcguild.com/api.php?api_key=xxxxxxxxxxxxxx';
my $email   = '[email protected]';
my $from_email = '[email protected]';
my $smtp_server = 'smtp.fastmail.fm';
my $smtp_user = '[email protected]';
my $smtp_pass = 'XXXXXXXX';
my $sendemail = "sendemail -f $from_email -t $email -s $smtp_server -xu $smtp_user -o message-format=raw -xp $smtp_pass -m";
my $worker_prefix = 'xxx';
my $filename = '/tmp/.guild_timeouts';
my $dead_thres = 150.0;

my $h_timeout = {};

sub main {
my @lines = ();

read_timeouts();

sysexec('localhost', "wget --output-document=- $request 2>/dev/null", \@lines);
return 1 if ($lines[0] =~ /Failed to connect to/);
my $json = JSON->new->allow_nonref;
my $guild = $json->decode($lines[0]);
print_out("\$guild: $guild");

my $workers = $guild->{'workers'};
my $total = 0.0;
my @dead;
my @choped_dead;
for my $name (sort keys (%$workers)) {
my $worker_name = $workers->{$name}->{worker_name};
my $hash_rate   = $workers->{$name}->{hash_rate};
print "$workers->{$name}->{worker_name}: $workers->{$name}->{hash_rate}\n";
if ($hash_rate < $dead_thres) {
push (@dead, $worker_name);
} else {
update_flag($worker_name, 1);
}
$total += $hash_rate;
}
print "Total: $total\n";

my $must_send = 0;
for my $name (@dead) {
my $choped = $name; $choped =~ s/$worker_prefix//g;
push @choped_dead, $choped;
if (down_flag($name)) {
$must_send = 1;
update_flag($name, 0);
}
}
if ($must_send) {
@lines = ();
print_log("Will send: @dead");
sysexec('localhost', $sendemail . " '@choped_dead'", \@lines);
}
write_timeouts();
print "\n";
}

sub update_flag {
my $worker = shift;
my $flag = shift;

$h_timeout->{$worker} = $flag;
}

sub down_flag {
my $worker = shift;

my $t = exists $h_timeout->{$worker} ? $h_timeout->{$worker} : 1;
return $t;
}

sub read_timeouts {
my $f;
local $/=undef;
open($f, '<', $filename) or return;
my $txt = <$f>;
eval $txt;
close $f;
}

sub write_timeouts {
my $f;
open($f, '>', $filename) or return;
my $txt = Data::Dumper->Dump([$h_timeout], ['$h_timeout']);
print $f $txt;
close $f;
}

my $local_hostname = 'localhost';
sub sysexec {
my ($host, $cmd, $lines) = @_;
my $aaa;
my $rc = 1;
local *KID;

return 1 if (!defined $host || $host eq '');
if ($host ne $local_hostname) {
$cmd = "ssh -o PreferredAuthentications=publickey $host -t '$cmd' 2>/dev/null";
}
my $pid = open(KID, "$cmd |");
if ($pid) { # parent
while (defined($aaa = )) {
push(@$lines, $aaa);
}
close(KID);
$rc = ($?>> 8);
} else {
print_log("FATAL: Could not execute command '$cmd'");
return 1;
}
return $rc;
}

sub print_log {
my ($txt) = @_;
print $txt . "\n";
}

&main();

1;