File Coverage

blib/lib/Mojolicious/Plugin/AnyData.pm

Criterion	Covered	Total	%
statement	10	47	21.2
branch	0	18	0.0
condition	0	20	0.0
subroutine	4	11	36.3
pod	3	5	60.0
total	17	101	16.8

################################################################################
#
#  Mojolicious::Plugin::AnyData 1.20
#
#    uses perl data in the memory like a database source.
#    author: Alexander Ponomarev
#
################################################################################

package Mojolicious::Plugin::AnyData;

BEGIN {
    $Mojolicious::Plugin::AnyData::VERSION = '1.20';
}

use Mojo::Base 'Mojolicious::Plugin';
use DBI;
use v5.10;

has 'dbh';
has 'app';

# Run on startup
sub register {
    my ($self, $app, $param) = @_;

    return if $app->mode && $app->mode eq 'production';

    my ($data, $data_file);
    my $func   = $param->{func};
    my $helper = $param->{helper};
    $helper ||= 'db';
    my $dbh  = DBI->connect('dbi:AnyData:(RaiseError=>1)');

    $self->dbh($dbh);
    $self->app($app);

    if ( $param->{load_data} ) {
	$self->load_data( $param->{load_data} );
    }

    if ( $func && ref $func eq 'ARRAY' && scalar @$func > 0 ) {
	#$dbh->func(@$func);
	$self->func(@$func);
    }

    $app->helper( $param->{helper} => sub { return $dbh } );
    $app->helper( any_data => sub { return $self } );
}

# Load data into the memory using different ways
# in: arrayref (data structure) or scalar (filename)
sub load_data {
    my ($self, $data) = @_;

    return unless $self->dbh && $self->app;

    if ( ref $data ) {
    	$self->ad_import($data);
    }
    else {
    	my $data_file = $data;
    	$data = $self->app->plugin(config => {
    	    file => $data_file,
    	    stash_key => 'any_data',
    	});
    	$self->ad_import($data);
    }
}

# Provide the wrapper method for DBD::AnyData::func method
sub func {
    my ($self, $table_name, $table_type, $table_data, $table_method) = @_;

    return unless $self->dbh;

    $self->dbh->func($table_name, 'ad_clear');
    $self->dbh->func( $table_name, $table_type, $table_data, $table_method );
}

# Execute DBD::AnyData::func method in order to load data into the memory
sub ad_import {
    my ($self, $data) = @_;

    return unless $self->dbh;

    if ( $data && ref $data eq 'HASH' && keys %$data > 0 ) {
    	TABLE:
	for my $table_name ( keys %$data ) {
	    next TABLE unless ref $data->{$table_name} eq 'ARRAY';

	    $self->dbh->func($table_name, 'ad_clear');
    	    $self->dbh->func($table_name, 'ARRAY', $data->{$table_name}, 'ad_import');
        }
    }
}

sub version {
    my ($self) = @_;

    return $Mojolicious::Plugin::AnyData::VERSION;
}

1;
__END__