centreon-plugins/apps/centreon/local/mode/notsodummy.pm

281 lines
10 KiB
Perl

#
# Copyright 2021 Centreon (http://www.centreon.com/)
#
# Centreon is a full-fledged industry-strength solution that meets
# the needs in IT infrastructure and application monitoring for
# service performance.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
package apps::centreon::local::mode::notsodummy;
use base qw(centreon::plugins::mode);
use strict;
use warnings;
use centreon::plugins::statefile;
use Digest::MD5 qw(md5_hex);
my %errors_service = (0 => 'OK', 1 => 'WARNING', 2 => 'CRITICAL', 3 => 'UNKNOWN');
my %errors_host = (0 => 'UP', 1 => 'DOWN');
my %errors_hash = ('UP' => 'OK', 'DOWN' => 'CRITICAL');
sub new {
my ($class, %options) = @_;
my $self = $class->SUPER::new(package => __PACKAGE__, %options, force_new_perfdata => 1);
bless $self, $class;
$options{options}->add_options(arguments => {
"status-sequence:s" => { name => 'status_sequence' },
"host" => { name => 'host' },
"output:s" => { name => 'output' },
"metrics-count:s" => { name => 'metrics_count' },
"metrics-name:s" => { name => 'metrics_name', default => 'metrics.number' },
"metrics-values-range:s" => { name => 'metrics_values_range' },
"show-sequence" => { name => 'show_sequence' },
"show-index" => { name => 'show_index' },
"restart-sequence" => { name => 'restart_sequence' },
});
$self->{cache} = centreon::plugins::statefile->new(%options);
return $self;
}
sub check_options {
my ($self, %options) = @_;
$self->SUPER::init(%options);
if (!defined($self->{option_results}->{status_sequence}) || $self->{option_results}->{status_sequence} eq '') {
$self->{output}->add_option_msg(short_msg => "Need to specify --status-sequence option.");
$self->{output}->option_exit();
}
if (!defined($self->{option_results}->{output}) || $self->{option_results}->{output} eq '') {
$self->{output}->add_option_msg(short_msg => "Need to specify --output option.");
$self->{output}->option_exit();
}
foreach my $status (split(',', $self->{option_results}->{status_sequence})) {
if (!defined($self->{option_results}->{host}) && $status !~ /^[0-3]$/ && $status !~ /ok|warning|critical|unknown/i) {
$self->{output}->add_option_msg(short_msg => "Status should be in '0,1,2,3' or 'ok,warning,critical,unknown' (case isensitive).");
$self->{output}->option_exit();
}
if (defined($self->{option_results}->{host}) && $status !~ /^[0-1]$/ && $status !~ /up|down/i) {
$self->{output}->add_option_msg(short_msg => "Status should be in '0,1' or 'up,down' (case isensitive).");
$self->{output}->option_exit();
}
push @{$self->{status_sequence}}, $status;
}
if (defined($self->{option_results}->{metrics_count}) && $self->{option_results}->{metrics_count} < 1) {
$self->{output}->add_option_msg(short_msg => "Need to specify --metrics-count value > 0.");
$self->{output}->option_exit();
}
if (defined($self->{option_results}->{metrics_values_range}) && $self->{option_results}->{metrics_values_range} =~ /(([0-9-.,]*):)?([0-9-.,]*)/) {
$self->{metrics_range_start} = $2;
$self->{metrics_range_end} = $3;
}
$self->{metrics_range_start} = 0 if (!defined($self->{metrics_range_start}) || $self->{metrics_range_start} eq '');
$self->{metrics_range_end} = 100 if (!defined($self->{metrics_range_end}) || $self->{metrics_range_end} eq '');
if ($self->{metrics_range_start} !~ /^-?\d+$/ || $self->{metrics_range_end} !~ /^-?\d+$/) {
$self->{output}->add_option_msg(short_msg => "Need to specify --metrics-values-range where range start and range end are integer.");
$self->{output}->option_exit();
}
if ($self->{metrics_range_start} > $self->{metrics_range_end}) {
$self->{output}->add_option_msg(short_msg => "Need to specify --metrics-values-range where range start is lower than range end.");
$self->{output}->option_exit();
}
$self->{cache}->check_options(option_results => $self->{option_results});
}
sub get_next_status {
my ($self, %options) = @_;
my $index;
my $has_cache_file = $options{statefile}->read(statefile => 'centreon_notsodummy_' .
md5_hex(@{$self->{status_sequence}}) . '_' . md5_hex($self->{option_results}->{output}));
if ($has_cache_file == 0 || $self->{option_results}->{restart_sequence}) {
$index = 0;
my $datas = {
last_timestamp => time(),
status_sequence => $self->{status_sequence},
status_sequence_index => $index
};
$options{statefile}->write(data => $datas);
} else {
$index = $options{statefile}->get(name => 'status_sequence_index');
$index = ($index < scalar(@{$self->{status_sequence}} - 1)) ? $index + 1 : 0;
my $datas = {
last_timestamp => time(),
status_sequence => $self->{status_sequence},
status_sequence_index => $index
};
$options{statefile}->write(data => $datas);
}
return $self->{status_sequence}[$index], $index;
}
sub get_sequence_output {
my ($self, %options) = @_;
my @sequence_output;
my $i = 0;
foreach my $status (split(',', $self->{option_results}->{status_sequence})) {
$status = $errors_service{$status} if ($status =~ /^[0-3]$/ && !defined($self->{option_results}->{host}));
$status = $errors_host{$status} if ($status =~ /^[0-1]$/ && defined($self->{option_results}->{host}));
push @sequence_output, uc($status) if ($i == $options{index});
push @sequence_output, lc($status) if ($i != $options{index});
$i++
}
return join(',', @sequence_output);
}
sub run {
my ($self, %options) = @_;
my ($status, $index) = $self->get_next_status(statefile => $self->{cache});
my $status_label = $status;
if (defined($self->{option_results}->{host})) {
$status_label = $errors_host{$status} if ($status =~ /^[0-1]$/);
$status = $errors_host{$status} if ($status =~ /^[0-1]$/);
$status = $errors_hash{uc($status)};
} else {
$status_label = $errors_service{$status} if ($status =~ /^[0-3]$/);
$status = $errors_service{$status} if ($status =~ /^[0-3]$/);
}
my $output = $self->{option_results}->{output};
$output .= ' [' . $self->get_sequence_output(index => $index) . ']' if ($self->{option_results}->{show_sequence});
$self->{output}->output_add(
severity => $status,
short_msg => uc($status_label) . ': ' . $output
);
$self->{output}->output_add(
long_msg => "Current status '" . uc($status_label) . "'"
);
$self->{output}->output_add(
long_msg => "Sequence '" . $self->get_sequence_output(index => $index) . "'"
);
if (defined($self->{option_results}->{metrics_count}) > 0) {
for (my $i = 1; $i <= $self->{option_results}->{metrics_count}; $i++) {
my $metric_name = $self->{option_results}->{metrics_name} . '.' . $i;
my $metric_value = $self->{metrics_range_start} + int(rand($self->{metrics_range_end} - $self->{metrics_range_start})) + 1;
$self->{output}->perfdata_add(
nlabel => $metric_name,
value => $metric_value,
min => $self->{metrics_range_start},
max => $self->{metrics_range_end}
);
$self->{output}->output_add(
long_msg => "Metric '" . $metric_name . "' value is '" . $metric_value . "'"
);
}
}
if (defined($self->{option_results}->{show_index})) {
$self->{output}->perfdata_add(
nlabel => 'sequence.index.position',
value => ++$index,
min => 1,
max => scalar(@{$self->{status_sequence}})
);
}
$self->{output}->display(nolabel => 1);
$self->{output}->exit();
}
1;
__END__
=head1 MODE
Do a not-so-dummy check.
Sequence and sequence index are stored in cache file. Use --statefile* options
to defined the way cache file is managed.
Examples:
perl centreon_plugin.pl --plugin=apps::centreon::local::plugin
--mode=not-so-dummy --status-sequence='ok,warning,ok,critical,critical,critical'
--output='Not so dummy service' --show-sequence --statefile-dir='/tmp'
perl centreon_plugin.pl --plugin=apps::centreon::local::plugin
--mode=not-so-dummy --status-sequence='up,down,down' --host
--output='Not so dummy host'
perl centreon_plugin.pl --plugin=apps::centreon::local::plugin
--mode=not-so-dummy --status-sequence='ok,ok,ok' --output='Not so dummy'
--metrics-count=5 --metrics-name='met.rics' --metrics-values-range='-15:42'
=over 8
=item B<--status-sequence>
Comma separated sequence of statuses from which the mode should pick is
return code from.
(Example: --status-sequence='ok,critical,ok,ok' or --status-sequence='up,up,down' --host)
(Should be numeric value between 0 and 3, or string in ok, warning, critical, unknown, up, down).
=item B<--host>
To be set if sequence is for host statuses.
=item B<--output>
Output to be returned.
=item B<--metrics-count>
Number of metrics to generate.
=item B<--metrics-name>
Name of the metrics (Default: 'metrics.number').
Metrics are suffixed by a number between 1 and metrics count.
=item B<--metrics-values-range>
Range of values from which metrics values can be picked (Default: '0:100').
=item B<--show-sequence>
Show the sequence is the output (in addition to the defined output).
=item B<--show-index>
Show the index as a metric (in addition to the defined metrics count).
=item B<--restart-sequence>
Restart the sequence from the beginning (ie. reset the sequence in cache file).
=back
=cut