File Coverage

blib/lib/DBIx/Simple/DataSection.pm

Criterion	Covered	Total	%
statement	15	47	31.9
branch	0	14	0.0
condition	0	2	0.0
subroutine	5	11	45.4
pod	2	5	40.0
total	22	79	27.8

package DBIx::Simple::DataSection;
use strict;
use warnings;
use base 'DBIx::Simple';
our $VERSION = '0.02';

use Carp;
use Data::Section::Simple;

sub new {
    my $package = scalar caller;
    my $self    = shift->SUPER::new(@_);
    $self->force_utf8();
    $self->{package} = $package;
    $self->_init;
    $self;
}

sub force_utf8 {
    my $self = shift;
    return unless $self->{dbd};
    my $driver_name = $self->{dbd};
    if ( $driver_name eq 'Pg' ) {
        $self->{dbh}->{pg_enable_utf8} = 1;
    }
    elsif ( $driver_name eq 'mysql' ) {
        $self->{dbh}->{mysql_enable_utf8} = 1;
    }
    elsif ( $driver_name eq 'SQLite' ) {
        $self->{dbh}->{unicode} = 1;
    }
}

sub connect {
    my $self = shift->SUPER::connect(@_);
    $self->force_utf8();
    $self->{package} ||= scalar caller(0);
    $self->_init;
    $self;
}

sub _init {
    my $self = shift;
    $self->{section} = Data::Section::Simple->new( $self->{package} );
    $self->{cache}   = {};
}

sub query_by_sql {
    my ( $self, $sql_name, @binds ) = @_;
    my $query = $self->get_sql($sql_name);
    $self->SUPER::query( $query, @binds );
}

sub get_sql {
    my ( $self, $sql_name ) = @_;

    if ( my $sql = $self->{cache}{$sql_name} ) {
        return $sql;
    }
    my $sql = $self->{section}->get_data_section($sql_name);
    if ($sql) {
        $self->{cache}{$sql_name} = $sql if $self->{use_cache};
        return $sql;
    }
    croak "could not find sql: $sql_name in __DATA__ section";
}

1;

__END__