磁盘IO监控脚本

#!/usr/bin/perl -w
use strict;
use Regexp::Common;
use Getopt::Long;
my %values;
my $result;
my ($iostat,$data,$perfdata);
my ($warning,$critical,$disk);
$result = GetOptions("disk=s"      =>      \$disk,
                     "d=s"         =>      \$disk,
                     "warning=i"   =>      \$warning,
                     "w=i"         =>      \$warning,
                     "critical=i"  =>      \$critical,
                     "c=i"         =>      \$critical,
                     "help"        =>      \&help,
                     );
$warning ||= 75;
$critical ||= 80;
$disk ||= undef;
&main();
sub main(){
        &set_iostat();
        &get_data();
        if ( $values{'%util'} < $warning )
        {
            print "TEST_IO OK - " . "$data| $perfdata\n";
        }
        elsif ( $values{'%util'} >= $warning )
        {
            print "TEST_IO WARNING - " . "$data| $perfdata\n";
        }
        elsif ( $values{'%util'} >= $critical )
        {
            print "TEST_IO CRITICAL - " . "$data| $perfdata\n";
        }
        else #No clue what value is there!
        {
             print "UNKNOWN - Error in command output\n";
        }
}
sub set_iostat(){
        if (defined $disk){
                $iostat = "iostat " . "$disk";
        }else{
                $iostat = "iostat";
        }
}
sub get_data(){
        &set_iostat();
        my $output = `$iostat -d -x 1 3 |tail -n 3`;
        unless($output) {
            print "Error executing iostat command\n";
        }
        $output =~ /Device.*?/gs;
        foreach my $key (qw(rrqm/s wrqm/s r/s w/s rsec/s wsec/s avgrq-sz avgqu-sz await svctm %util)){
                if($output =~ /\G.*?($RE{num}{real})/gs){
                        $data .= sprintf("%s %s ", $key, $1);
                }
        }
        %values = split ' ', $data;
        $data = '';
        foreach my $key(qw(r/s w/s avgqu-sz await svctm %util)){
            $data .= sprintf("%s %s ",$key,$values{$key});
            #排除(await、svctm、%util)的perfdata输出
            # if( $key eq 'r/s'|| $key eq 'w/s'|| $key eq 'avgqu-sz'){
                $perfdata .= $key . "=" . $values{$key} . " ";
            # }
        }
}
sub help{
    print "Usage: -d|--disk=</path/to/iostat> -w|--warning=<warn> -c|--critical=<crit>\n";
}