diff --git a/centreon/plugins/dbi.pm b/centreon/plugins/dbi.pm index f512594d9..4629d72b9 100644 --- a/centreon/plugins/dbi.pm +++ b/centreon/plugins/dbi.pm @@ -78,6 +78,9 @@ sub new { $self->{username} = undef; $self->{password} = undef; + # Sometimes, we need to set ENV + $self->{env} = undef; + return $self; } @@ -116,10 +119,11 @@ sub check_options { $self->{data_source} = (defined($self->{option_results}->{data_source})) ? shift(@{$self->{option_results}->{data_source}}) : undef; $self->{username} = (defined($self->{option_results}->{username})) ? shift(@{$self->{option_results}->{username}}) : undef; $self->{password} = (defined($self->{option_results}->{password})) ? shift(@{$self->{option_results}->{password}}) : undef; + $self->{env} = (defined($self->{option_results}->{env})) ? shift(@{$self->{option_results}->{env}}) : undef; $self->{sql_errors_exit} = $self->{option_results}->{sql_errors_exit}; if (!defined($self->{data_source}) || $self->{data_source} eq '') { - $self->{output}->add_option_msg(short_msg => "Need to specify database arguments."); + $self->{output}->add_option_msg(short_msg => "Need to specify data_source arguments."); $self->{output}->option_exit(exit_litteral => $self->{sql_errors_exit}); } @@ -160,6 +164,13 @@ sub connect { my ($self, %options) = @_; my $dontquit = (defined($options{dontquit}) && $options{dontquit} == 1) ? 1 : 0; + # Set ENV + if (defined($self->{env})) { + foreach (keys %{$self->{env}}) { + $ENV{$_} = $self->{env}->{$_}; + } + } + $self->{instance} = DBI->connect( "DBI:". $self->{data_source}, $self->{username}, @@ -243,7 +254,7 @@ dbi class =item B<--datasource> -Hostname to query (required). +Datasource (required. Depends of database server). =item B<--username> diff --git a/database/informix/mode/connectiontime.pm b/database/informix/mode/connectiontime.pm new file mode 100644 index 000000000..ddadbc9e4 --- /dev/null +++ b/database/informix/mode/connectiontime.pm @@ -0,0 +1,123 @@ +################################################################################ +# Copyright 2005-2013 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 database::informix::mode::connectiontime; + +use base qw(centreon::plugins::mode); + +use strict; +use warnings; +use Time::HiRes; +use POSIX; + +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', }, + }); + + 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->{warn1} . "'."); + $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->{critical} . "'."); + $self->{output}->option_exit(); + } +} + +sub run { + my ($self, %options) = @_; + # $options{sql} = sqlmode object + $self->{sql} = $options{sql}; + + my $now = Time::HiRes::time(); + my ($exit, $msg_error) = $self->{sql}->connect(dontquit => 1); + my $now2 = Time::HiRes::time(); + + if ($exit == -1) { + $self->{output}->output_add(severity => 'CRITICAL', + short_msg => $msg_error); + } else { + my $milliseconds = $now2 - $now; + $milliseconds = floor($milliseconds * 1000); + my $exit_code = $self->{perfdata}->threshold_check(value => $milliseconds, threshold => [ { label => 'critical', 'exit_litteral' => 'critical' }, { label => 'warning', exit_litteral => 'warning' } ]); + $self->{output}->output_add(severity => $exit_code, + short_msg => sprintf("Connection established in %.3fs.", $milliseconds / 1000)); + $self->{output}->perfdata_add(label => 'connection_time', unit => 'ms', + value => $milliseconds, + warning => $self->{perfdata}->get_perfdata_for_output(label => 'warning'), + critical => $self->{perfdata}->get_perfdata_for_output(label => 'critical'), + min => 0); + } + + $self->{output}->display(); + $self->{output}->exit(); +} + +1; + +__END__ + +=head1 MODE + +Check Informix connection time. + +=over 8 + +=item B<--warning> + +Threshold warning in milliseconds. + +=item B<--critical> + +Threshold critical in milliseconds. + +=back + +=cut diff --git a/database/informix/plugin.pm b/database/informix/plugin.pm new file mode 100644 index 000000000..cff1642bb --- /dev/null +++ b/database/informix/plugin.pm @@ -0,0 +1,122 @@ +################################################################################ +# Copyright 2005-2013 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 database::informix::plugin; + +use strict; +use warnings; +use base qw(centreon::plugins::script_sql); +use File::Temp qw(tempfile); + +sub new { + my ($class, %options) = @_; + + my $self = $class->SUPER::new(package => __PACKAGE__, %options); + bless $self, $class; + # $options->{options} = options object + + $self->{version} = '0.1'; + %{$self->{modes}} = ( + 'connection-time' => 'database::informix::mode::connectiontime', + ); + + return $self; +} + +sub init { + my ($self, %options) = @_; + + $self->{options}->add_options( + arguments => { + 'host:s@' => { name => 'db_host' }, + 'port:s@' => { name => 'db_port' }, + 'instance:s@' => { name => 'db_instance' }, + } + ); + $self->{options}->parse_options(); + my $options_result = $self->{options}->get_options(); + $self->{options}->clean(); + + if (defined($options_result->{db_host})) { + @{$self->{sqldefault}->{dbi}} = (); + for (my $i = 0; $i < scalar(@{$options_result->{db_host}}); $i++) { + if (!defined($options_result->{db_instance}[$i])) { + $self->{output}->add_option_msg(short_msg => "Need to specify instance argument."); + $self->{output}->option_exit(); + } + if (!defined($options_result->{db_port}[$i])) { + $self->{output}->add_option_msg(short_msg => "Need to specify port argument."); + $self->{output}->option_exit(); + } + + my ($handle, $path) = tempfile("sqlhosts_" . $options_result->{db_instance}[$i] . "\_XXXXX", UNLINK => 1, TMPDIR => 1); + print $handle $options_result->{db_instance}[$i] . " onsoctcp " . $options_result->{db_host}[$i] . " " . $options_result->{db_port}[$i] . "\n"; + close($handle); + $self->{sqldefault}->{dbi}[$i] = { data_source => 'Informix:sysmaster@' . $options_result->{db_instance}[$i], + env => { INFORMIXSQLHOSTS => File::Spec->rel2abs($path), + INFORMIXSERVER => $options_result->{db_instance}[$i] } }; + } + } + + $self->SUPER::init(%options); +} + +1; + +__END__ + +=head1 PLUGIN DESCRIPTION + +Check Informix Server. + +=over 8 + +You need to specify the following options. + +=item B<--host> + +Hostname to query. + +=item B<--port> + +Database Server Port. + +=item B<--instance> + +Database Instance Name. + +=back + +=cut