Ticket #143: munin-quota-usage_.2

#!/usr/bin/perl                                                                                                           
#                                                                                                                         
# Plugin to monitor quota on a specified device. Needs repquota and root privs                                            
#                                                                                                                         
# Usage: place in /etc/munin/node.d/quota-usage_<dev> (or link it there using                                             
# ln -s), for example quota-usage_hda3. Use underscores instead of slashes, for                                           
# example to monitor /dev/mapper/vol-foo, name this quota-usage_mapper_vol-foo                                            
#                                                                                                                         
# This plugin uses the soft and hard quota data for warning and critical levels                                           
# respectively.                                                                                                           
#                                                                                                                         
# Parameters understood:                                                                                                  
#                                                                                                                         
#       config   (required)                                                                                               
#                                                                                                                         
#%# family=manual                                                                                                         
#                                                                                                                         
use strict;
use warnings;

# We do some magic to allow device strings with underscore to mean a /                                                    
# So to monitor /dev/mapper/vol-foo use quota-usage_mapper_vol-foo                                                        
my @tmp = split(/_/, $0);
shift @tmp;
my $dev = join("/", @tmp);
my %users;

open (REP, "/usr/sbin/repquota /dev/$dev |") or exit 22;
while (<REP>) {
        chomp;
        if (/^-+$/../^$/) {
                my @data = split(/ +/);
                if ( @data > 2 && $data[2] > 0 ) {
                    $users{$data[0]}[0] = $data[2]; # current usage                                                       
                    $users{$data[0]}[1] = $data[3]; # Soft quota                                                          
                    $users{$data[0]}[2] = $data[4]; # Hard quota                                                          
                }
        }
}
close REP;


my @sorted_users;
@sorted_users = sort keys %users;    # sorted by username                                                                 



my $user;
my @value;

if ($ARGV[0] and $ARGV[0] eq "config") {
        print "graph_title Filesystem usage by user (quota based) on $dev\n";
        print "graph_args --base 1024 --lower-limit 0\n";
        print "graph_vlabel bytes\n";
        print "graph_category disk\n";

        foreach (@sorted_users) {
            my $user = $_;
            my ($uid, $name) = (getpwnam($user))[2,6];
            $user =~ s/-/_/g;
            $name = (split(/,/, $name))[0];
            printf "%s.label %s\n", $user, $name;
            if ( $users{$user}[1] > 0 ) {
                printf "%s.warning %s\n", $user, $users{$user}[1];
            }
            if ( $users{$user}[2] > 0 ) {
                printf "%s.critical %s\n", $user, $users{$user}[2];
            }
            printf "%s.cdef %s,1024,*\n", $user, $user;
            if (($uid < 500)) {
                printf "%s.graph no\n", $user;
            }
        }
} else {
        foreach (@sorted_users) {
            my $user = $_;
            my $esc = $user;
            $esc =~ s/-/_/g;
            printf "%s.value %s\n", $esc, $users{$user}[0];
        }
}

# vim: set ts=4 sw=4: