API for rrdtool

git-svn-id: https://svn.code.sf.net/p/xcat/code/xcat-core/trunk@3593 8638fb3e-16cb-4fca-ae20-7b5d299a9bcd
This commit is contained in:
ellen56 2009-06-17 06:57:37 +00:00
parent 6ccc776d32
commit b231f0c745

View File

@ -0,0 +1,313 @@
#!/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;