314 lines
7.5 KiB
Perl
314 lines
7.5 KiB
Perl
|
#!/usr/bin/env perl
|
||
|
|
#IBM(c) 2007 EPL license http://www.eclipse.org/legal/epl-v10.html
|
||
|
|
package xCAT_monitoring::rrdutil;
|
||
|
|
use strict;
|
||
|
|
use IO::Socket;
|
||
|
|
BEGIN
|
||
|
|
{
|
||
|
|
$::XCATROOT = $ENV{'XCATROOT'} ? $ENV{'XCATROOT'} : '/opt/xcat';
|
||
|
|
}
|
||
|
|
|
||
|
|
#Modules to use:
|
||
|
|
use lib "$::XCATROOT/lib/perl";
|
||
|
|
use xCAT::Utils;
|
||
|
|
|
||
|
|
################################################
|
||
|
|
#sub start_RRD_server
|
||
|
|
#Description:
|
||
|
|
# add "rrdsrv $port/tcp #RRD server" to /etc/services
|
||
|
|
# add "rrdsrv stream tcp no wait root /usr/bin/rrdtool rrdtool - $dir"
|
||
|
|
# to /etc/inetd.conf
|
||
|
|
# restart inetd
|
||
|
|
#Input:
|
||
|
|
# $port Port number of RRD server,
|
||
|
|
# $dir directory to save *.rrd
|
||
|
|
#return:
|
||
|
|
# 0 success
|
||
|
|
# !0 fail
|
||
|
|
################################################
|
||
|
|
sub start_RRD_server
|
||
|
|
{
|
||
|
|
my ($port, $dir) = @_;
|
||
|
|
my $cmd = undef;
|
||
|
|
my @old = ();
|
||
|
|
my @new = ();
|
||
|
|
my $offset = 0;
|
||
|
|
@old = xCAT::Utils->runcmd("cat /etc/services", -2);
|
||
|
|
push @new, "rrdsrv $port/tcp #RRD server";
|
||
|
|
foreach (@old) {
|
||
|
|
if ($_ =~ /rrdsrv/){
|
||
|
|
splice(@old, $offset, 1, @new);
|
||
|
|
} else {
|
||
|
|
$offset++;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
open FILE, ">/etc/services.new" or return -1;
|
||
|
|
print FILE @old;
|
||
|
|
close FILE;
|
||
|
|
$cmd = "mv -f /etc/services.new /etc/services";
|
||
|
|
xCAT::Utils->runcmd($cmd, -2);
|
||
|
|
|
||
|
|
if(! -d $dir){
|
||
|
|
$cmd = "mkdir -p $dir";
|
||
|
|
xCAT::Utils->runcmd($cmd, -2);
|
||
|
|
} else {
|
||
|
|
$cmd = "rm -rf $dir/*";
|
||
|
|
xCAT::Utils->runcmd($cmd, -2);
|
||
|
|
}
|
||
|
|
@old = ();
|
||
|
|
@new = ();
|
||
|
|
@old = xCAT::Utils->runcmd("cat /etc/inetd.conf", -2);
|
||
|
|
$offset = 0;
|
||
|
|
push @new, "rrdsrv stream tcp no wait root /usr/bin/rrdtool rrdtool - $dir";
|
||
|
|
foreach (@old) {
|
||
|
|
if ($_ =~ /rrdsrv/){
|
||
|
|
splice(@old, $offset, 1, @new);
|
||
|
|
} else {
|
||
|
|
$offset++;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
open FILE, ">/etc/inetd.conf.new" or return -1;
|
||
|
|
print FILE @old;
|
||
|
|
close FILE;
|
||
|
|
xCAT::Utils->runcmd("mv -f /etc/inetd.conf.new /etc/inetd.conf", -2);
|
||
|
|
|
||
|
|
if(xCAT::Utils->isAIX()){
|
||
|
|
xCAT::Utils->runcmd("stopsrc -s inetd", 0);
|
||
|
|
xCAT::Utils->runcmd("startsrc -s inetd", 0);
|
||
|
|
}
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
################################################
|
||
|
|
#sub stop_RRD_server
|
||
|
|
#Description:
|
||
|
|
# remove "rrdsrv $port/tcp #RRD server" from /etc/services
|
||
|
|
# remove "rrdsrv stream tcp no wait root /usr/bin/rrdtool rrdtool - $dir"
|
||
|
|
# from /etc/inetd.conf
|
||
|
|
# restart inetd
|
||
|
|
#Input:
|
||
|
|
# None
|
||
|
|
#return:
|
||
|
|
# 0 success
|
||
|
|
# !0 fail
|
||
|
|
################################################
|
||
|
|
sub stop_RRD_server
|
||
|
|
{
|
||
|
|
my @old = ();
|
||
|
|
my $offset = 0;
|
||
|
|
@old = xCAT::Utils->runcmd("cat /etc/services", -2);
|
||
|
|
foreach (@old) {
|
||
|
|
if ($_ =~ /rrdsrv/){
|
||
|
|
splice(@old, $offset, 1);
|
||
|
|
} else {
|
||
|
|
$offset++;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
open FILE, ">/etc/services.new" or return -1;
|
||
|
|
print FILE @old;
|
||
|
|
close FILE;
|
||
|
|
xCAT::Utils->runcmd("mv -f /etc/services.new /etc/services", -1);
|
||
|
|
|
||
|
|
@old = ();
|
||
|
|
@old = xCAT::Utils->runcmd("cat /etc/inetd.conf", -1);
|
||
|
|
$offset = 0;
|
||
|
|
foreach (@old) {
|
||
|
|
if ($_ =~ /rrdsrv/){
|
||
|
|
splice(@old, $offset, 1);
|
||
|
|
} else {
|
||
|
|
$offset++;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
open FILE, ">/etc/inetd.conf.new" or return -1;
|
||
|
|
print FILE @old;
|
||
|
|
close FILE;
|
||
|
|
xCAT::Utils->runcmd("mv -f /etc/inetd.conf.new /etc/inetd.conf", -2);
|
||
|
|
if(xCAT::Utils->isAIX()){
|
||
|
|
xCAT::Utils->runcmd("stopsrc -s inetd", 0);
|
||
|
|
xCAT::Utils->runcmd("startsrc -s inetd", 0);
|
||
|
|
}
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
################################################
|
||
|
|
#sub runrrdcmd_remote
|
||
|
|
#Description:
|
||
|
|
# Run the given RRD cmd on remote RRD server,
|
||
|
|
# and return the output in an array.
|
||
|
|
#Input:
|
||
|
|
# $cmd RRD command
|
||
|
|
# $host IP or hostname of RRD server
|
||
|
|
# $port Port number of RRD server,
|
||
|
|
#return:
|
||
|
|
# output of command
|
||
|
|
################################################
|
||
|
|
sub runrrdcmd_remote
|
||
|
|
{
|
||
|
|
my($cmd, $host, $port) = @_;
|
||
|
|
|
||
|
|
my $socket = IO::Socket::INET->new
|
||
|
|
(PeerAddr=>$host,
|
||
|
|
PeerPort=>$port,
|
||
|
|
Proto=>'tcp',
|
||
|
|
Type=>SOCK_STREAM);
|
||
|
|
|
||
|
|
if(! $socket){
|
||
|
|
print "ERROR: to connect with $host:$port\n";
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
|
||
|
|
print $socket "$cmd\n";
|
||
|
|
my $output = [];
|
||
|
|
my $line = undef;
|
||
|
|
while($line = <$socket>){
|
||
|
|
push @$output, $line;
|
||
|
|
if($line =~ /(OK|ERROR)/){
|
||
|
|
last;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
print $socket "quit";
|
||
|
|
close($socket);
|
||
|
|
return $output;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
################################################
|
||
|
|
#sub RRD_create
|
||
|
|
#Description:
|
||
|
|
# RRD_create will overwrite a RRdb if it already exists,
|
||
|
|
#Input:
|
||
|
|
# $rrd filename of RRD to create
|
||
|
|
# $sum if $sum != 0 to create a database file for SUMMARY information
|
||
|
|
# $step the base interval in seconds with which data will be fed into the RRD
|
||
|
|
# $start_time the time in seconds when the first value should be added to the RRD
|
||
|
|
# $ds_type ds-name:GAUGE | COUNTER | DERIVE | ABSOLUTE
|
||
|
|
# $data_source the IP or hostname of remote host
|
||
|
|
#return:
|
||
|
|
# 0 sucess
|
||
|
|
# !0 fail
|
||
|
|
################################################
|
||
|
|
sub RRD_create
|
||
|
|
{
|
||
|
|
my ($rrd, $sum, $step, $start_time, $ds_type, $data_source) = @_;
|
||
|
|
my $output = [];
|
||
|
|
my $heartbeat = 8 * $step;
|
||
|
|
my $cmd = "create $rrd --start $start_time --step $step DS:sum:$ds_type:$heartbeat:U:U";
|
||
|
|
if($sum){
|
||
|
|
$cmd = $cmd." DS:num:$ds_type:$heartbeat:U:U";
|
||
|
|
}
|
||
|
|
#TODO: Specified custom RR archives here?
|
||
|
|
$cmd = $cmd." RRA:AVERAGE:0.5:1:244 RRA:AVERAGE:0.5:24:244 RRA:AVERAGE:0.5:168:244 RRA:AVERAGE:0.5:672:244 RRA:AVERAGE:0.5:5760:374";
|
||
|
|
if(defined($data_source)){
|
||
|
|
$output = &runrrdcmd_remote($cmd, $data_source, 13900);
|
||
|
|
} else {
|
||
|
|
@$output = xCAT::Utils->runcmd("rrdtool $cmd", 0);
|
||
|
|
}
|
||
|
|
my $line = pop(@$output);
|
||
|
|
if($line =~ /ERROR/){
|
||
|
|
return -1;
|
||
|
|
} else {
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
################################################
|
||
|
|
#sub RRD_update
|
||
|
|
#Description:
|
||
|
|
# RRD_update
|
||
|
|
#Input:
|
||
|
|
# $rrd filename of RRD to update
|
||
|
|
# $sum sum of all numeric metrics
|
||
|
|
# $num number of all numeric metrics, should be null for a host metrics
|
||
|
|
# $process_time update time
|
||
|
|
# $data_source the IP or hostname of remote host
|
||
|
|
#return:
|
||
|
|
# 0 sucess
|
||
|
|
# !0 fail
|
||
|
|
################################################
|
||
|
|
sub RRD_update
|
||
|
|
{
|
||
|
|
my ($rrd, $sum, $num, $process_time, $data_source) = @_;
|
||
|
|
my $output = [];
|
||
|
|
my $cmd = "update $rrd";
|
||
|
|
if($num ne "null"){
|
||
|
|
$cmd = $cmd." $process_time:$sum:$num";
|
||
|
|
} else {
|
||
|
|
$cmd = $cmd." $process_time:$sum";
|
||
|
|
}
|
||
|
|
if(defined($data_source)){
|
||
|
|
$output = &runrrdcmd_remote($cmd, $data_source, 13900);
|
||
|
|
} else {
|
||
|
|
@$output = xCAT::Utils->runcmd("rrdtool $cmd", 0);
|
||
|
|
}
|
||
|
|
my $line = pop(@$output);
|
||
|
|
if($line =~ /ERROR/){
|
||
|
|
return -1;
|
||
|
|
} else {
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
################################################
|
||
|
|
#sub RRD_fetch
|
||
|
|
#Description:
|
||
|
|
# RRD_fetch
|
||
|
|
#Input:
|
||
|
|
# $rrd filename of RRD to update
|
||
|
|
# $start_time start of time series
|
||
|
|
# $end_time end of time series
|
||
|
|
# $data_source the IP or hostname of remote host
|
||
|
|
#return:
|
||
|
|
# 0 sucess
|
||
|
|
# !0 fail
|
||
|
|
################################################
|
||
|
|
sub RRD_fetch
|
||
|
|
{
|
||
|
|
my ($rrd, $start_time, $end_time, $data_source) = @_;
|
||
|
|
my $output = [];
|
||
|
|
my $resolution = undef;
|
||
|
|
my $cmd = undef;
|
||
|
|
$cmd = "fetch $rrd AVERAGE -s $start_time -e $end_time";
|
||
|
|
if(defined($data_source)){
|
||
|
|
$output = &runrrdcmd_remote($cmd, $data_source, 13900);
|
||
|
|
} else {
|
||
|
|
@$output = xCAT::Utils->runcmd("rrdtool $cmd", 0);
|
||
|
|
}
|
||
|
|
return $output;
|
||
|
|
}
|
||
|
|
|
||
|
|
################################################
|
||
|
|
#sub push_data_to_rrd
|
||
|
|
#Description:
|
||
|
|
# push_data_to_rrd
|
||
|
|
#Input:
|
||
|
|
# $rrd filename of RRD to update
|
||
|
|
# $sum sum of all numeric metrics
|
||
|
|
# $num number of all numeric metrics, should be null for a host metrics
|
||
|
|
# $step the base interval in seconds with which data will be fed into the RRD
|
||
|
|
# $process_time update time
|
||
|
|
# $ds_type ds-name:GAUGE | COUNTER | DERIVE | ABSOLUTE
|
||
|
|
# $data_source the IP or hostname of remote host
|
||
|
|
#return:
|
||
|
|
# 0 sucess
|
||
|
|
# !0 fail
|
||
|
|
################################################
|
||
|
|
|
||
|
|
sub push_data_to_rrd
|
||
|
|
{
|
||
|
|
my $ret = 0;
|
||
|
|
my($rrd, $sum, $num, $step, $process_time, $ds_type, $data_source) = @_;
|
||
|
|
my $summary = $num eq 'null' ? 0 : 1;
|
||
|
|
if(! -f $rrd){
|
||
|
|
$ret = RRD_create($rrd, $summary, $step, $process_time-$step, $ds_type, $data_source);
|
||
|
|
if($ret != 0){
|
||
|
|
return $ret;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
$ret = RRD_update($rrd, $sum, $num, $process_time, $data_source);
|
||
|
|
return $ret;
|
||
|
|
}
|
||
|
|
|
||
|
|
1;
|