From 819835640d25eb931efbe1932921bfb76802c592 Mon Sep 17 00:00:00 2001
From: garnier-quentin <garnier.quentin@gmail.com>
Date: Thu, 16 Dec 2021 10:55:40 +0100
Subject: [PATCH] wip

---
 .../src/centreon/script/centreon_vmware.pm    |  1 +
 .../vmware/src/centreon/vmware/cmdbase.pm     |  8 +-
 .../src/centreon/vmware/cmdcpucluster.pm      | 98 +++++++++++++++++++
 .../vmware/src/centreon/vmware/cmdcpuhost.pm  | 27 +++--
 .../src/centreon/vmware/cmddatastoreusage.pm  |  4 +
 .../vmware/src/centreon/vmware/common.pm      |  1 +
 6 files changed, 125 insertions(+), 14 deletions(-)
 create mode 100644 connectors/vmware/src/centreon/vmware/cmdcpucluster.pm

diff --git a/connectors/vmware/src/centreon/script/centreon_vmware.pm b/connectors/vmware/src/centreon/script/centreon_vmware.pm
index c1063cb99..15a1599c5 100644
--- a/connectors/vmware/src/centreon/script/centreon_vmware.pm
+++ b/connectors/vmware/src/centreon/script/centreon_vmware.pm
@@ -61,6 +61,7 @@ my @load_modules = (
     'centreon::vmware::cmdalarmdatacenter',
     'centreon::vmware::cmdalarmhost',
     'centreon::vmware::cmdcountvmhost',
+    'centreon::vmware::cmdcpucluster',
     'centreon::vmware::cmdcpuhost',
     'centreon::vmware::cmdcpuvm',
     'centreon::vmware::cmddatastorecountvm',
diff --git a/connectors/vmware/src/centreon/vmware/cmdbase.pm b/connectors/vmware/src/centreon/vmware/cmdbase.pm
index 9cbf1659e..5aea8144e 100644
--- a/connectors/vmware/src/centreon/vmware/cmdbase.pm
+++ b/connectors/vmware/src/centreon/vmware/cmdbase.pm
@@ -31,7 +31,7 @@ sub new {
 
     $self->{logger} = $options{logger};
     $self->{global_case_insensitive} = defined($options{case_insensitive}) ? $options{case_insensitive} : 0;
-    
+
     return $self;
 }
 
@@ -55,7 +55,7 @@ sub class_handle_ALRM {
 
 sub handle_ALRM {
     my $self = shift;
-    
+
     $self->{logger}->writeLogError('Child process autokill!!');
     exit(0);
 }
@@ -80,12 +80,12 @@ sub set_connector {
 
 sub initArgs {
     my ($self, %options) = @_;
-    
+
     foreach (keys %{$options{arguments}}) {
         $self->{$_} = $options{arguments}->{$_};
     }
     centreon::vmware::common::init_response(identity => $options{arguments}->{identity});
-    
+
     if ($self->{global_case_insensitive} == 0 && defined($self->{case_insensitive})) {
         $self->{global_case_insensitive} = 1;
     }
diff --git a/connectors/vmware/src/centreon/vmware/cmdcpucluster.pm b/connectors/vmware/src/centreon/vmware/cmdcpucluster.pm
new file mode 100644
index 000000000..05f4e5742
--- /dev/null
+++ b/connectors/vmware/src/centreon/vmware/cmdcpucluster.pm
@@ -0,0 +1,98 @@
+# Copyright 2015 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 centreon::vmware::cmdcpucluster;
+
+use base qw(centreon::vmware::cmdbase);
+
+use strict;
+use warnings;
+use centreon::vmware::common;
+
+sub new {
+    my ($class, %options) = @_;
+    my $self = $class->SUPER::new(%options);
+    bless $self, $class;
+    
+    $self->{commandName} = 'cpucluster';
+    
+    return $self;
+}
+
+sub checkArgs {
+    my ($self, %options) = @_;
+
+    if (defined($options{arguments}->{cluster_name}) && $options{arguments}->{cluster_name} eq '') {
+        centreon::vmware::common::set_response(code => 100, short_message => 'Argument error: cluster name cannot be null');
+        return 1;
+    }
+
+    return 0;
+}
+
+sub run {
+    my $self = shift;
+
+    if (!($self->{connector}->{perfcounter_speriod} > 0)) {
+        centreon::vmware::common::set_response(code => -1, short_message => "Can't retrieve perf counters");
+        return ;
+    }
+
+    my $filters = $self->build_filter(label => 'name', search_option => 'cluster_name', is_regexp => 'filter');
+    my @properties = ('name');
+    my $views = centreon::vmware::common::search_entities(command => $self, view_type => 'ClusterComputeResource', properties => \@properties, filter => $filters);
+    return if (!defined($views));
+
+    my @instances = ('*');
+    my $values = centreon::vmware::common::generic_performance_values_historic(
+        $self->{connector},
+        $views, 
+        [
+            { label => 'cpu.usage.average',    'instances' => \@instances},
+            { label => 'cpu.usagemhz.average', 'instances' => \@instances}
+        ],
+        $self->{connector}->{perfcounter_speriod},
+        sampling_period => $self->{sampling_period},
+        time_shift => $self->{time_shift},
+        skip_undef_counter => 1, multiples => 1, multiples_result_by_entity => 1
+    );
+    return if (centreon::vmware::common::performance_errors($self->{connector}, $values) == 1);
+
+    my $interval_min = centreon::vmware::common::get_interval_min(
+        speriod => $self->{connector}->{perfcounter_speriod}, 
+        sampling_period => $self->{sampling_period},
+        time_shift => $self->{time_shift}
+    );
+
+    my $data = {};
+    foreach my $view (@$views) {
+        my $entity_value = $view->{mo_ref}->{value};
+        $data->{$entity_value} = { name => $view->{name} };
+
+        my $total_cpu_average = centreon::vmware::common::simplify_number(centreon::vmware::common::convert_number($values->{$entity_value}->{$self->{connector}->{perfcounter_cache}->{'cpu.usage.average'}->{'key'} . ":"} * 0.01));
+        my $total_cpu_mhz_average = centreon::vmware::common::simplify_number(centreon::vmware::common::convert_number($values->{$entity_value}->{$self->{connector}->{perfcounter_cache}->{'cpu.usagemhz.average'}->{'key'} . ":"}));
+
+        $data->{$entity_value}->{'interval_min'} = $interval_min;
+        $data->{$entity_value}->{'cpu.usage.average'} = $total_cpu_average;
+        $data->{$entity_value}->{'cpu.usagemhz.average'} = $total_cpu_mhz_average;
+    }
+    
+    centreon::vmware::common::set_response(data => $data);
+}
+
+1;
diff --git a/connectors/vmware/src/centreon/vmware/cmdcpuhost.pm b/connectors/vmware/src/centreon/vmware/cmdcpuhost.pm
index 31c60739f..508289e34 100644
--- a/connectors/vmware/src/centreon/vmware/cmdcpuhost.pm
+++ b/connectors/vmware/src/centreon/vmware/cmdcpuhost.pm
@@ -59,18 +59,25 @@ sub run {
     return if (!defined($result));
 
     my @instances = ('*');
-    my $values = centreon::vmware::common::generic_performance_values_historic($self->{connector},
-                            $result, 
-                            [{'label' => 'cpu.usage.average',    'instances' => \@instances},
-                             {'label' => 'cpu.usagemhz.average', 'instances' => \@instances}],
-                            $self->{connector}->{perfcounter_speriod},
-                            sampling_period => $self->{sampling_period}, time_shift => $self->{time_shift},
-                            skip_undef_counter => 1, multiples => 1, multiples_result_by_entity => 1);
+    my $values = centreon::vmware::common::generic_performance_values_historic(
+        $self->{connector},
+        $result, 
+        [
+            { label => 'cpu.usage.average',    'instances' => \@instances},
+            { label => 'cpu.usagemhz.average', 'instances' => \@instances}
+        ],
+        $self->{connector}->{perfcounter_speriod},
+        sampling_period => $self->{sampling_period},
+        time_shift => $self->{time_shift},
+        skip_undef_counter => 1, multiples => 1, multiples_result_by_entity => 1
+    );
     return if (centreon::vmware::common::performance_errors($self->{connector}, $values) == 1);
 
-    my $interval_min = centreon::vmware::common::get_interval_min(speriod => $self->{connector}->{perfcounter_speriod}, 
-                                                                  sampling_period => $self->{sampling_period}, time_shift => $self->{time_shift});
-    
+    my $interval_min = centreon::vmware::common::get_interval_min(
+        speriod => $self->{connector}->{perfcounter_speriod}, 
+        sampling_period => $self->{sampling_period}, time_shift => $self->{time_shift}
+    );
+
     my $data = {};
     foreach my $entity_view (@$result) {
         my $entity_value = $entity_view->{mo_ref}->{value};
diff --git a/connectors/vmware/src/centreon/vmware/cmddatastoreusage.pm b/connectors/vmware/src/centreon/vmware/cmddatastoreusage.pm
index 6faf56d62..1c38e06f0 100644
--- a/connectors/vmware/src/centreon/vmware/cmddatastoreusage.pm
+++ b/connectors/vmware/src/centreon/vmware/cmddatastoreusage.pm
@@ -88,6 +88,10 @@ sub run {
 
         next if (centreon::vmware::common::is_accessible(accessible => $entity_view->summary->accessible) == 0);
 
+        if (defined($self->{refresh})) {
+            $entity_view->RefreshDatastore();
+        }
+
         # capacity 0...
         if ($entity_view->summary->capacity <= 0) {
             $data->{$entity_value}->{size} = 0;
diff --git a/connectors/vmware/src/centreon/vmware/common.pm b/connectors/vmware/src/centreon/vmware/common.pm
index 0773e1d87..f67468962 100644
--- a/connectors/vmware/src/centreon/vmware/common.pm
+++ b/connectors/vmware/src/centreon/vmware/common.pm
@@ -486,6 +486,7 @@ sub cache_perf_counters {
             $obj_vmware->{perfcounter_speriod} = 20;
         }
     };
+
     if ($@) {
         $obj_vmware->{logger}->writeLogError("'" . $obj_vmware->{whoaim} . "' $@");
         return 1;