centreon-plugins/database/db2/dbi.pm

#
# Copyright 2021 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::db2::dbi;

use base qw(centreon::plugins::dbi);

use strict;
use warnings;

sub connect {
    my ($self, %options) = @_;

    if ($self->{data_source} =~ /PROTOCOL=TCPIP/) {
        $self->{data_source} .= 'UID=' . $self->{username} . ';'
            if ($self->{data_source} !~ /UID/ && defined($self->{username}));
        $self->{data_source} .= 'PWD=' . $self->{password} . ';'
            if ($self->{data_source} !~ /PWD/ && defined($self->{password}));
    }
    $self->SUPER::connect(%options);
}

sub get_dbh {
    my ($self, %options) = @_;

    return $self->{instance};
}

sub get_database_name {
    my ($self, %options) = @_;

    my $dbname = 'unknown';
    $dbname = $1 if ($self->{data_source} =~ /DATABASE=([^;]*?)/i);
    $dbname = $1 if ($self->{data_source} =~ /DBI:DB2:(.*)/);
    $dbname =~ s/^\s+//g;
    $dbname =~ s/\s+$//g;
    return $dbname;
}

1;

__END__