147 lines
4.7 KiB
PHP
147 lines
4.7 KiB
PHP
<?php
|
|
/**
|
|
* Graphite.php
|
|
*
|
|
* -Description-
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
*
|
|
* @link https://www.librenms.org
|
|
* @copyright 2020 Tony Murray
|
|
* @copyright 2017 Falk Stern <https://github.com/fstern/>
|
|
* @author Tony Murray <murraytony@gmail.com>
|
|
*/
|
|
|
|
namespace LibreNMS\Data\Store;
|
|
|
|
use Carbon\Carbon;
|
|
use LibreNMS\Config;
|
|
use LibreNMS\Data\Measure\Measurement;
|
|
use Log;
|
|
|
|
class Graphite extends BaseDatastore
|
|
{
|
|
protected $connection;
|
|
|
|
protected $prefix;
|
|
|
|
public function __construct(\Socket\Raw\Factory $socketFactory)
|
|
{
|
|
parent::__construct();
|
|
$host = Config::get('graphite.host');
|
|
$port = Config::get('graphite.port', 2003);
|
|
try {
|
|
$this->connection = $socketFactory->createClient("$host:$port");
|
|
} catch (\Exception $e) {
|
|
d_echo($e->getMessage());
|
|
}
|
|
|
|
if ($this->connection) {
|
|
Log::notice("Graphite connection made to $host");
|
|
} else {
|
|
Log::error("Graphite connection to $host has failed!");
|
|
}
|
|
|
|
$this->prefix = Config::get('graphite.prefix', '');
|
|
}
|
|
|
|
public function getName()
|
|
{
|
|
return 'Graphite';
|
|
}
|
|
|
|
public static function isEnabled()
|
|
{
|
|
return Config::get('graphite.enable', false);
|
|
}
|
|
|
|
/**
|
|
* Datastore-independent function which should be used for all polled metrics.
|
|
*
|
|
* RRD Tags:
|
|
* rrd_def RrdDefinition
|
|
* rrd_name array|string: the rrd filename, will be processed with rrd_name()
|
|
* rrd_oldname array|string: old rrd filename to rename, will be processed with rrd_name()
|
|
* rrd_step int: rrd step, defaults to 300
|
|
*
|
|
* @param array $device
|
|
* @param string $measurement Name of this measurement
|
|
* @param array $tags tags for the data (or to control rrdtool)
|
|
* @param array|mixed $fields The data to update in an associative array, the order must be consistent with rrd_def,
|
|
* single values are allowed and will be paired with $measurement
|
|
*/
|
|
public function put($device, $measurement, $tags, $fields)
|
|
{
|
|
if (! $this->connection) {
|
|
d_echo("Graphite Error: not connected\n");
|
|
|
|
return;
|
|
}
|
|
|
|
$timestamp = Carbon::now()->timestamp;
|
|
|
|
if ($measurement == 'ports') {
|
|
$measurement = 'ports|' . $tags['ifName'];
|
|
}
|
|
|
|
// metrics will be built as prefix.hostname.measurement.field value timestamp
|
|
// metric fields can not contain . as this is used by graphite as a field separator
|
|
$hostname = preg_replace('/\./', '_', $device['hostname']);
|
|
$measurement = preg_replace(['/\./', '/\//'], '_', $measurement);
|
|
$measurement = preg_replace('/\|/', '.', $measurement);
|
|
$measurement_name = preg_replace('/\./', '_', $tags['rrd_name']);
|
|
if (is_array($measurement_name)) {
|
|
$ms_name = implode('.', $measurement_name);
|
|
} else {
|
|
$ms_name = $measurement_name;
|
|
}
|
|
// remove the port-id tags from the metric
|
|
if (preg_match('/^port-id\d+/', $ms_name)) {
|
|
$ms_name = '';
|
|
}
|
|
|
|
foreach ($fields as $k => $v) {
|
|
// Send zero for fields without values
|
|
if (empty($v)) {
|
|
$v = 0;
|
|
}
|
|
$metric = implode('.', array_filter([$this->prefix, $hostname, $measurement, $ms_name, $k]));
|
|
$this->writeData($metric, $v, $timestamp);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* @param $metric
|
|
* @param $value
|
|
* @param $timestamp
|
|
*/
|
|
private function writeData($metric, $value, $timestamp)
|
|
{
|
|
try {
|
|
$stat = Measurement::start('write');
|
|
|
|
// Further sanitize the full metric before sending, whitespace isn't allowed
|
|
$metric = preg_replace('/\s+/', '_', $metric);
|
|
|
|
$line = implode(' ', [$metric, $value, $timestamp]);
|
|
Log::debug("Sending to Graphite: $line\n");
|
|
$this->connection->write("$line\n");
|
|
|
|
$this->recordStatistic($stat->end());
|
|
} catch (\Socket\Raw\Exception $e) {
|
|
Log::error('Graphite write error: ' . $e->getMessage());
|
|
}
|
|
}
|
|
}
|