From 2b81a461a688a3b2670cb1aba0a6a5e458155212 Mon Sep 17 00:00:00 2001
From: qgarnier <qgarnier@centreon.com>
Date: Tue, 5 Dec 2017 10:46:06 +0100
Subject: [PATCH] working on jboss mode

---
 .../java/jboss/jmx/mode/datasourceusage.pm    | 164 ++++++++++++++++++
 .../java/jboss/jmx/mode/listdatasources.pm    | 123 +++++++++++++
 .../apps/java/jboss/jmx/plugin.pm             |  50 ++++++
 3 files changed, 337 insertions(+)
 create mode 100644 centreon-plugins/apps/java/jboss/jmx/mode/datasourceusage.pm
 create mode 100644 centreon-plugins/apps/java/jboss/jmx/mode/listdatasources.pm
 create mode 100644 centreon-plugins/apps/java/jboss/jmx/plugin.pm

diff --git a/centreon-plugins/apps/java/jboss/jmx/mode/datasourceusage.pm b/centreon-plugins/apps/java/jboss/jmx/mode/datasourceusage.pm
new file mode 100644
index 000000000..7888a1869
--- /dev/null
+++ b/centreon-plugins/apps/java/jboss/jmx/mode/datasourceusage.pm
@@ -0,0 +1,164 @@
+#
+# Copyright 2017 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::java::jboss::jmx::mode::datasourceusage;
+
+use base qw(centreon::plugins::templates::counter);
+
+use strict;
+use warnings;
+use Digest::MD5 qw(md5_hex);
+
+sub set_counters {
+    my ($self, %options) = @_;
+
+    $self->{maps_counters_type} = [
+        { name => 'datasource', type => 1, cb_prefix_output => 'prefix_ds_output', message_multiple => 'All datasources are ok' },
+    ];
+
+    $self->{maps_counters}->{datasource} = [
+        { label => 'active-con', set => {
+                key_values => [ { name => 'ActiveCount' }, { name => 'display' } ],
+                output_template => 'Current Active Connections : %s',
+                perfdatas => [
+                    { label => 'active_con', value => 'ActiveCount_absolute', template => '%s', min => 0,
+                      label_extra_instance => 1, instance_use => 'display_absolute' },
+                ],
+            }
+        },
+        { label => 'available-con', set => {
+                key_values => [ { name => 'AvailableCount' }, { name => 'display' } ],
+                output_template => 'Current Available Connections : %s',
+                perfdatas => [
+                    { label => 'available_con', value => 'AvailableCount_absolute', template => '%s', min => 0,
+                      label_extra_instance => 1, instance_use => 'display_absolute' },
+                ],
+            }
+        },
+        { label => 'in-use-con', set => {
+                key_values => [ { name => 'InUseCount' }, { name => 'display' } ],
+                output_template => 'Current In Use Connections : %s',
+                perfdatas => [
+                    { label => 'in_use_con', value => 'InUseCount_absolute', template => '%s', min => 0,
+                      label_extra_instance => 1, instance_use => 'display_absolute' },
+                ],
+            }
+        },
+        { label => 'created-con', set => {
+                key_values => [ { name => 'CreatedCount', diff => 1 }, { name => 'display' } ],
+                output_template => 'Created Connections : %s',
+                perfdatas => [
+                    { label => 'created_con', value => 'CreatedCount_absolute', template => '%s', min => 0,
+                      label_extra_instance => 1, instance_use => 'display_absolute' },
+                ],
+            }
+        },
+    ];
+}
+
+sub prefix_ds_output {
+    my ($self, %options) = @_;
+
+    return "Datasource '" . $options{instance_value}->{display} . "' ";
+}
+
+sub new {
+    my ($class, %options) = @_;
+    my $self = $class->SUPER::new(package => __PACKAGE__, %options, statefile => 1);
+    bless $self, $class;
+    
+    $self->{version} = '1.0';
+    $options{options}->add_options(arguments =>
+                                {
+                                "filter-name:s" => { name => 'filter_name' },
+                                });
+    return $self;
+}
+
+sub manage_selection {
+    my ($self, %options) = @_;
+
+    my $request = [
+        { mbean => "jboss.jca:name=*,service=ManagedConnectionPool", attributes => 
+             [ { name => 'AvailableConnectionCount' }, { name => 'ConnectionCount' }, { name => 'ConnectionCreatedCount' }, { name => 'InUseConnectionCount' } ] },
+        { mbean => "jboss.as:data-source=*,statistics=pool,subsystem=datasources", attributes => 
+             [ { name => 'AvailableCount' }, { name => 'ActiveCount' }, { name => 'CreatedCount' }, { name => 'InUseCount' } ] },
+    ];
+    
+    my $result = $options{custom}->get_attributes(request => $request, nothing_quit => 1);
+
+    $self->{datasource} = {};
+    foreach my $key (keys %$result) {
+        $key =~ /(?:[:,])(?:name|data-source)=(.*?)(?:,|$)/;
+        my $ds_name = $1;
+        $ds_name =~ s/^"(.*)"$/$1/;
+        
+        if (defined($self->{option_results}->{filter_name}) && $self->{option_results}->{filter_name} ne '' &&
+            $ds_name !~ /$self->{option_results}->{filter_name}/) {
+            $self->{output}->output_add(long_msg => "skipping  '" . $ds_name . "': no matching filter.", debug => 1);
+            next;
+        }
+     
+        $self->{datasource}->{$ds_name} = {
+            display => $ds_name,
+            AvailableCount => defined($result->{$key}->{AvailableConnectionCount}) ? $result->{$key}->{AvailableConnectionCount} : $result->{$key}->{AvailableCount},
+            ActiveCount => defined($result->{$key}->{ConnectionCount}) ? $result->{$key}->{ConnectionCount} : $result->{$key}->{ActiveCount},
+            CreatedCount => defined($result->{$key}->{ConnectionCreatedCount}) ? $result->{$key}->{ConnectionCreatedCount} : $result->{$key}->{CreatedCount},
+            InUseCount => defined($result->{$key}->{InUseConnectionCount}) ? $result->{$key}->{InUseConnectionCount} : $result->{$key}->{InUseCount},
+        };
+    }
+    
+    $self->{cache_name} = "jboss_" . $self->{mode} . '_' . md5_hex($options{custom}->get_connection_info()) . '_' .
+        (defined($self->{option_results}->{filter_counters}) ? md5_hex($self->{option_results}->{filter_counters}) : md5_hex('all')) . '_' .
+        (defined($self->{option_results}->{filter_name}) ? md5_hex($self->{option_results}->{filter_name}) : md5_hex('all'));
+}
+
+1;
+
+__END__
+
+=head1 MODE
+
+Check data sources usage.
+
+=over 8
+
+=item B<--filter-counters>
+
+Only display some counters (regexp can be used).
+Example: --filter-counters='num-active'
+
+=item B<--filter-name>
+
+Filter datasource name (can be a regexp).
+
+=item B<--warning-*>
+
+Threshold warning.
+Can be: 'active-con', 'available-con', 'created-con', 'in-use-con'.
+
+=item B<--critical-*>
+
+Threshold critical.
+Can be: 'active-con', 'available-con', 'created-con', 'in-use-con'.
+
+=back
+
+=cut
diff --git a/centreon-plugins/apps/java/jboss/jmx/mode/listdatasources.pm b/centreon-plugins/apps/java/jboss/jmx/mode/listdatasources.pm
new file mode 100644
index 000000000..74c828e0b
--- /dev/null
+++ b/centreon-plugins/apps/java/jboss/jmx/mode/listdatasources.pm
@@ -0,0 +1,123 @@
+#
+# Copyright 2017 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::java::jboss::jmx::mode::listdatasources;
+
+use base qw(centreon::plugins::mode);
+
+use strict;
+use warnings;
+
+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 =>
+                                {
+                                  "filter-name:s"   => { name => 'filter_name' },
+                                });
+    $self->{ds} = {};
+    return $self;
+}
+
+sub check_options {
+    my ($self, %options) = @_;
+    $self->SUPER::init(%options);
+}
+
+sub manage_selection {
+    my ($self, %options) = @_;
+
+    my $request = [
+        { mbean => "jboss.jca:name=*,service=ManagedConnectionPool", attributes => 
+             [ { name => 'ConnectionCount' } ] },
+        { mbean => "jboss.as:data-source=*,statistics=pool,subsystem=datasources", attributes => 
+             [ { name => 'ActiveCount' } ] },
+    ];
+    my $result = $options{custom}->get_attributes(request => $request);
+
+    foreach my $mbean (keys %{$result}) {
+        $mbean =~ /(?:[:,])(?:data-source|name)=(.*?)(?:,|$)/;
+        my $name = $1;
+        $name =~ s/^"(.*)"$/$1/;
+        
+        if (defined($self->{option_results}->{filter_name}) && $self->{option_results}->{filter_name} ne '' &&
+            $name !~ /$self->{option_results}->{filter_name}/) {
+            $self->{output}->output_add(long_msg => "skipping '" . $name . "': no matching filter.", debug => 1);
+            next;
+        }
+        
+        $self->{ds}->{$name} = { 
+            name => $name,
+        };
+    }
+}
+
+sub run {
+    my ($self, %options) = @_;
+  
+    $self->manage_selection(%options);
+    foreach my $instance (sort keys %{$self->{ds}}) { 
+        $self->{output}->output_add(long_msg => '[name = ' . $self->{ds}->{$instance}->{name} . "]");
+    }
+    
+    $self->{output}->output_add(severity => 'OK',
+                                short_msg => 'List data sources:');
+    $self->{output}->display(nolabel => 1, force_ignore_perfdata => 1, force_long_output => 1);
+    $self->{output}->exit();
+}
+
+sub disco_format {
+    my ($self, %options) = @_;
+    
+    $self->{output}->add_disco_format(elements => ['name']);
+}
+
+sub disco_show {
+    my ($self, %options) = @_;
+
+    $self->manage_selection(%options);
+    foreach my $instance (sort keys %{$self->{ds}}) {             
+        $self->{output}->add_disco_entry(
+            %{$self->{ds}->{$instance}}
+        );
+    }
+}
+
+1;
+
+__END__
+
+=head1 MODE
+
+List data sources.
+
+=over 8
+
+=item B<--filter-name>
+
+Filter by name (can be a regexp).
+
+=back
+
+=cut
+    
diff --git a/centreon-plugins/apps/java/jboss/jmx/plugin.pm b/centreon-plugins/apps/java/jboss/jmx/plugin.pm
new file mode 100644
index 000000000..378cd1951
--- /dev/null
+++ b/centreon-plugins/apps/java/jboss/jmx/plugin.pm
@@ -0,0 +1,50 @@
+#
+# Copyright 2017 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::java::jboss::jmx::plugin;
+
+use strict;
+use warnings;
+use base qw(centreon::plugins::script_custom);
+
+sub new {
+    my ($class, %options) = @_;
+    my $self = $class->SUPER::new(package => __PACKAGE__, %options);
+    bless $self, $class;
+
+    $self->{version} = '0.1';
+    %{$self->{modes}} = (
+                          'datasource-usage' => 'apps::java::jboss::jmx::mode::datasourceusage',
+                          'list-datasources' => 'apps::java::jboss::jmx::mode::listdatasources',
+                         );
+
+    $self->{custom_modes}{jolokia} = 'centreon::common::protocols::jmx::custom::jolokia';
+    return $self;
+}
+
+1;
+
+__END__
+
+=head1 PLUGIN DESCRIPTION
+
+Check Jboss in JMX. Need Jolokia agent.
+
+=cut