diff --git a/centreon-plugins/database/postgres/dbi.pm b/centreon-plugins/database/postgres/dbi.pm new file mode 100644 index 000000000..b09c07fe2 --- /dev/null +++ b/centreon-plugins/database/postgres/dbi.pm @@ -0,0 +1,80 @@ +# +# Copyright 2023 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 database::postgres::dbi; + +use base qw(centreon::plugins::dbi); + +use strict; +use warnings; + +sub connect { + my ($self, %options) = @_; + my $dontquit = (defined($options{dontquit}) && $options{dontquit} == 1) ? 1 : 0; + + return if (defined($self->{instance})); + + # Set ENV + if (defined($self->{env})) { + foreach (keys %{$self->{env}}) { + $ENV{$_} = $self->{env}->{$_}; + } + } + + $self->set_signal_handlers(); + my $connect_error; + eval { + $ENV{PGCONNECT_TIMEOUT} = $self->{timeout} if (defined($self->{timeout})); + $self->{instance} = DBI->connect( + "DBI:". $self->{data_source}, + $self->{username}, + $self->{password}, + { RaiseError => 0, PrintError => 0, AutoCommit => 1, %{$self->{connect_options_hash}} } + ); + }; + if ($@) { + $connect_error = $@; + } + + $self->prepare_destroy(); + + if (!defined($self->{instance})) { + my $err_msg = sprintf( + 'Cannot connect: %s', + defined($DBI::errstr) ? $DBI::errstr : + (defined($connect_error) ? $connect_error : '(no error string)') + ); + if ($dontquit == 0) { + $self->{output}->add_option_msg(short_msg => $err_msg); + $self->{output}->option_exit(exit_litteral => $self->{sql_errors_exit}); + } + return (-1, $err_msg); + } + + if (defined($self->{connect_query}) && $self->{connect_query} ne '') { + $self->query(query => $self->{connect_query}); + } + $self->set_version(); + return 0; +} + +1; + +__END__ diff --git a/centreon-plugins/database/postgres/plugin.pm b/centreon-plugins/database/postgres/plugin.pm index 9d3cbee54..152282474 100644 --- a/centreon-plugins/database/postgres/plugin.pm +++ b/centreon-plugins/database/postgres/plugin.pm @@ -47,6 +47,7 @@ sub new { 'vacuum' => 'database::postgres::mode::vacuum' }; + $self->{sql_modes}->{dbi} = 'database::postgres::dbi'; $self->{sql_modes}->{psqlcmd} = 'database::postgres::psqlcmd'; return $self; }