ViewVC Help
View File | Revision Log | Show Annotations | Revision Graph | Root Listing
root/i-scream/projects/cms/source/reports/rrdgraphing/graph.pl
Revision: 1.5
Committed: Tue May 21 15:01:43 2002 UTC (22 years, 6 months ago) by tdb
Content type: text/plain
Branch: MAIN
Changes since 1.4: +104 -34 lines
Log Message:
Changed to have a shared config file. Also parses command line options to
make it more verbose or quiet.

File Contents

# Content
1 #!/usr/bin/perl -w
2
3 #
4 # i-scream central monitoring system
5 # Copyright (C) 2000-2002 i-scream
6 #
7 # This program is free software; you can redistribute it and/or
8 # modify it under the terms of the GNU General Public License
9 # as published by the Free Software Foundation; either version 2
10 # of the License, or (at your option) any later version.
11 #
12 # This program is distributed in the hope that it will be useful,
13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 # GNU General Public License for more details.
16 #
17 # You should have received a copy of the GNU General Public License
18 # along with this program; if not, write to the Free Software
19 # Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
20 #
21
22 # -----------------------------------------------------------
23 # i-scream graph generation script
24 # http://www.i-scream.org.uk
25 #
26 # Generates graphs from rrd databases for i-scream data.
27 #
28 # $Author: tdb $
29 # $Id: graph.pl,v 1.4 2002/05/21 11:37:56 tdb Exp $
30 #------------------------------------------------------------
31
32 ## TODO
33 # possibly make more configurable?
34 # -- allow configurable periods of graphs
35 # -- comments, types, etc
36
37 my($version) = '$Id$';
38
39 $| = 1;
40
41 use strict;
42 use Getopt::Std;
43 use RRDs;
44
45 # define variables that will be read from the config
46 # nb. keep this insync with the config file!
47 use vars qw{
48 $imgdir $rrddir
49 $maxrrdage $maximgage $deleterrds $deleteimgs
50 $hex_slash $hex_underscore
51 $rrdstep $retry_wait
52 $verbose $quiet
53 };
54
55 # default locate of the config file
56 my($configfile) = "rrdgraphing.conf";
57
58 # check for command line arguments
59 my(%opts);
60 my($ret) = getopts('hvqVc:', \%opts);
61
62 # if invalid argument given, $ret will not be 1
63 &usage() if $ret != 1;
64
65 # first process the arguments which might mean we exit now
66
67 # -h is usage
68 if($opts{h}) {
69 &usage();
70 }
71 # -V is version
72 if($opts{V}) {
73 print "graph.pl version: $version\n";
74 exit(1);
75 }
76
77 # Then try getting the config
78
79 # -c specifies the config file location
80 if($opts{c}) {
81 $configfile = $opts{c};
82 }
83 # suck in the config
84 &log("reading config from $configfile\n");
85 do $configfile;
86
87 # Then any options we might want to override the config with
88
89 # -v is verbose
90 if($opts{v}) {
91 $verbose = $opts{v};
92 }
93 # -q is verbose
94 if($opts{q}) {
95 $quiet = $opts{q};
96 # if we're meant to be quiet, we can hardly be verbose!
97 $verbose = 0;
98 }
99
100
101 # Read the contents of the base directory
102 # and pull out the list of subdirectories (except . and .. :)
103 opendir(DIR, $rrddir);
104 my(@rrddirlist) = grep { -d "$rrddir/$_" && !/^\.$/ && !/^\.\.$/ } readdir(DIR);
105 closedir DIR;
106
107 # look through each directoty, as they might
108 # contain rrds for a particular machine
109 foreach my $machine (@rrddirlist) {
110 # Read the contents of the directory
111 opendir(DIR, "$rrddir/$machine");
112 my(@rrdlist) = grep { /\.rrd$/ && -f "$rrddir/$machine/$_" } readdir(DIR);
113 closedir DIR;
114
115 # See what rrd we have, and generate the graphs accordingly
116 foreach my $rrd (@rrdlist) {
117 chomp $rrd;
118 # stat the file
119 my($dev,$ino,$mode,$nlink,$uid,$gid,$rdev,$size,$atime,$mtime,
120 $ctime,$blksize,$blocks) = stat("$rrddir/$machine/$rrd");
121 # check if it's old enough to be deleted
122 if((time - $mtime) > $maxrrdage) {
123 # do we delete the rrd, or just ignore it?
124 if($deleterrds) {
125 # if so, delete it
126 unlink("$rrddir/$machine/$rrd");
127 &log("deleted old rrd $rrddir/$machine/$rrd\n");
128 }
129 else {
130 &log("ignored old rrd $rrddir/$machine/$rrd\n");
131 }
132 # no more processing required for this rrd
133 next;
134 }
135 if($rrd =~ /^(cpu)\.rrd$/) {
136 my(@data);
137 my(@rawdata);
138 push @data, "LINE2:$1:idle:idle#00FF00:idle cpu";
139 push @data, "LINE2:$1:user:user#0000FF:user cpu";
140 push @data, "LINE2:$1:kernel:kernel#00FFFF:kernel cpu";
141 push @data, "LINE2:$1:swap:swap#FF00FF:swap cpu";
142 push @data, "LINE2:$1:iowait:iowait#FF0000:iowait cpu";
143 push @rawdata, "--upper-limit=100";
144 &makegraph($machine, $1, "CPU Usage for $machine", \@data, \@rawdata);
145 }
146 if($rrd =~ /^(mem)\.rrd$/) {
147 my(@data);
148 my(@rawdata);
149 # we don't actually want to display free memory,
150 # although we need it to do inuse...
151 push @data, "NONE:$1:free:free#CCCCFF:free memory";
152 push @data, "LINE2:$1:total:total#0000FF:total memory";
153 # calculate inuse
154 push @rawdata, "CDEF:inuse=total,free,-";
155 # and add it to the graph
156 push @rawdata, "AREA:inuse#CCCCFF:memory in use";
157 push @rawdata, "--base=1024";
158 &makegraph($machine, $1, "Memory Usage for $machine", \@data, \@rawdata);
159 }
160 if($rrd =~ /^(load)\.rrd$/) {
161 my(@data);
162 push @data, "LINE2:$1:load1:load1#CCCCFF:1 minute load average";
163 push @data, "LINE2:$1:load5:load5#7777FF:5 minute load average";
164 push @data, "LINE2:$1:load15:load15#0000FF:15 minute load average";
165 &makegraph($machine, $1, "Loads for $machine", \@data);
166 }
167 if($rrd =~ /^(proc)\.rrd$/) {
168 my(@data);
169 push @data, "LINE2:$1:cpu:cpu#00FF00:cpu processes";
170 push @data, "LINE2:$1:sleeping:sleeping#0000FF:sleeping processes";
171 push @data, "LINE2:$1:stopped:stopped#00FFFF:stopped processes";
172 push @data, "LINE2:$1:total:total#FF00FF:total processes";
173 push @data, "LINE2:$1:zombie:zombie#FF0000:zombie processes";
174 &makegraph($machine, $1, "Processes on $machine", \@data);
175 }
176 if($rrd =~ /^(swap)\.rrd$/) {
177 my(@data);
178 my(@rawdata);
179 # we don't actually want to display free swap,
180 # although we need it to do inuse...
181 push @data, "NONE:$1:free:free#CCCCFF:free swap";
182 push @data, "LINE2:$1:total:total#0000FF:total swap";
183 # calculate inuse
184 push @rawdata, "CDEF:inuse=total,free,-";
185 # and add it to the graph
186 push @rawdata, "AREA:inuse#CCCCFF:swap in use";
187 push @rawdata, "--base=1024";
188 &makegraph($machine, $1, "Swap Usage for $machine", \@data, \@rawdata);
189 }
190 if($rrd =~ /^(users)\.rrd$/) {
191 my(@data);
192 push @data, "AREA:$1:count:count#CCCCFF:user count";
193 &makegraph($machine, $1, "User Count for $machine", \@data);
194 }
195 if($rrd =~ /^(disk)-(\S+).rrd$/) {
196 my(@data);
197 my(@rawdata);
198 push @data, "LINE2:$1-$2:kbytes:kbytes#0000FF:total size";
199 push @data, "AREA:$1-$2:used:used#CCCCFF:used";
200 push @rawdata, "--base=1024";
201 my($type) = $1;
202 my($name) = $2;
203 my($nicename) = $2;
204 $nicename =~ s/$hex_slash/\//g;
205 $nicename =~ s/$hex_underscore/_/g;
206 &makegraph($machine, "$type-$name", "Disk Usage for $machine on $nicename", \@data, \@rawdata);
207 }
208 # probably a queue with a name like this :)
209 if($rrd =~ /^(\d+)_0\.rrd$/) {
210 my(@data);
211 my(@rawdata);
212 my($baserrd) = $1;
213 my($i) = 0;
214 while( -f "$rrddir/$machine/$baserrd\_$i.rrd" ) {
215 push @data, "LINE2:$baserrd\_$i:size:size$i" . &get_colour($i) . ":queue$i size ";
216 ++$i;
217 }
218 push @data, "LINE2:$baserrd\_0:total:total#FF0000:packets/sec - currently";
219 push @rawdata, "GPRINT:total:LAST:%lf %spackets/sec";
220 my($comment);
221 if(-f "$rrddir/$machine/$baserrd.def") {
222 open(DEF, "$rrddir/$machine/$baserrd.def");
223 $comment = <DEF>;
224 chomp $comment if defined $comment;
225 }
226 $comment = "unknown queue" if not defined $comment;
227 &makegraph($machine, $baserrd, $comment, \@data, \@rawdata);
228 }
229 }
230 # have a last check, maybe we can remove the directory now?
231 # (only if we're deleting stuff)
232 if($deleterrds) {
233 # Read the contents of the directory
234 opendir(DIR, "$rrddir/$machine");
235 my(@dirlist) = grep { !/^\.$/ && !/^\.\.$/ } readdir(DIR);
236 closedir DIR;
237 if($#dirlist == -1) {
238 rmdir "$rrddir/$machine";
239 &log("deleting empty rrd directory $rrddir/$machine\n");
240 }
241 }
242 }
243
244 if($deleteimgs) {
245 # Read the contents of the graphs directory
246 # and pull out the list of subdirectories (except . and .. :)
247 opendir(DIR, $imgdir);
248 my(@imgdirlist) = grep { -d "$imgdir/$_" && !/^\.$/ && !/^\.\.$/ } readdir(DIR);
249 closedir DIR;
250
251 # look through each directoty, as they might
252 # contain images for a particular machine
253 foreach my $machine (@imgdirlist) {
254 # Read the contents of the directory
255 opendir(DIR, "$imgdir/$machine");
256 my(@imglist) = grep { /\.png$/ && -f "$imgdir/$machine/$_" } readdir(DIR);
257 closedir DIR;
258
259 # See what rrd we have, and generate the graphs accordingly
260 foreach my $img (@imglist) {
261 chomp $img;
262 # stat the img
263 my($dev,$ino,$mode,$nlink,$uid,$gid,$rdev,$size,$atime,$mtime,
264 $ctime,$blksize,$blocks) = stat("$imgdir/$machine/$img");
265 # check if it's old enough to be deleted
266 if((time - $mtime) > $maximgage) {
267 # if so, delete it
268 unlink("$imgdir/$machine/$img");
269 &log("deleted old image $imgdir/$machine/$img\n");
270 }
271 }
272 # have a last check, maybe we can remove the directory now?
273 # Read the contents of the directory
274 opendir(DIR, "$imgdir/$machine");
275 my(@dirlist) = grep { !/^\.$/ && !/^\.\.$/ } readdir(DIR);
276 closedir DIR;
277 if($#dirlist == -1) {
278 rmdir "$imgdir/$machine";
279 &log("deleted empty image directory $imgdir/$machine\n");
280 }
281 }
282 }
283
284 exit(0);
285
286
287 #
288 # subroutine to make some graphs
289 #
290 # $machine = name of the machine
291 # (eg. kernow.ukc.ac.uk)
292 # $type = the type of graph for the machine
293 # (eg. cpu)
294 # $title = the title for the graph
295 # (eg. kernow CPU usage)
296 # $dataref = a reference to an array containing information for the graph
297 # elements of format: "gtype:rrdname:dsname:name#colour:comment with spaces"
298 # (if gtype is "NONE" only a DEF of 'name' will be defined, no line will be plotted)
299 # $rawcmdref = a reference to an array containing raw rrd commands
300 # elements a single command each, no spaces
301 #
302
303 sub makegraph() {
304 my($machine, $type, $title, $dataref, $rawcmdref) = @_;
305 # pass in these arrays by reference
306 my(@data) = @$dataref if defined $dataref;
307 my(@rawcmd) = @$rawcmdref if defined $rawcmdref;
308 # check if directory exists for images
309 if(! -d "$imgdir/$machine") {
310 # not sure on this umask, but it seems to work?
311 mkdir "$imgdir/$machine", 0777;
312 &log("created directory $imgdir/$machine\n");
313 }
314 my(@rrdcmd);
315 foreach my $dataitem (@data) {
316 # dataitem should be: "gtype:rrdname:dsname:name#colour:comment with spaces"
317 # (if gtype is "NONE" only a DEF of 'name' will be defined, no line will be plotted)
318 if($dataitem =~ /^(\S+):(\S+):(\S+):(\S+)#(.{6}):(.*)$/) {
319 push @rrdcmd, "DEF:$4=$rrddir/$machine/$2.rrd:$3:AVERAGE";
320 if($1 ne "NONE") {
321 push @rrdcmd, "$1:$4#$5:$6";
322 }
323 }
324 }
325 push @rrdcmd, "--title=$title";
326 push @rrdcmd, "--imgformat=PNG";
327 push @rrdcmd, "--lower-limit=0";
328 # not entirely convinced this is good...
329 push @rrdcmd, "--alt-autoscale-max";
330 # add any further raw commands
331 push @rrdcmd, @rawcmd;
332 RRDs::graph ("$imgdir/$machine/$type-3h.png", "--start=-10800", @rrdcmd);
333 my($err_3h) = RRDs::error;
334 &log("created $imgdir/$machine/$type-3h.png\n") unless $err_3h;
335 &error("Error generating 3h graph for $machine/$type: $err_3h\n") if $err_3h;
336 RRDs::graph ("$imgdir/$machine/$type-1d.png", "--start=-86400", @rrdcmd);
337 my($err_1d) = RRDs::error;
338 &log("created $imgdir/$machine/$type-1d.png\n") unless $err_1d;
339 &error("Error generating 1d graph for $machine/$type: $err_1d\n") if $err_1d;
340 RRDs::graph ("$imgdir/$machine/$type-1w.png", "--start=-604800", @rrdcmd);
341 my($err_1w) = RRDs::error;
342 &log("created $imgdir/$machine/$type-1w.png\n") unless $err_1w;
343 &error("Error generating 1w graph for $machine/$type: $err_1w\n") if $err_1w;
344 RRDs::graph ("$imgdir/$machine/$type-1m.png", "--start=-2678400", @rrdcmd);
345 my($err_1m) = RRDs::error;
346 &log("created $imgdir/$machine/$type-1m.png\n") unless $err_1m;
347 &error("Error generating 1m graph for $machine/$type: $err_1m\n") if $err_1m;
348 RRDs::graph ("$imgdir/$machine/$type-1y.png", "--start=-31536000", @rrdcmd);
349 my($err_1y) = RRDs::error;
350 &log("created $imgdir/$machine/$type-1y.png\n") unless $err_1y;
351 &error("Error generating 1y graph for $machine/$type: $err_1y\n") if $err_1y;
352 return;
353 }
354
355 # hacky subroutine to return a colour
356 # could be done much better somehow :/
357 sub get_colour {
358 my($col) = @_;
359 if($col == 0) {
360 return "#0000FF";
361 }
362 elsif($col == 1) {
363 return "#00FF00";
364 }
365 elsif($col == 2) {
366 return "#FF00FF";
367 }
368 elsif($col == 3) {
369 return "#FFFF00";
370 }
371 elsif($col == 4) {
372 return "#00FFFF";
373 }
374 else {
375 return "#000066";
376 }
377 }
378
379 # prints out usage information then exits
380 sub usage() {
381 print "Usage: graph.pl [options]\n";
382 print "Options\n";
383 print " -c config Specifies the configuration file\n";
384 print " default: rrdgraphing.conf\n";
385 print " -v Be verbose about what's happening\n";
386 print " -q Be quiet, even supress errors\n";
387 print " -V Print version number\n";
388 print " -h Prints this help page\n";
389 exit(1);
390 }
391
392 # prints a log message if verbose is turned on
393 sub log() {
394 my($msg) = @_;
395 print $msg if $verbose;
396 }
397
398 # prints an error message unless quiet is turned on
399 sub error() {
400 my($msg) = @_;
401 print STDERR $msg unless $quiet;
402 }