weathermap

Subversion Repositories:
[/] [lib/] [datasources/] [WeatherMapDataSource_dsstats.php] - Rev 126 Go to most recent revision
Compare with Previous - Blame - Download
<?php

include_once(dirname(__FILE__)."/../ds-common.php");

class WeatherMapDataSource_dsstats extends WeatherMapDataSource {

        function Init(&$map)
        {
                global $config;
                if($map->context=='cacti')
                {
                        if( !function_exists('db_fetch_row') )
                        {
                                debug("ReadData DSStats: Cacti database library not found. [DSSTATS001]\n");
                                return(FALSE);
                        }
                        if(function_exists("api_plugin_is_enabled"))
                        {
                                if(! api_plugin_is_enabled('dsstats'))
                                {
                                        debug("ReadData DSStats: DSStats plugin not enabled (new-style). [DSSTATS002B]\n");
                                        return(FALSE);
                                 }
                        }
                        else
                        {               
                                if( !isset($plugins) || !in_array('dsstats',$plugins))
                                {
                                        debug("ReadData DSStats: DSStats plugin not enabled (old-style). [DSSTATS002A]\n");
                                        return(FALSE);
                                }
                        }               
                                                
                        $sql = "show tables";
                        $result = db_fetch_assoc($sql) or die (mysql_error());
                        $tables = array();
                        
                        foreach($result as $index => $arr) {
                                foreach ($arr as $t) {
                                        $tables[] = $t;
                                }
                        }
                        
                        if( !in_array('data_source_stats_hourly_last', $tables) )
                        {
                                debug('ReadData DSStats: data_source_stats_hourly_last database table not found. [DSSTATS003]\n');
                                return(FALSE);
                        }                       
                                                
                        return(TRUE);
                }

                return(FALSE);
        }

# dsstats:<datatype>:<local_data_id>:<rrd_name_in>:<rrd_name_out>

        function Recognise($targetstring)
        {
                if(preg_match("/^dsstats:([a-z]+):(\d+):([\-a-zA-Z0-9_]+):([\-a-zA-Z0-9_]+)$/",$targetstring,$matches))
                {
                        return TRUE;
                }
                elseif(preg_match("/^dsstats:(\d+):([\-a-zA-Z0-9_]+):([\-a-zA-Z0-9_]+)$/",$targetstring,$matches))
                {
                        return TRUE;
                }
                else
                {
                        return FALSE;
                }
        }

        
        
        // Actually read data from a data source, and return it
        // returns a 3-part array (invalue, outvalue and datavalid time_t)
        // invalue and outvalue should be -1,-1 if there is no valid data
        // data_time is intended to allow more informed graphing in the future
        function ReadData($targetstring, &$map, &$item)
        {
                global $config;
                
                $dsnames[IN] = "traffic_in";
                $dsnames[OUT] = "traffic_out";
                $data[IN] = NULL;
                $data[OUT] = NULL;

                $inbw = NULL;
                $outbw = NULL;
                $data_time = 0;

                $table = "";
                $keyfield = "rrd_name";
                $datatype = "";
                $field = "";

                if(preg_match("/^dsstats:(\d+):([\-a-zA-Z0-9_]+):([\-a-zA-Z0-9_]+)$/",$targetstring,$matches))
                {
                        $local_data_id = $matches[1];
                        $dsnames[IN] = $matches[2];
                        $dsnames[OUT] = $matches[3];

                        $datatype = "last";

                        if($map->get_hint("dsstats_default_type") != '') {
                                        $datatype = $map->get_hint("dsstats_default_type");
                                        debug("Default datatype changed to ".$datatype.".\n");
                        }
                }elseif(preg_match("/^dsstats:([a-z]+):(\d+):([\-a-zA-Z0-9_]+):([\-a-zA-Z0-9_]+)$/",$targetstring,$matches))
                {
                        $dsnames[IN] = $matches[3];
                        $dsnames[OUT] = $matches[4];
                        $datatype = $matches[1];
                        $local_data_id = $matches[2];
                }       

                if( substr($datatype,0,5) == "daily") $table = "data_source_stats_daily";
                if( substr($datatype,0,6) == "weekly") $table = "data_source_stats_weekly";
                if( substr($datatype,0,7) == "monthly") $table = "data_source_stats_monthly";
                if( substr($datatype,0,6) == "hourly") $table = "data_source_stats_hourly";
                if( substr($datatype,0,6) == "yearly") $table = "data_source_stats_yearly";

                if( substr($datatype,-7) == "average" ) $field = "average";
                if( substr($datatype,-4) == "peak" ) $field = "peak";

                if($datatype == "last")
                {
                        $field = "calculated";
                        $table = "data_source_stats_hourly_last";
                }
                
                if($datatype == "wm")
                {
                        $field = "last_calc";
                        $table = "weathermap_data";
                        $keyfield = "data_source_name";
                }
                
                if($table != "" and $field != "")
                {
                        $SQL = sprintf("select %s as name, %s as result from %s where local_data_id=%d and (%s='%s' or %s='%s')", 
                                        $keyfield, $field, 
                                        $table, $local_data_id, $keyfield, 
                                        mysql_escape_string($dsnames[IN]), $keyfield, mysql_escape_string($dsnames[OUT])
                                );

                        $results = db_fetch_assoc($SQL);
                        if(sizeof($results)>0)
                        {
                                foreach ($results as $result)
                                {
                                        foreach ( array(IN,OUT) as $dir)
                                        {
                                                if( ($dsnames[$dir] == $result['name']) && ($result['result'] != -90909090909) && ($result['result'] !='U') )
                                                {
                                                        $data[$dir] = $result['result'];                
                                                }
                                        }
                                }
                        }
                        
                        if($datatype=='wm' && ($data[IN] == NULL || $data[OUT] == NULL) )
                        {
                                debug("Didn't get data for 'wm' source. Inserting new tasks.");
                                // insert the required details into weathermap_data, so it will be picked up next time
                                $SQL = sprintf("select data_template_data.data_source_path as path from data_template_data,data_template_rrd where data_template_data.local_data_id=data_template_rrd.local_data_id and data_template_rrd.local_data_id=%d",
                                                $local_data_id
                                                );
                                $result = db_fetch_row($SQL);
                                if(sizeof($result)>0)
                                {
                                        $db_rrdname = $result['path'];
                                        debug("Filename is $db_rrdname");
                                        foreach (array(IN,OUT) as $dir)
                                        {
                                                if($data[$dir] === NULL)
                                                {
                                                        $SQLins = "insert into weathermap_data (rrdfile, data_source_name, sequence, local_data_id) values ('" . 
                                                                mysql_real_escape_string($db_rrdname) . "','" . 
                                                                mysql_real_escape_string($dsnames[$dir]) . "', 0," . 
                                                                $local_data_id.")";
                                                        // warn($SQLins);
                                                        db_execute($SQLins);
                                                }
                                        }
                                }
                                else
                                {
                                        warn("DSStats ReadData: Failed to find a filename for DS id $local_data_id [WMDSTATS01]");
                                }
                        }
                }

                // fill all that other information (ifSpeed, etc)
                if($local_data_id>0) UpdateCactiData($item, $local_data_id);
                
                debug ("DSStats ReadData: Returning (".($data[IN]===NULL?'NULL':$data[IN]).",".($data[OUT]===NULL?'NULL':$data[OUT]).",$data_time)\n");
                
                return( array($data[IN], $data[OUT], $data_time) );
        }
}

// vim:ts=4:sw=4:
?>