From 95c70c919837f588389f525fe2cf4aba479e9f91 Mon Sep 17 00:00:00 2001 From: Pim van den Berg Date: Thu, 31 Dec 2009 11:42:56 +0100 Subject: rewrite of type classes A constructor is added to the Type_Default class. The constructor will parse GET values (such as host, plugin, pinstance, type, tinstance, seconds), create an array of all needed rrd files to generate a graph and substract identifiers from these rrd files. Because of the constructor (and related functions) it is not needed to define an array of tinstances to be grouped and shown in one graph. Also $obj->args don't have to be defined per plugin. This will result in smaller plugin files. The type classes are based on the fact that a plugin has multiple type instances OR multiple rrd data sources. This is called the source and is retrieved by rrd_get_sources in each rrd_gen_graph function. Also variables in function rrd_gen_graph have been renamed to better ones. --- type/Default.class.php | 142 ++++++++++++++++++++++++++++-------------- type/GenericIO.class.php | 51 ++++++++------- type/GenericStacked.class.php | 59 +++++++----------- 3 files changed, 147 insertions(+), 105 deletions(-) (limited to 'type') diff --git a/type/Default.class.php b/type/Default.class.php index 6b1ab5c..0c992e2 100644 --- a/type/Default.class.php +++ b/type/Default.class.php @@ -5,7 +5,8 @@ class Type_Default { var $datadir; var $args; - var $data_sources; + var $seconds; + var $data_sources = array('value'); var $order; var $ds_names; var $colors; @@ -16,6 +17,29 @@ class Type_Default { var $width; var $heigth; + var $files; + var $tinstances; + var $identifiers; + + function __construct($datadir) { + $this->datadir = $datadir; + $this->parse_get(); + $this->rrd_files(); + $this->identifiers = $this->file2identifier($this->files); + } + + # parse $_GET values + function parse_get() { + $this->args = array( + 'host' => $_GET['h'], + 'plugin' => $_GET['p'], + 'pinstance' => $_GET['pi'], + 'type' => $_GET['t'], + 'tinstance' => $_GET['ti'], + ); + $this->seconds = $_GET['s']; + } + function validate_color($color) { if (!preg_match('/^[0-9a-f]{6}$/', $color)) return '000000'; @@ -46,26 +70,41 @@ class Type_Default { return $c[r].$c[g].$c[b]; } - function identifier($host, $plugin, $pinst, $type, $tinst) { - $identifier = sprintf('%s/%s%s%s/%s%s%s', $host, - $plugin, strlen($pinst) ? '-' : '', $pinst, - $type, strlen($tinst) ? '-' : '', $tinst); + function rrd_files() { + $files = $this->get_filenames(); - if (is_file($this->datadir.'/'.$identifier.'.rrd')) - return $identifier; - else - return FALSE; + foreach($files as $filename) { + preg_match("#^$this->datadir/{$this->args['host']}/[\w\d]+-?([\w\d-]+)?/[\w\d]+-?([\w\d-]+)?\.rrd#", $filename, $matches); + + $this->tinstances[] = $matches[2]; + $this->files[$matches[2]] = $filename; + } + + sort($this->tinstances); + ksort($this->files); } - function get_filename($tinstance=NULL) { - if (!is_array($this->args['tinstance']) && $tinstance == NULL) - $tinstance = $this->args['tinstance']; + function get_filenames() { + $identifier = sprintf('%s/%s%s%s/%s%s%s', $this->args['host'], + $this->args['plugin'], strlen($this->args['pinstance']) ? '-' : '', $this->args['pinstance'], + $this->args['type'], strlen($this->args['tinstance']) ? '-' : '', $this->args['tinstance']); + + $wildcard = strlen($this->args['tinstance']) ? '' : '*'; - $identifier = $this->identifier($this->args['host'], - $this->args['plugin'], $this->args['pinstance'], - $this->args['type'], $tinstance); + $files = glob($this->datadir .'/'. $identifier . $wildcard . '.rrd'); - return $this->datadir.'/'.$identifier.'.rrd'; + return $files; + } + + function file2identifier($files) { + foreach($files as $key => $file) { + if (is_file($file)) { + $files[$key] = preg_replace("#^$this->datadir/#", '', $files[$key]); + $files[$key] = preg_replace('#\.rrd$#', '', $files[$key]); + } + } + + return $files; } function rrd_graph($debug=false) { @@ -96,48 +135,57 @@ class Type_Default { return $rrdgraph; } - function rrd_gen_graph() { - $filename = $this->get_filename(); + function rrd_get_sources() { + # is the source spread over multiple files? + if (is_array($this->files) && count($this->files)>1) { + # and must it be ordered? + if (is_array($this->order)) { + $this->tinstances = array_intersect($this->order, $this->tinstances); + } + # use tinstances as sources + $sources = $this->tinstances; + } + # or one file with multiple data_sources + else { + # use data_sources as sources + $sources = $this->data_sources; + } + return $sources; + } + function rrd_gen_graph() { $rrdgraph = $this->rrd_options(); - if (is_array($this->args['tinstance'])) - $array = is_array($this->order) ? $this->order : $this->args['tinstance']; - else - $array = $this->data_sources; + $sources = $this->rrd_get_sources(); $i=0; - foreach ($array as $value) { - if (is_array($this->args['tinstance'])) { - $filename = $this->get_filename($value); - $ds = $this->data_sources[0]; - } else { - $filename = $this->get_filename(); - $ds = $value; + foreach ($this->tinstances as $tinstance) { + foreach ($this->data_sources as $ds) { + $rrdgraph[] = sprintf('DEF:min_%s=%s:%s:MIN', $sources[$i], $this->files[$tinstance], $ds); + $rrdgraph[] = sprintf('DEF:avg_%s=%s:%s:AVERAGE', $sources[$i], $this->files[$tinstance], $ds); + $rrdgraph[] = sprintf('DEF:max_%s=%s:%s:MAX', $sources[$i], $this->files[$tinstance], $ds); + $i++; } - $rrdgraph[] = sprintf('DEF:min%s=%s:%s:MIN', $i, $filename, $ds); - $rrdgraph[] = sprintf('DEF:avg%s=%s:%s:AVERAGE', $i, $filename, $ds); - $rrdgraph[] = sprintf('DEF:max%s=%s:%s:MAX', $i, $filename, $ds); - $i++; } - if (!is_array($this->args['tinstance'])) { - $rrdgraph[] = sprintf('AREA:max0#%s', $this->get_faded_color($this->colors[$this->data_sources[0]])); - $rrdgraph[] = sprintf('AREA:min0#%s', 'ffffff'); + if(count($this->files)<=1) { + foreach ($sources as $source) { + $rrdgraph[] = sprintf('AREA:max_%s#%s', $source, $this->get_faded_color($this->colors[$source])); + $rrdgraph[] = sprintf('AREA:min_%s#%s', $source, 'ffffff'); + break; # only 1 area to draw + } } - $i=0; - foreach ($array as $value) { - $dsname = $this->ds_names[$value] != '' ? $this->ds_names[$value] : $value; - $color = is_array($this->colors) ? $this->colors[$value]: $this->colors; - $rrdgraph[] = sprintf('LINE1:avg%d#%s:\'%s\'', $i, $this->validate_color($color), $dsname); - $rrdgraph[] = sprintf('GPRINT:min%d:MIN:\'%s Min,\'', $i, $this->rrd_format); - $rrdgraph[] = sprintf('GPRINT:avg%d:AVERAGE:\'%s Avg,\'', $i, $this->rrd_format); - $rrdgraph[] = sprintf('GPRINT:max%d:MAX:\'%s Max,\'', $i, $this->rrd_format); - $rrdgraph[] = sprintf('GPRINT:avg%d:LAST:\'%s Last\\l\'', $i, $this->rrd_format); - $i++; + foreach ($sources as $source) { + $dsname = $this->ds_names[$source] != '' ? $this->ds_names[$source] : $source; + $color = is_array($this->colors) ? $this->colors[$source]: $this->colors; + $rrdgraph[] = sprintf('LINE1:avg_%s#%s:\'%s\'', $source, $this->validate_color($color), $dsname); + $rrdgraph[] = sprintf('GPRINT:min_%s:MIN:\'%s Min,\'', $source, $this->rrd_format); + $rrdgraph[] = sprintf('GPRINT:avg_%s:AVERAGE:\'%s Avg,\'', $source, $this->rrd_format); + $rrdgraph[] = sprintf('GPRINT:max_%s:MAX:\'%s Max,\'', $source, $this->rrd_format); + $rrdgraph[] = sprintf('GPRINT:avg_%s:LAST:\'%s Last\\l\'', $source, $this->rrd_format); } - + return $rrdgraph; } } diff --git a/type/GenericIO.class.php b/type/GenericIO.class.php index e91cb16..ee04316 100644 --- a/type/GenericIO.class.php +++ b/type/GenericIO.class.php @@ -5,45 +5,54 @@ require_once 'Default.class.php'; class Type_GenericIO extends Type_Default { function rrd_gen_graph() { - $filename = $this->get_filename(); - $rrdgraph = $this->rrd_options(); + $sources = $this->rrd_get_sources(); + if ($this->scale) $raw = '_raw'; - foreach($this->data_sources as $ds) { - $rrdgraph[] = sprintf('DEF:min_%s%s=%s:%s:MIN', $ds, $raw, $filename, $ds); - $rrdgraph[] = sprintf('DEF:avg_%s%s=%s:%s:AVERAGE', $ds, $raw, $filename, $ds); - $rrdgraph[] = sprintf('DEF:max_%s%s=%s:%s:MAX', $ds, $raw, $filename, $ds); + $i=0; + foreach ($this->tinstances as $tinstance) { + foreach ($this->data_sources as $ds) { + $rrdgraph[] = sprintf('DEF:min_%s%s=%s:%s:MIN', $sources[$i], $raw, $this->files[$tinstance], $ds); + $rrdgraph[] = sprintf('DEF:avg_%s%s=%s:%s:AVERAGE', $sources[$i], $raw, $this->files[$tinstance], $ds); + $rrdgraph[] = sprintf('DEF:max_%s%s=%s:%s:MAX', $sources[$i], $raw, $this->files[$tinstance], $ds); + $i++; + } } if ($this->scale) { - foreach($this->data_sources as $ds) { - $rrdgraph[] = sprintf('CDEF:min_%s=min_%s_raw,%s,*', $ds, $ds, $this->scale); - $rrdgraph[] = sprintf('CDEF:avg_%s=avg_%s_raw,%s,*', $ds, $ds, $this->scale); - $rrdgraph[] = sprintf('CDEF:max_%s=max_%s_raw,%s,*', $ds, $ds, $this->scale); + $i=0; + foreach ($this->tinstances as $tinstance) { + foreach ($this->data_sources as $ds) { + $rrdgraph[] = sprintf('CDEF:min_%s=min_%1$s_raw,%s,*', $sources[$i], $this->scale); + $rrdgraph[] = sprintf('CDEF:avg_%s=avg_%1$s_raw,%s,*', $sources[$i], $this->scale); + $rrdgraph[] = sprintf('CDEF:max_%s=max_%1$s_raw,%s,*', $sources[$i], $this->scale); + $i++; + } } } $rrdgraph[] = sprintf('CDEF:overlap=avg_%s,avg_%s,LT,avg_%1$s,avg_%2$s,IF', - $this->data_sources[0], $this->data_sources[1]); + $sources[0], $sources[1]); - foreach($this->data_sources as $ds) { - $rrdgraph[] = sprintf('AREA:avg_%s#%s', $ds, $this->get_faded_color($this->colors[$ds])); + foreach($sources as $source) { + $rrdgraph[] = sprintf('AREA:avg_%s#%s', $source, $this->get_faded_color($this->colors[$source])); } $rrdgraph[] = sprintf('AREA:overlap#%s', $this->get_faded_color( - $this->get_faded_color($this->colors[$this->data_sources[0]]), - $this->get_faded_color($this->colors[$this->data_sources[1]]) + $this->get_faded_color($this->colors[$sources[0]]), + $this->get_faded_color($this->colors[$sources[1]]) ) ); - foreach($this->data_sources as $ds) { - $rrdgraph[] = sprintf('LINE1:avg_%s#%s:\'%s\'', $ds, $this->colors[$ds], $this->ds_names[$ds]); - $rrdgraph[] = sprintf('GPRINT:min_%s:MIN:\'%s Min,\'', $ds, $this->rrd_format); - $rrdgraph[] = sprintf('GPRINT:avg_%s:AVERAGE:\'%s Avg,\'', $ds, $this->rrd_format); - $rrdgraph[] = sprintf('GPRINT:max_%s:MAX:\'%s Max,\'', $ds, $this->rrd_format); - $rrdgraph[] = sprintf('GPRINT:avg_%s:LAST:\'%s Last\l\'', $ds, $this->rrd_format); + foreach($sources as $source) { + $dsname = $this->ds_names[$source] != '' ? $this->ds_names[$source] : $source; + $rrdgraph[] = sprintf('LINE1:avg_%s#%s:\'%s\'', $source, $this->colors[$source], $dsname); + $rrdgraph[] = sprintf('GPRINT:min_%s:MIN:\'%s Min,\'', $source, $this->rrd_format); + $rrdgraph[] = sprintf('GPRINT:avg_%s:AVERAGE:\'%s Avg,\'', $source, $this->rrd_format); + $rrdgraph[] = sprintf('GPRINT:max_%s:MAX:\'%s Max,\'', $source, $this->rrd_format); + $rrdgraph[] = sprintf('GPRINT:avg_%s:LAST:\'%s Last\l\'', $source, $this->rrd_format); } return $rrdgraph; diff --git a/type/GenericStacked.class.php b/type/GenericStacked.class.php index f024336..7595385 100644 --- a/type/GenericStacked.class.php +++ b/type/GenericStacked.class.php @@ -3,56 +3,41 @@ require_once 'Default.class.php'; class Type_GenericStacked extends Type_Default { - + function rrd_gen_graph() { $rrdgraph = $this->rrd_options(); - if (is_array($this->args['tinstance'])) - if (is_array($this->order)) - $array = array_intersect($this->order, $this->args['tinstance']); - else - $array = $this->args['tinstance']; - else - $array = $this->data_sources; + $sources = $this->rrd_get_sources(); $i=0; - foreach ($array as $value) { - if (is_array($this->args['tinstance'])) { - $filename = $this->get_filename($value); - $ds = $this->data_sources[0]; - } else { - $filename = $this->get_filename(); - $ds = $value; + foreach ($this->tinstances as $tinstance) { + foreach ($this->data_sources as $ds) { + $rrdgraph[] = sprintf('DEF:min_%s=%s:%s:MIN', $sources[$i], $this->files[$tinstance], $ds); + $rrdgraph[] = sprintf('DEF:avg_%s=%s:%s:AVERAGE', $sources[$i], $this->files[$tinstance], $ds); + $rrdgraph[] = sprintf('DEF:max_%s=%s:%s:MAX', $sources[$i], $this->files[$tinstance], $ds); + $i++; } - $rrdgraph[] = sprintf('DEF:min%s=%s:%s:MIN', $i, $filename, $ds); - $rrdgraph[] = sprintf('DEF:avg%s=%s:%s:AVERAGE', $i, $filename, $ds); - $rrdgraph[] = sprintf('DEF:max%s=%s:%s:MAX', $i, $filename, $ds); - $i++; } - for ($i=count($array)-1 ; $i>=0 ; $i--) { - if ($i == (count($array)-1)) - $rrdgraph[] = sprintf('CDEF:cdef%d=avg%d', $i, $i); + for ($i=count($sources)-1 ; $i>=0 ; $i--) { + if ($i == (count($sources)-1)) + $rrdgraph[] = sprintf('CDEF:area_%s=avg_%1$s', $sources[$i]); else - $rrdgraph[] = sprintf('CDEF:cdef%d=cdef%d,avg%d,+', $i, $i+1, $i); + $rrdgraph[] = sprintf('CDEF:area_%s=area_%s,avg_%1$s,+', $sources[$i], $sources[$i+1]); } - $i=0; - foreach ($array as $value) { - $color = $this->get_faded_color($this->colors[$value]); - $rrdgraph[] = sprintf('AREA:cdef%d#%s', $i, $color); - $i++; + foreach ($sources as $source) { + $color = $this->get_faded_color($this->colors[$source]); + $rrdgraph[] = sprintf('AREA:area_%s#%s', $source, $color); } - $i=0; - foreach ($array as $value) { - $dsname = $this->ds_names[$value] != '' ? $this->ds_names[$value] : $value; - $rrdgraph[] = sprintf('LINE1:cdef%d#%s:\'%s\'', $i, $this->validate_color($this->colors[$value]), $dsname); - $rrdgraph[] = sprintf('GPRINT:min%d:MIN:\'%s Min,\'', $i, $this->rrd_format); - $rrdgraph[] = sprintf('GPRINT:avg%d:AVERAGE:\'%s Avg,\'', $i, $this->rrd_format); - $rrdgraph[] = sprintf('GPRINT:max%d:MAX:\'%s Max,\'', $i, $this->rrd_format); - $rrdgraph[] = sprintf('GPRINT:avg%d:LAST:\'%s Last\\l\'', $i, $this->rrd_format); - $i++; + foreach ($sources as $source) { + $dsname = $this->ds_names[$source] != '' ? $this->ds_names[$source] : $source; + $rrdgraph[] = sprintf('LINE1:area_%s#%s:\'%s\'', $source, $this->validate_color($this->colors[$source]), $dsname); + $rrdgraph[] = sprintf('GPRINT:min_%s:MIN:\'%s Min,\'', $source, $this->rrd_format); + $rrdgraph[] = sprintf('GPRINT:avg_%s:AVERAGE:\'%s Avg,\'', $source, $this->rrd_format); + $rrdgraph[] = sprintf('GPRINT:max_%s:MAX:\'%s Max,\'', $source, $this->rrd_format); + $rrdgraph[] = sprintf('GPRINT:avg_%s:LAST:\'%s Last\\l\'', $source, $this->rrd_format); } return $rrdgraph; -- cgit v1.1