From ae21740df13f9e23ac861239c651c6c304bd3b7f Mon Sep 17 00:00:00 2001 From: Quentin Garnier Date: Thu, 12 Mar 2015 14:29:48 +1100 Subject: [PATCH] + add mode for service problems --- .../centreon/mysql/mode/countnotifications.pm | 5 +- apps/centreon/mysql/mode/countproblems.pm | 180 ++++++++++++++++++ 2 files changed, 184 insertions(+), 1 deletion(-) create mode 100644 apps/centreon/mysql/mode/countproblems.pm diff --git a/apps/centreon/mysql/mode/countnotifications.pm b/apps/centreon/mysql/mode/countnotifications.pm index c93fa83fe..677a36437 100644 --- a/apps/centreon/mysql/mode/countnotifications.pm +++ b/apps/centreon/mysql/mode/countnotifications.pm @@ -78,11 +78,14 @@ sub execute { my ($self, %options) = @_; $self->{sql}->connect(); - $self->{sql}->query(query => "SELECT name, count(NULLIF(log_id, 0)) as num FROM " . $self->{option_results}->{centreon_storage_database} . ".instances LEFT JOIN " . $self->{option_results}->{centreon_storage_database} . ".logs ON logs.ctime > " . $options{time} . " AND logs.msg_type IN ('2', '3') AND logs.instance_name = instances.name WHERE deleted = '0' GROUP BY instance_name"); + $self->{sql}->query(query => "SELECT name, count(NULLIF(log_id, 0)) as num FROM " . $self->{option_results}->{centreon_storage_database} . ".instances LEFT JOIN " . $self->{option_results}->{centreon_storage_database} . ".logs ON logs.ctime > " . $options{time} . " AND logs.msg_type IN ('2', '3') AND logs.instance_name = instances.name WHERE deleted = '0' GROUP BY name"); my $total_notifications = 0; while ((my $row = $self->{sql}->fetchrow_hashref())) { $self->{output}->output_add(long_msg => sprintf("%d sent notifications from %s", $row->{num}, $row->{name})); $total_notifications += $row->{num}; + $self->{output}->perfdata_add(label => 'notifications_' . $row->{name}, + value => $row->{num}, + min => 0); } my $exit_code = $self->{perfdata}->threshold_check(value => $total_notifications, threshold => [ { label => 'critical', exit_litteral => 'critical' }, { label => 'warning', exit_litteral => 'warning' } ]); diff --git a/apps/centreon/mysql/mode/countproblems.pm b/apps/centreon/mysql/mode/countproblems.pm new file mode 100644 index 000000000..8b2aecbc0 --- /dev/null +++ b/apps/centreon/mysql/mode/countproblems.pm @@ -0,0 +1,180 @@ +################################################################################ +# Copyright 2005-2015 MERETHIS +# Centreon is developped by : Julien Mathis and Romain Le Merlus under +# GPL Licence 2.0. +# +# 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. +# +# 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 . +# +# Linking this program statically or dynamically with other modules is making a +# combined work based on this program. Thus, the terms and conditions of the GNU +# General Public License cover the whole combination. +# +# As a special exception, the copyright holders of this program give MERETHIS +# permission to link this program with independent modules to produce an executable, +# regardless of the license terms of these independent modules, and to copy and +# distribute the resulting executable under terms of MERETHIS choice, provided that +# MERETHIS also meet, for each linked independent module, the terms and conditions +# of the license of that module. An independent module is a module which is not +# derived from this program. If you modify this program, you may extend this +# exception to your version of the program, but you are not obliged to do so. If you +# do not wish to do so, delete this exception statement from your version. +# +# For more information : contact@centreon.com +# Authors : Quentin Garnier +# +#################################################################################### + +package apps::centreon::mysql::mode::countproblems; + +use base qw(centreon::plugins::mode); + +use strict; +use warnings; +use centreon::plugins::statefile; + +sub new { + my ($class, %options) = @_; + my $self = $class->SUPER::new(package => __PACKAGE__, %options); + bless $self, $class; + + $self->{version} = '1.0'; + $options{options}->add_options(arguments => + { + "warning:s" => { name => 'warning' }, + "critical:s" => { name => 'critical' }, + "centreon-storage-database:s" => { name => 'centreon_storage_database', default => 'centreon_storage' }, + }); + $self->{statefile_cache} = centreon::plugins::statefile->new(%options); + + return $self; +} + +sub check_options { + my ($self, %options) = @_; + $self->SUPER::init(%options); + + if (($self->{perfdata}->threshold_validate(label => 'warning', value => $self->{option_results}->{warning})) == 0) { + $self->{output}->add_option_msg(short_msg => "Wrong warning threshold '" . $self->{option_results}->{warning} . "'."); + $self->{output}->option_exit(); + } + if (($self->{perfdata}->threshold_validate(label => 'critical', value => $self->{option_results}->{critical})) == 0) { + $self->{output}->add_option_msg(short_msg => "Wrong critical threshold '" . $self->{option_results}->{critical}. "'."); + $self->{output}->option_exit(); + } + + $self->{statefile_cache}->check_options(%options); +} + +sub execute { + my ($self, %options) = @_; + + $self->{sql}->connect(); + $self->{sql}->query(query => "SELECT name, msg_type, status, count(NULLIF(log_id, 0)) as num FROM " . $self->{option_results}->{centreon_storage_database} . ".instances LEFT JOIN " . $self->{option_results}->{centreon_storage_database} . ".logs ON logs.ctime > " . $options{time} . " AND logs.msg_type IN ('0', '1') AND type = '1' AND status NOT IN ('0') AND logs.instance_name = instances.name WHERE deleted = '0' GROUP BY name, msg_type, status"); + + my $total_problems = { total => 0, hosts => 0, services => 0 }; + my $total_problems_by_poller = {}; + while ((my $row = $self->{sql}->fetchrow_hashref())) { + if (!defined($total_problems_by_poller->{$row->{name}})) { + $total_problems_by_poller->{$row->{name}} = { 0_1 => { label_perf => 'host_down', label => 'host down', num => 0 }, + 1_1 => { label_perf => 'service_warning', label => 'service warning', num => 0 }, + 1_2 => { label_perf => 'service_critical', label => 'service critical', num => 0 }, + 1_3 => { label_perf => 'service_unknown', label => 'service unknown', num => 0 }}; + } + + if ($row->{num} != 0 && defined($total_problems_by_poller->{$row->{name}}->{$row->{msg_type} . '_' . $row->{status}})) { + $total_problems_by_poller->{$row->{name}}->{$row->{msg_type} . '_' . $row->{status}}->{num} = $row->{num}; + if ($row->{msg_type} == 0) { + $total_problems->{hosts} += $row->{num}; + } else { + $total_problems->{services} += $row->{num}; + } + $total_problems->{total} += $row->{num}; + } + } + + $self->{output}->output_add(long_msg => sprintf("%d total hosts problems", $total_problems->{services})); + $self->{output}->output_add(long_msg => sprintf("%d total services problems", $total_problems->{services})); + foreach my $poller (sort keys %{$total_problems_by_poller}) { + foreach my $id (sort keys %{$total_problems_by_poller->{$poller}}) { + $self->{output}->output_add(long_msg => sprintf("%d %s problems on %s", + $total_problems_by_poller->{$poller}->{$id}->{num}, + $total_problems_by_poller->{$poller}->{$id}->{label}, + $poller)); + $self->{output}->perfdata_add(label => $total_problems_by_poller->{$poller}->{$id}->{label_perf} . "_" . $poller, + value => $total_problems_by_poller->{$poller}->{$id}->{num}, + min => 0); + } + } + + my $exit_code = $self->{perfdata}->threshold_check(value => $total_problems->{total}, threshold => [ { label => 'critical', exit_litteral => 'critical' }, { label => 'warning', exit_litteral => 'warning' } ]); + $self->{output}->output_add(severity => $exit_code, + short_msg => sprintf("%d total problems", $total_problems->{total})); + $self->{output}->perfdata_add(label => 'total', + value => $total_problems->{total}, + warning => $self->{perfdata}->get_perfdata_for_output(label => 'warning'), + critical => $self->{perfdata}->get_perfdata_for_output(label => 'critical'), + min => 0); + $self->{output}->perfdata_add(label => 'total_hosts', + value => $total_problems->{hosts}, + min => 0); + $self->{output}->perfdata_add(label => 'total_services', + value => $total_problems->{services}, + min => 0); +} + +sub run { + my ($self, %options) = @_; + # $options{sql} = sqlmode object + $self->{sql} = $options{sql}; + + $self->{statefile_cache}->read(statefile => 'mysql_' . $self->{mode} . '_' . $self->{sql}->get_unique_id4save()); + my $old_timestamp = $self->{statefile_cache}->get(name => 'last_timestamp'); + my $new_datas = { last_timestamp => time() }; + $self->{statefile_cache}->write(data => $new_datas); + + if (!defined($old_timestamp)) { + $self->{output}->output_add(severity => 'OK', + short_msg => "Buffer creation..."); + } else { + $self->execute(time => $old_timestamp); + } + + $self->{output}->display(); + $self->{output}->exit(); +} + +1; + +__END__ + +=head1 MODE + +Check the number of problems (works only with centreon-broker). +The mode should be used with mysql plugin and dyn-mode option. + +=over 8 + +=item B<--centreon-storage-database> + +Centreon storage database name (default: 'centreon_storage'). + +=item B<--warning> + +Threshold warning. + +=item B<--critical> + +Threshold critical. + +=back + +=cut