Files
cacti/cli/repair_graphs.php

203 lines
7.5 KiB
PHP
Executable File

#!/usr/bin/env php
<?php
/*
+-------------------------------------------------------------------------+
| Copyright (C) 2004-2023 The Cacti Group |
| |
| 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 2 |
| 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. |
+-------------------------------------------------------------------------+
| Cacti: The Complete RRDTool-based Graphing Solution |
+-------------------------------------------------------------------------+
| This code is designed, written, and maintained by the Cacti Group. See |
| about.php and/or the AUTHORS file for specific developer information. |
+-------------------------------------------------------------------------+
| http://www.cacti.net/ |
+-------------------------------------------------------------------------+
*/
/* Original script is located here https://forums.cacti.net/viewtopic.php?t=35816, but this one was modified quite a lot */
require(__DIR__ . '/../include/cli_check.php');
require_once($config['base_path'] . '/lib/poller.php');
require_once($config['base_path'] . '/lib/utility.php');
require_once($config['base_path'] . '/lib/template.php');
/* switch to main database for cli's */
if ($config['poller_id'] > 1) {
db_switch_remote_to_main();
}
/* process calling arguments */
$parms = $_SERVER["argv"];
array_shift($parms);
$execute = false;
$show_sql = false;
unset($host_id);
unset($graph_template_id);
unset($data_template_id);
foreach($parms as $parameter) {
if (strpos($parameter, '=')) {
list($arg, $value) = explode('=', $parameter);
} else {
$arg = $parameter;
$value = '';
}
switch ($arg) {
case "--execute":
$execute = true;
break;
case "--show-sql":
$show_sql = true;
break;
case "--host-id":
$host_id = trim($value);
if (!is_numeric($host_id)) {
print "ERROR: You must supply a valid host-id to run this script!\n";
exit(1);
}
break;
case "--graph-template-id":
$graph_template_id = $value;
if (!is_numeric($graph_template_id)) {
print "ERROR: You must supply a numeric graph-template-id!\n";
exit(1);
}
break;
case "--data-template-id":
$data_template_id = $value;
if (!is_numeric($data_template_id)) {
print "ERROR: You must supply a numeric data-template-id!\n";
exit(1);
}
break;
case "--version":
case "-v":
case "-V":
display_version();
exit(0);
case "--help":
case "-h":
case "-H":
display_help();
exit(0);
default:
print "ERROR: Invalid Parameter " . $parameter . "\n\n";
display_help();
exit(1);
}
}
if (!$show_sql && !$execute) {
display_help();
exit(1);
}
if (!isset($data_template_id)) {
print "ERROR: You must supply a valid data-template-id!\n";
exit(1);
}
if (!isset($graph_template_id)) {
print "ERROR: You must supply a valid graph-template-id!\n";
exit(1);
}
if ($execute) {
print "NOTE: Repairing Graphs\n";
} else {
print "NOTE: Performing Check of Graphs\n";
}
// Get all graphs for supplied graph template
$graph = db_fetch_assoc("SELECT *
FROM graph_local
WHERE " . (!isset($host_id) ? '' : "host_id=".$host_id." AND ") . " graph_template_id=" . $graph_template_id . "");
if (cacti_sizeof($graph)) {
if (!$show_sql) {
print "\nCorrupted graphs:\n";
}
foreach($graph as $g) {
// Get datasource for supplied data template for current host
$ds = db_fetch_assoc("SELECT * FROM data_local where host_id=" . $g["host_id"] . " and data_template_id=" . $data_template_id);
if (!cacti_sizeof($ds)) {
continue;
}
$ds = $ds[0];
// Get rrd for found datasource
$rrd_data = db_fetch_assoc("SELECT * FROM data_template_rrd where local_data_id=" . $ds["id"]);
if (!cacti_sizeof($rrd_data)) {
print "Could not get correct rrd id for datasource=" . $ds["id"] . "\n";
continue;
}
/*
// Here we will find graph items that should point to our data template
// Get templated rrd id for given data template
select id from data_template_rrd where local_data_template_rrd_id=0 and local_data_id=0 and data_template_id=520
// Get templated graph->rrd association
select id from graph_templates_item where local_graph_template_item_id=0 and local_graph_id=0 and task_item_id=
// Get graph associations which corresponds to supplied data template
select id from graph_templates_item where local_graph_id=$g["id"] and local_graph_template_item_id in
// But I'm too lazy to write such a lot of code, so let's better make one long query below
*/
$graph_templates_items_wrong = db_fetch_assoc("select id, task_item_id from graph_templates_item WHERE task_item_id!=" . $rrd_data[0]["id"] . " and graph_template_id=" . $graph_template_id . " and local_graph_id=" . $g["id"] . " and local_graph_template_item_id in (select id from graph_templates_item where local_graph_template_item_id=0 and local_graph_id=0 and task_item_id=(select id from data_template_rrd where local_data_template_rrd_id=0 and local_data_id=0 and data_template_id=" . $data_template_id . "))");
if (!cacti_sizeof($graph_templates_items_wrong)) {
// Everything correct here.
continue;
} else {
foreach($graph_templates_items_wrong as $graph_templates_item_wrong) {
// Here is a list of graph_templates_item ids to be fixed and their wrong task_item_id
$graph_templates_item[] = $graph_templates_item_wrong["id"];
$task_item_id[] = $graph_templates_item_wrong["task_item_id"];
}
}
print "Host " . $g["host_id"] . ", graph " . $g["id"] . ", graph item " . implode(",",$graph_templates_item) . ", task_item_id " . implode(",",$task_item_id) . "->" . $rrd_data[0]["id"] . "\n";
$query = "UPDATE graph_templates_item SET task_item_id=" . $rrd_data[0]["id"] . " WHERE task_item_id!=" . $rrd_data[0]["id"] . " and graph_template_id=" . $graph_template_id . " and local_graph_id=" . $g["id"] . " and id in (" . implode(",",$graph_templates_item) . ")";
if ($show_sql) {
print $query . ";\n";
}
if ($execute) {
db_execute($query);
}
unset($graph_templates_item);
unset($task_item_id);
}
}
function display_version() {
$version = get_cacti_cli_version();
print "Cacti Graph Repair Tool, Version $version, " . COPYRIGHT_YEARS . PHP_EOL;
}
/* display_help - displays the usage of the function */
function display_help() {
print "usage: repair_graphs.php [--host-id=ID] --data-template-id=[ID]\n";
print " --graph-template-id=[ID] [--show-sql] [--execute]\n\n";
print "Cacti utility for repairing graph<->datasource relationship via a command line interface.\n\n";
print "--execute - Perform the repair\n";
print "--show-sql - Show SQL lines for the repair (optional)\n";
print "--host-id=id - The host_id to repair or leave empty to process all hosts\n";
print "--data-template-id=id - The numerical ID of the data template to be fixed\n";
print "--graph-template-id=id - The numerical ID of the graph template to be fixed\n";
}