comment out debug lines for production use again
[php-utility-classes.git] / include / classes / rrdstat.php-class
index 575f7dff589f6b2f38579eaf0ccad6ca85916b12..68887290c9fbfc3e5c29c174f81b4f6fb9d5cebf 100644 (file)
@@ -2,7 +2,11 @@
 // ************ RRD status class **************
 class rrdstat {
 
-  var $rrd_file = 'sample.rrd';
+  var $rrd_file = null;
+
+  var $config_raw = null;
+  var $config_graph = null;
+  var $config_page = null;
 
   var $rrd_fields = array();
   var $rra_base = array();
@@ -15,19 +19,21 @@ class rrdstat {
     // ***** init RRD stat module *****
     $this->set_def($init_info);
 
-    if (!is_writeable($this->rrd_file)) {
-      if (!file_exists($this->rrd_file)) {
-        if (touch($this->rrd_file)) { $this->create(); }
-        else { trigger_error('RRD file can not be created', E_USER_WARNING); }
+    if (!is_null($this->rrd_file)) {
+      if (!is_writeable($this->rrd_file)) {
+        if (!file_exists($this->rrd_file)) {
+          if (touch($this->rrd_file)) { $this->create(); }
+          else { trigger_error('RRD file can not be created', E_USER_WARNING); }
+        }
+        else {
+          if (is_readable($this->rrd_file)) { $this->status = 'readonly'; }
+          else { trigger_error('RRD file is not readable', E_USER_WARNING); }
+        }
       }
       else {
-        if (is_readable($this->rrd_file)) { $this->status = 'readonly'; }
-        else { trigger_error('RRD file is not readable', E_USER_WARNING); }
+        $this->status = 'ok';
       }
     }
-    else {
-      $this->status = 'ok';
-    }
   }
 
   function set_def($init_info = null) {
@@ -81,6 +87,10 @@ class rrdstat {
     }
 
     $this->rra_add_max = isset($iinfo['rra_add_max'])?$iinfo['rra_add_max']:true;
+
+    if (isset($iinfo['graph'])) { $this->config_graph = $iinfo['graph']; }
+    if (isset($iinfo['page'])) { $this->config_page = $iinfo['page']; }
+    $this->config_raw = $iinfo;
   }
 
   function create() {
@@ -137,38 +147,148 @@ class rrdstat {
   return ($return_var == 0);
   }
 
-  function graph($filename, $timeframe, $extra = null) {
-    // create an RRD graph
-    static $gColors;
+  function fetch($cf = 'AVERAGE', $resolution = null, $start = null, $end = null) {
+    // fetch data from a RRD
+    if (!in_array($this->status, array('ok','readonly'))) { trigger_error('Error: rrd status is '.$this->status, E_USER_WARNING); return false; }
+
+    if (!in_array($cf, array('AVERAGE','MIN','MAX','LAST'))) { $cf = 'AVERAGE'; }
+    if (!is_numeric($resolution)) { $resolution = $this->rrd_step; }
+    if (!is_numeric($end)) { $end = $this->last_update(); }
+    elseif ($end < 0) { $end += $this->last_update(); }
+    $end = intval($end/$resolution)*$resolution;
+    if (!is_numeric($start)) { $start = $end; }
+    elseif ($start < 0) { $start += $end; }
+    $start = intval($start/$resolution)*$resolution;
+
+    $fetch_cmd = 'rrdtool fetch '.$this->rrd_file.' '.$cf.' --resolution '.$resolution.' --start '.$start.' --end '.$end;
+    $return = `$fetch_cmd 2>&1`;
+
+    if (strpos($return, 'ERROR') !== false) {
+      trigger_error('rrd fetch error: '.$return, E_USER_WARNING);
+      $fresult = false;
+    }
+    else {
+      $fresult = array();
+      $rows = explode("\n", $return);
+      $fields = preg_split('/\s+/', array_shift($rows));
+      if (array_shift($fields) == 'timestamp') {
+        $fresult[0] = $fields;
+        foreach ($rows as $row) {
+          if (strlen(trim($row))) {
+            $rvals = preg_split('/\s+/', $row);
+            $rtime = array_shift($rvals);
+            $rv_array = array();
+            foreach ($rvals as $key=>$rval) {
+              $rv_array[$fields[$key]] = ($rval=='nan')?null:floatval($rval);
+            }
+            $fresult[$rtime] = $rv_array;
+          }
+        }
+      }
+    }
+  return $fresult;
+  }
 
+  function graph($timeframe = 'day', $special = null, $extra = null) {
+    // create a RRD graph
+    static $gColors;
     if (!isset($gColors)) {
       $gColors = array('#00CC00','#0000FF','#000000','#FF0000','#00FF00','#FFFF00','#FF00FF','#00FFFF','#808080','#800000','#008000','#000080','#808000','#800080','#008080','#C0C0C0');
     }
 
-    if (is_null($filename) || !strlen($filename)) { $filename = str_replace('.rrd', '-%tf.png', $this->rrd_file); }
-    $fname = str_replace('%tf', $timeframe, $filename);
-    if (substr($fname, -4) != '.png') { $fname .= '.png'; }
+    if (!in_array($this->status, array('ok','readonly'))) { trigger_error('Error: rrd status is '.$this->status, E_USER_WARNING); return false; }
 
-    // [-s|--start time] [-e|--end time] [-S|--step seconds]
-    if ($timeframe == 'day') { $timeinfo = '--start -86400'; }
-    elseif ($timeframe == 'week') { $timeinfo = '--start -604800'; }
-    elseif ($timeframe == 'month') { $timeinfo = '--start -2678400'; }
-    elseif ($timeframe == 'year') { $timeinfo = '--start -31622400'; }
-    else { $timeinfo = '--start -'.$this->rrd_step*500; }
+    // assemble configuration
+    $gconf = $this->config_graph;
+    if (!is_null($special) && is_array($this->config_raw['graph'][$special])) {
+      if (is_array($gconf)) { $gconf = array_merge($gconf, $this->config_raw['graph'][$special]); }
+      else { $gconf = $this->config_raw['graph'][$special]; }
+    }
+    if (is_array($extra)) {
+      if (is_array($gconf)) { $gconf = array_merge($gconf, $extra); }
+      else { $gconf = $extra; }
+    }
+
+    if (isset($gconf['format']) && ($gconf['format'] == 'SVG')) {
+      $format = $gconf['format']; $fmt_ext = '.svg';
+    }
+    elseif (isset($gconf['format']) && ($gconf['format'] == 'EPS')) {
+      $format = $gconf['format']; $fmt_ext = '.eps';
+    }
+    elseif (isset($gconf['format']) && ($gconf['format'] == 'PDF')) {
+      $format = $gconf['format']; $fmt_ext = '.pdf';
+    }
+    else {
+      $format = 'PNG'; $fmt_ext = '.png';
+    }
+
+    if (isset($gconf['filename'])) { $fname = $gconf['filename']; }
+    else { $fname = str_replace('.rrd', '-%t%f', $this->rrd_file); }
+    $fname = str_replace('%t', $timeframe, $fname);
+    $fname = str_replace('%f', $fmt_ext, $fname);
+    if (substr($fname, -strlen($fmt_ext)) != $fmt_ext) { $fname .= $fmt_ext; }
 
     $graphrows = array(); $gC = 0;
-    if (isset($extra['rows']) && count($extra['rows'])) {
-      foreach ($extra['rows'] as $erow) {
+    $gDefs = ''; $gGraphs = ''; $addSpecial = '';
+
+    if ($timeframe == 'day') {
+      $duration = isset($gconf['duration'])?$gconf['duration']:30*3600; // 30 hours
+      $slice = isset($gconf['slice'])?$gconf['slice']:300; // 5 minutes
+      // vertical lines at day borders
+      $addSpecial .= ' VRULE:'.strtotime(date('Y-m-d')).'#FF0000';
+      $addSpecial .= ' VRULE:'.strtotime(date('Y-m-d').' -1 day').'#FF0000';
+      if (!isset($gconf['grid_x'])) { $gconf['grid_x'] = 'HOUR:1:HOUR:6:HOUR:2:0:%-H'; }
+    }
+    elseif ($timeframe == 'week') {
+      $duration = isset($gconf['duration'])?$gconf['duration']:8*86400; // 8 days
+      $slice = isset($gconf['slice'])?$gconf['slice']:1800; // 30 minutes
+      // vertical lines at week borders
+      $addSpecial .= ' VRULE:'.strtotime(date('Y-m-d').' '.(-date('w')+1).' day').'#FF0000';
+      $addSpecial .= ' VRULE:'.strtotime(date('Y-m-d').' '.(-date('w')-6).' day').'#FF0000';
+    }
+    elseif ($timeframe == 'month') {
+      $duration = isset($gconf['duration'])?$gconf['duration']:36*86400; // 36 days
+      $slice = isset($gconf['slice'])?$gconf['slice']:7200; // 2 hours
+      // vertical lines at month borders
+      $addSpecial .= ' VRULE:'.strtotime(date('Y-m-01')).'#FF0000';
+      $addSpecial .= ' VRULE:'.strtotime(date('Y-m-01').' -1 month').'#FF0000';
+    }
+    elseif ($timeframe == 'year') {
+      $duration = isset($gconf['duration'])?$gconf['duration']:396*86400; // 365+31 days
+      $slice = isset($gconf['slice'])?$gconf['slice']:86400; // 1 day
+      // vertical lines at month borders
+      $addSpecial .= ' VRULE:'.strtotime(date('Y-01-01')).'#FF0000';
+      $addSpecial .= ' VRULE:'.strtotime(date('Y-01-01').' -1 year').'#FF0000';
+    }
+    else {
+      $duration = isset($gconf['duration'])?$gconf['duration']:$this->rrd_step*500; // 500 steps
+      $slice = isset($gconf['slice'])?$gconf['slice']:$this->rrd_step; // whatever our step is
+    }
+
+    if (isset($gconf['rows']) && count($gconf['rows'])) {
+      foreach ($gconf['rows'] as $erow) {
         if (isset($erow['name']) && strlen($erow['name'])) {
+          if (!isset($erow['scale']) && isset($gconf['scale'])) { $erow['scale'] = $gconf['scale']; }
           $grow = array();
           $grow['dType'] = isset($erow['dType'])?$erow['dType']:'DEF';
-          $grow['name'] = $erow['name'];
+          $grow['name'] = $erow['name'].(isset($erow['scale'])?'_tmp':'');
           $grow['dsname'] = isset($erow['dsname'])?$erow['dsname']:$erow['name'];
           $grow['cf'] = isset($erow['cf'])?$erow['cf']:'AVERAGE';
+          if (isset($erow['rpn_expr'])) { $grow['rpn_expr'] = $erow['rpn_expr']; }
+          if (isset($erow['scale'])) {
+            $graphrows[] = $grow;
+            $grow = array();
+            $grow['dType'] = 'CDEF';
+            $grow['name'] = $erow['name'];
+            $grow['rpn_expr'] = $erow['name'].'_tmp,'.$erow['scale'].',*';
+          }
           $grow['gType'] = isset($erow['gType'])?$erow['gType']:'LINE1';
           $grow['color'] = isset($erow['color'])?$erow['color']:$gColors[$gC++];
           if ($gC >= count($gColors)) { $gC = 0; }
-          if (isset($erow['legend'])) { $grow['legend'] = $erow['legend']; }
+          if (isset($erow['legend'])) {
+            $grow['legend'] = $erow['legend'];
+            if (!isset($gconf['show_legend'])) { $gconf['show_legend'] = true; }
+          }
           if (isset($erow['stack'])) { $grow['stack'] = ($erow['stack'] == true); }
           $graphrows[] = $grow;
         }
@@ -178,45 +298,91 @@ class rrdstat {
       foreach ($this->rrd_fields as $ds) {
         $grow = array();
         $grow['dType'] = 'DEF';
-        $grow['name'] = $ds['name'];
+        $grow['name'] = $ds['name'].(isset($gconf['scale'])?'_tmp':'');
         $grow['dsname'] = $ds['name'];
         $grow['cf'] = 'AVERAGE';
+        if (isset($gconf['scale'])) {
+          $graphrows[] = $grow;
+          $grow = array();
+          $grow['dType'] = 'CDEF';
+          $grow['name'] = $ds['name'];
+          $grow['rpn_expr'] = $ds['name'].'_tmp,'.$gconf['scale'].',*';
+        }
         $grow['gType'] = ($ds['name']=='ds0')?'AREA':'LINE1';
         $grow['color'] = $gColors[$gC++]; if ($gC >= count($gColors)) { $gC = 0; }
         $graphrows[] = $grow;
       }
     }
 
-    $gDefs = ''; $gGraphs = '';
+    $endtime = isset($gconf['time_end'])?$gconf['time_end']:(is_numeric($this->last_update())?$this->last_update():time());
+    $gOpts = ' --start '.($endtime-$duration).' --end '.$endtime.' --step '.$slice;
+    if (isset($gconf['label_top'])) { $gOpts .= ' --title '.$this->text_quote($gconf['label_top']); }
+    if (isset($gconf['label_y'])) { $gOpts .= ' --vertical-label '.$this->text_quote($gconf['label_y']); }
+    if (isset($gconf['width'])) { $gOpts .= ' --width '.$gconf['width']; }
+    if (isset($gconf['height'])) { $gOpts .= ' --height '.$gconf['height'];
+      if (($gconf['height'] <= 32) && isset($gconf['thumb']) && ($gconf['thumb'])) { $gOpts .= ' --only-graph'; }
+    }
+    if (!isset($gconf['show_legend']) || (!$gconf['show_legend'])) { $gOpts .= ' --no-legend'; }
+    if (isset($gconf['min_y'])) { $gOpts .= ' --lower-limit '.$gconf['min_y']; }
+    if (isset($gconf['max_y'])) { $gOpts .= ' --upper-limit '.$gconf['max_y']; }
+    if (isset($gconf['fix_scale_y']) && $gconf['fix_scale_y']) { $gOpts .= ' --rigid'; }
+    if (isset($gconf['grid_x'])) { $gOpts .= ' --x-grid '.$gconf['grid_x']; }
+    if (isset($gconf['grid_y'])) { $gOpts .= ' --y-grid '.$gconf['grid_y']; }
+    if (isset($gconf['units_exponent'])) { $gOpts .= ' --units-exponent '.$gconf['units_exponent']; }
+    if (isset($gconf['units_length'])) { $gOpts .= ' --units-length '.$gconf['units_length']; }
+    if (!isset($gconf['force_recreate']) || (!$gconf['force_recreate'])) { $gOpts .= ' --lazy'; }
+    if (isset($gconf['force_color']) && is_array($gconf['force_color'])) {
+      foreach ($gconf['force_color'] as $ctag=>$cval) { $gOpts .= ' --color '.$ctag.$cval; }
+    }
+    if (isset($gconf['force_font']) && is_array($gconf['force_font'])) {
+      foreach ($gconf['force_font'] as $ctag=>$cval) { $gOpts .= ' --font '.$ctag.$cval; }
+    }
+    if (isset($gconf['units_binary']) && $gconf['units_binary']) { $gOpts .= ' --base 1024'; }
+
     foreach ($graphrows as $grow) {
-      $gDefs .= ' '.$grow['dType'].':'.$grow['name'].'='.$this->rrd_file.':'.$grow['dsname'].':'.$grow['cf'];
-      // XXX: current rrdtools version doesn't support STACK flag, only STACK type
-      if (isset($grow['stack']) && $grow['stack']) { $grow['gType'] = 'STACK'; }
-      $gGraphs .= ' '.$grow['gType'].':'.$grow['name'].$grow['color'];
-      if (isset($grow['legend'])) { $gGraphs .= ':'.$this->text_quote($grow['legend']); }
-      // XXX: when STACK flag is supported, remove above STACK if-command and uncomment the one below
-      //if (isset($grow['stack']) && $grow['stack']) { $gGraphs .= ':STACK'; }
+      if (isset($grow['dType']) && strlen($grow['dType'])) {
+        $gDefs .= ' '.$grow['dType'].':'.$grow['name'].'=';
+        $gDefs .= ($grow['dType']=='DEF')?$this->rrd_file.':'.$grow['dsname'].':'.$grow['cf']:$grow['rpn_expr'];
+      }
+      if (isset($grow['gType']) && strlen($grow['gType'])) {
+        // XXX: current rrdtools version doesn't support STACK flag, only STACK type
+        if (isset($grow['stack']) && $grow['stack']) { $grow['gType'] = 'STACK'; }
+        $gGraphs .= ' '.$grow['gType'].':'.$grow['name'].$grow['color'];
+        if (isset($grow['legend'])) { $gGraphs .= ':'.$this->text_quote($grow['legend']); }
+        // XXX: when STACK flag is supported, remove above STACK if-command and uncomment the one below
+        //if (isset($grow['stack']) && $grow['stack']) { $gGraphs .= ':STACK'; }
+      }
     }
 
-    $graph_cmd = 'rrdtool graph '.$fname.' '.$timeinfo.$gDefs.$gGraphs;
+    $graph_cmd = 'rrdtool graph '.str_replace('*', '\*', $fname.$gOpts.$gDefs.$gGraphs.$addSpecial);
     $return = `$graph_cmd 2>&1`;
 
-    $retval = 0;
     if (strpos($return, 'ERROR') !== false) {
       trigger_error('rrd graph error: '.$return, E_USER_WARNING);
-      $retval = 1;
+//       $return = $graph_cmd."\n\n".$return;
     }
-  return $retval;
+  return $return;
   }
 
   function simple_html($page_extras = null, $graph_extras = null) {
     // create a simple (MRTG-like) HTML page and return it in a string
     $basename = str_replace('.rrd', '', $this->rrd_file);
+
+    // assemble configuration
+    $pconf = $this->config_page;
+    if (is_array($page_extras)) {
+      if (is_array($pconf)) { $pconf = array_merge($pconf, $page_extras); }
+      else { $pconf = $page_extras; }
+    }
+
     $out = '<html><head><title>'.$basename.' - RRD statistics</title></head><body>';
 
+    $out .= '<p>Last Update: '.date('Y-m-d H:i:s', $this->last_update()).'</p>';
+
     if (in_array($this->status, array('ok','readonly'))) {
       foreach (array('day','week','month','year') as $tframe) {
-        $this->graph(null, $tframe, $graph_extras);
+        $ret = $this->graph($tframe, null, $graph_extras);
+//         $out .= '<p>'.nl2br($ret).'</p>';
         $out .= '<p><img src="'.$basename.'-'.$tframe.'.png"></p>';
       }
     }
@@ -228,6 +394,17 @@ class rrdstat {
   return $out;
   }
 
+  function last_update() {
+    // fetch time of last update in this RRD file
+    static $last_update;
+    if (!isset($last_update) && in_array($this->status, array('ok','readonly'))) {
+      $last_cmd = 'rrdtool last '.$this->rrd_file;
+      $return = trim(`$last_cmd 2>&1`);
+      $last_update = is_numeric($return)?$return:null;
+    }
+  return isset($last_update)?$last_update:null;
+  }
+
   function text_quote($text) { return '"'.str_replace('"', '\"', str_replace(':', '\:', $text)).'"'; }
 }
 ?>