From feb8e81329c900784b7a0559824dbba94b874f5f Mon Sep 17 00:00:00 2001 From: itoussies <65223458+itoussies@users.noreply.github.com> Date: Wed, 15 Jun 2022 22:49:40 +0200 Subject: [PATCH] Migrate AWS SNS & SQS Service Disco to Host Disco (#3739) Co-authored-by: itoussies --- .../cloud/aws/sns/mode/discovery.pm | 104 ++++++++++++++++++ centreon-plugins/cloud/aws/sns/plugin.pm | 5 +- .../cloud/aws/sqs/mode/discovery.pm | 104 ++++++++++++++++++ centreon-plugins/cloud/aws/sqs/plugin.pm | 5 +- 4 files changed, 214 insertions(+), 4 deletions(-) create mode 100644 centreon-plugins/cloud/aws/sns/mode/discovery.pm create mode 100644 centreon-plugins/cloud/aws/sqs/mode/discovery.pm diff --git a/centreon-plugins/cloud/aws/sns/mode/discovery.pm b/centreon-plugins/cloud/aws/sns/mode/discovery.pm new file mode 100644 index 000000000..629ff559c --- /dev/null +++ b/centreon-plugins/cloud/aws/sns/mode/discovery.pm @@ -0,0 +1,104 @@ +# +# Copyright 2022 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 cloud::aws::sns::mode::discovery; + +use base qw(centreon::plugins::mode); + +use strict; +use warnings; +use JSON::XS; + +sub new { + my ($class, %options) = @_; + my $self = $class->SUPER::new(package => __PACKAGE__, %options); + bless $self, $class; + + $options{options}->add_options(arguments => { + "prettify" => { name => 'prettify' }, + }); + + return $self; +} + +sub check_options { + my ($self, %options) = @_; + $self->SUPER::init(%options); +} + +sub run { + my ($self, %options) = @_; + + my @disco_data; + my $disco_stats; + + $disco_stats->{start_time} = time(); + + my $topics = $options{custom}->discovery( + service => 'sns', + command => 'list-topics' + ); + + foreach my $topic (@{$topics->{Topics}}) { + my %topic; + $topic{name} = $topic->{TopicArn}; + $topic{name} =~ s/(.*)\://g; + push @disco_data, \%topic; + } + + $disco_stats->{end_time} = time(); + $disco_stats->{duration} = $disco_stats->{end_time} - $disco_stats->{start_time}; + $disco_stats->{discovered_items} = @disco_data; + $disco_stats->{results} = \@disco_data; + + my $encoded_data; + eval { + if (defined($self->{option_results}->{prettify})) { + $encoded_data = JSON::XS->new->utf8->pretty->encode($disco_stats); + } else { + $encoded_data = JSON::XS->new->utf8->encode($disco_stats); + } + }; + if ($@) { + $encoded_data = '{"code":"encode_error","message":"Cannot encode discovered data into JSON format"}'; + } + + $self->{output}->output_add(short_msg => $encoded_data); + $self->{output}->display(nolabel => 1, force_ignore_perfdata => 1); + $self->{output}->exit(); +} + +1; + +__END__ + +=head1 MODE + +AWS SNS discovery. + +=over 8 + +=item B<--prettify> + +Prettify JSON output. + +=back + +=cut diff --git a/centreon-plugins/cloud/aws/sns/plugin.pm b/centreon-plugins/cloud/aws/sns/plugin.pm index cd8762779..74e24d298 100644 --- a/centreon-plugins/cloud/aws/sns/plugin.pm +++ b/centreon-plugins/cloud/aws/sns/plugin.pm @@ -31,8 +31,9 @@ sub new { $self->{version} = '1.0'; %{ $self->{modes} } = ( - 'notifications' => 'cloud::aws::sns::mode::notifications', - 'list-topics' => 'cloud::aws::sns::mode::listtopics' + 'discovery' => 'cloud::aws::sns::mode::discovery', + 'list-topics' => 'cloud::aws::sns::mode::listtopics', + 'notifications' => 'cloud::aws::sns::mode::notifications' ); $self->{custom_modes}{paws} = 'cloud::aws::custom::paws'; diff --git a/centreon-plugins/cloud/aws/sqs/mode/discovery.pm b/centreon-plugins/cloud/aws/sqs/mode/discovery.pm new file mode 100644 index 000000000..8e629bc9b --- /dev/null +++ b/centreon-plugins/cloud/aws/sqs/mode/discovery.pm @@ -0,0 +1,104 @@ +# +# Copyright 2022 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 cloud::aws::sqs::mode::discovery; + +use base qw(centreon::plugins::mode); + +use strict; +use warnings; +use JSON::XS; + +sub new { + my ($class, %options) = @_; + my $self = $class->SUPER::new(package => __PACKAGE__, %options); + bless $self, $class; + + $options{options}->add_options(arguments => { + "prettify" => { name => 'prettify' }, + }); + + return $self; +} + +sub check_options { + my ($self, %options) = @_; + $self->SUPER::init(%options); +} + +sub run { + my ($self, %options) = @_; + + my @disco_data; + my $disco_stats; + + $disco_stats->{start_time} = time(); + + my $queues = $options{custom}->discovery( + service => 'sqs', + command => 'list-queues' + ); + + foreach my $queue (@{$queues->{QueueUrls}}) { + my %queue; + $queue{name} = $queue; + $queue{name} =~ s/(.*)\///g; + push @disco_data, \%queue; + } + + $disco_stats->{end_time} = time(); + $disco_stats->{duration} = $disco_stats->{end_time} - $disco_stats->{start_time}; + $disco_stats->{discovered_items} = @disco_data; + $disco_stats->{results} = \@disco_data; + + my $encoded_data; + eval { + if (defined($self->{option_results}->{prettify})) { + $encoded_data = JSON::XS->new->utf8->pretty->encode($disco_stats); + } else { + $encoded_data = JSON::XS->new->utf8->encode($disco_stats); + } + }; + if ($@) { + $encoded_data = '{"code":"encode_error","message":"Cannot encode discovered data into JSON format"}'; + } + + $self->{output}->output_add(short_msg => $encoded_data); + $self->{output}->display(nolabel => 1, force_ignore_perfdata => 1); + $self->{output}->exit(); +} + +1; + +__END__ + +=head1 MODE + +AWS SNS discovery. + +=over 8 + +=item B<--prettify> + +Prettify JSON output. + +=back + +=cut diff --git a/centreon-plugins/cloud/aws/sqs/plugin.pm b/centreon-plugins/cloud/aws/sqs/plugin.pm index ed573628f..aa77cae1b 100644 --- a/centreon-plugins/cloud/aws/sqs/plugin.pm +++ b/centreon-plugins/cloud/aws/sqs/plugin.pm @@ -31,8 +31,9 @@ sub new { $self->{version} = '1.0'; %{ $self->{modes} } = ( - 'queues' => 'cloud::aws::sqs::mode::queues', - 'list-queues' => 'cloud::aws::sqs::mode::listqueues' + 'discovery' => 'cloud::aws::sqs::mode::discovery', + 'list-queues' => 'cloud::aws::sqs::mode::listqueues', + 'queues' => 'cloud::aws::sqs::mode::queues' ); $self->{custom_modes}{paws} = 'cloud::aws::custom::paws';