File Coverage

blib/lib/Deeme/Backend/DBI.pm

Criterion	Covered	Total	%
statement	145	158	91.7
branch	21	40	52.5
condition	4	6	66.6
subroutine	19	19	100.0
pod	1	8	12.5
total	190	231	82.2

package Deeme::Backend::DBI;

our $VERSION = '0.04';
use Deeme::Obj 'Deeme::Backend';
use DBI;
use Deeme::Utils qw(_serialize _deserialize);
use Carp 'croak';
use constant EVENT_TABLE     => "deeme_events";
use constant FUNCTIONS_TABLE => "deeme_functions";
has [qw(database _connection)];

sub new {
    my $self = shift;
    $self = $self->SUPER::new(@_);
    croak "No database string defined, database option missing"
        if ( !$self->database );
    $self->_setup;
    return $self;
}

sub _connect {
    return $_[0]->_connection(
        DBI->connect(
            ref $_[0]->database eq 'ARRAY'
            ? @{ $_[0]->database }
            : $_[0]->database
        )
    );
}

sub _disconnect {
    eval { $_[0]->_connection->commit() }
        if ( $_[0]->_connection->{AutoCommit} == 0 );
    $_[0]->_connection->disconnect;
}

sub _table_exists {
    my $self   = shift;
    my $db     = $self->_connection;
    my $table  = shift;
    my @tables = $db->tables( '', '', '', 'TABLE' );
    if (@tables) {
        for (@tables) {
            next unless $_;
            return 1 if $_ eq $table;
        }
    }
    else {
        eval {
            local $db->{PrintError} = 0;
            local $db->{RaiseError} = 1;
            $db->do(qq{SELECT * FROM $table WHERE 1 = 0 });
        };
        return 1 unless $@;
    }
    return 0;
}

sub _setup {
    my $self = shift;
    $self->_connect;
    #### Check event table, creating if not exists
    $self->_connection->do( "
        CREATE TABLE " . EVENT_TABLE . " (
          id       INTEGER PRIMARY KEY ,
          name    VARCHAR(100)
        )" ) if ( !$self->_table_exists(EVENT_TABLE) );
    #### Check functions table, creating if not exists

    $self->_connection->do( "
        CREATE TABLE " . FUNCTIONS_TABLE . " (
          id       INTEGER PRIMARY KEY,
          function TEXT,
          events_id    INT(10) NOT NULL,
          once INT(1) NOT NULL,
        foreign key (events_id) references " . EVENT_TABLE . " (id)
        )" ) if ( !$self->_table_exists(FUNCTIONS_TABLE) );
    $self->_disconnect;
}

sub events_get {
    my $self        = shift;
    my $name        = shift;
    my $deserialize = shift // 1;
    $self->_connect;
    my $sth         = $self->_connection->prepare( "
        SELECT
            f.function
        FROM
            " . FUNCTIONS_TABLE . " f
            INNER JOIN " . EVENT_TABLE . " e ON (f.events_id=e.id)
        WHERE e.name = ?
        " );
    $sth->execute($name);
    my $functions = [];
    while ( my $aref = $sth->fetchrow_arrayref ) {
        push( @{$functions}, @$aref );
    }
    $self->_disconnect;
    #deserializing subs and returning a reference
    return undef if ( @{$functions} == 0 );
    return [ map { _deserialize($_) } @{$functions} ]
        if ( $deserialize == 1 );

    return $functions;
}    #get events

sub events_reset {
    $_[0]->_connect;
    $_[0]->_connection->do( "DELETE FROM '" . EVENT_TABLE . "'" );
    $_[0]->_connection->do( "DELETE FROM '" . FUNCTIONS_TABLE . "'" );
    $_[0]->_disconnect;
}

sub events_onces {
    my $self = shift;
    my $name = shift;
    $self->_connect;
    my $sth  = $self->_connection->prepare( "
        SELECT
            f.once
        FROM
            " . FUNCTIONS_TABLE . " f
            INNER JOIN " . EVENT_TABLE . " e ON (f.events_id=e.id)
        WHERE e.name = '" . $name . "'
        " );
    $sth->execute;
    my @onces = ();
    while ( my $aref = $sth->fetchrow_arrayref ) {
        push( @onces, @$aref );
    }
    $self->_disconnect;
    #deserializing subs and returning a reference
    return @onces;
}    #get events

sub _event_name_to_id {
    return shift->_connection->selectrow_array(
        'SELECT id FROM ' . EVENT_TABLE . ' WHERE name = ? LIMIT 1',
        undef, shift );
}

sub once_update {
    my $self     = shift;
    my $name     = shift;
    my $onces    = shift;
     $self->_connect;
    my $event_id = $self->_event_name_to_id($name);
    my $sth      = $self->_connection->prepare( "
        SELECT
            id
        FROM
            " . FUNCTIONS_TABLE . "
        WHERE events_id = ?
        " );
    $sth->execute($event_id);
    my @functions_id = ();
    while ( my $aref = $sth->fetchrow_arrayref ) {
        push( @functions_id, @$aref );
    }
    $self->_connection->{AutoCommit} = 0;   # enable transactions, if possible
    $self->_connection->{RaiseError} = 1;
    my $sql = "UPDATE " . FUNCTIONS_TABLE . "
    SET once=?  WHERE id=? ";
    warn "-- Something unpredictable happened, please contact the mantainer"
        if ( scalar @{$onces} != scalar(@functions_id) );
    eval {
        $self->_connection->prepare($sql)
            ->execute( shift @{$onces}, shift @functions_id )
            unless ( @functions_id == 0 );
        $self->_connection->commit();
    };
    if ($@) {
        warn "-- Error inserting the functions in Deeme: $@\n";
        $self->_connection->rollback();
    }
    $self->_connection->{AutoCommit} = 1;    # disable transactions
    $self->_connection->{RaiseError} = 0;
         $self->_disconnect;
}

sub event_add {
    my $self = shift;
    my $name = shift;
    my $cb   = shift;
    my $once = shift // 0;
    $cb = _serialize($cb);
         $self->_connect;

    if ( my $e_id = $self->_event_name_to_id($name) ) {
        return $cb
            if $self->_connection->selectrow_array(
            'SELECT id FROM '
                . FUNCTIONS_TABLE
                . ' WHERE events_id = ? AND function=? LIMIT 1',
            undef, $e_id, $cb
            );
        $self->_connection->prepare(
                  "INSERT INTO "
                . FUNCTIONS_TABLE
                . " (events_id,function,once)
           VALUES(?,?,?)"
        )->execute( $e_id, $cb, $once );
    }
    else {
        $self->_connection->prepare(
                  "INSERT INTO "
                . EVENT_TABLE
                . " (name)
           VALUES(?)"
        )->execute($name);
        $self->_connection->prepare(
                  "INSERT INTO "
                . FUNCTIONS_TABLE
                . " (events_id,function,once)
           VALUES(?,?,?)"
            )->execute(
            $self->_connection->last_insert_id(
                undef, undef, FUNCTIONS_TABLE, undef
            ),
            $cb, $once
            );
    }
         $self->_disconnect;

    return $cb;
}

sub event_delete {
    my $self     = shift;
    my $name     = shift;
         $self->_connect;

    my $event_id = $self->_event_name_to_id($name);
    $self->_connection->{AutoCommit} = 0;   # enable transactions, if possible
    $self->_connection->{RaiseError} = 1;
    eval {
        $self->_connection->prepare(
            "DELETE
 FROM " . FUNCTIONS_TABLE . " WHERE events_id=?"
        )->execute($event_id);
        $self->_connection->prepare(
            "DELETE
 FROM " . EVENT_TABLE . " WHERE id=?"
        )->execute($event_id);
        $self->_connection->commit();
    };

    if ($@) {
        warn "-- Error inserting the functions in Deeme: $@\n";
        $self->_connection->rollback();
    }

    $self->_connection->{AutoCommit} = 1;   # enable transactions, if possible
    $self->_connection->{RaiseError} = 0;
         $self->_disconnect;

}    #delete event

sub event_update {
    my $self      = shift;
    my $name      = shift;
    my $functions = shift;
    my $serialize = shift // 1;
    return $self->event_delete($name) if ( scalar( @{$functions} ) == 0 );
     $self->_connect;

    my $event_id = $self->_event_name_to_id($name);

    $self->_connection->{AutoCommit} = 0;   # enable transactions, if possible
    $self->_connection->{RaiseError} = 1;
    eval {
        $self->_connection->prepare(
            "DELETE
 FROM " . FUNCTIONS_TABLE . " WHERE events_id=?"
        )->execute($event_id);
        my $sql
            = "INSERT INTO "
            . FUNCTIONS_TABLE
            . " (events_id,function,once)
           VALUES(?,?,?)";

        foreach my $f ( @{$functions} ) {
            if ( $serialize == 1 ) {
                $self->_connection->prepare($sql)
                    ->execute( $event_id, _serialize($f), 0 );
            }
            else {
                $self->_connection->prepare($sql)
                    ->execute( $event_id, $f, 0 );
            }

        }
        $self->_connection->commit();
    };

    if ($@) {
        warn "-- Error inserting the functions in Deeme: $@\n";
        $self->_connection->rollback();
    }

    $self->_connection->{AutoCommit} = 1;   # enable transactions, if possible
    $self->_connection->{RaiseError} = 0;
     $self->_disconnect;

}    #update event

1;
__END__