File Coverage

blib/lib/SQL/OOP/Dataset.pm

Criterion	Covered	Total	%
statement	73	80	91.2
branch	18	24	75.0
condition	3	9	33.3
subroutine	13	13	100.0
pod	6	7	85.7
total	113	133	84.9

package SQL::OOP::Dataset;
use strict;
use warnings;
use SQL::OOP::Base;
use SQL::OOP::ID;
use Scalar::Util qw(blessed);
use base qw(SQL::OOP::Base);

sub MODE_INSERT() {1} ## no critic
sub MODE_UPDATE() {2} ## no critic

### ---
### Constructor
### ---
sub new {
    my $class = shift @_;
    my $data_ref = (scalar @_ == 1) ? shift @_ : [@_];
    if (ref $data_ref eq 'HASH') {
        warn 'Deprecated: call the method with key-value array instead of hash';
        $data_ref = [%$data_ref];
    }
    my $self = bless {
        gen     => undef,
        array  => [],
    }, $class;

    return $self->append($data_ref);
}

### ---
### append elements
### ---
sub append {
    my $self = shift @_;
    my $data_ref = (scalar @_ == 1) ? shift @_ : [@_];
    if (ref $data_ref eq 'HASH') {
        warn 'Deprecated: call the method with key-value array instead of hash';
        $data_ref = [%$data_ref];
    }
    my @copied = @{$data_ref};
    $self->_init_gen;

    while (my($key, $val) = splice @copied, 0, 2) {
        push(@{$self->{array}}, SQL::OOP::ID->new($key)->to_string, $val);
    }

    return $self;
}

### ---
### Get binded values in array
### ---
sub bind {
    my $self = shift;
    my @copy = @{$self->{array}};
    my @vals;
    while (my ($k, $v) = splice @copy, 0, 2) {
        push(@vals, $v);
    }
    my @out = map {
        if (blessed($_)) {
            $_->bind;
        } else {
            $_;
        }
    } @vals;
    return @out if (wantarray);
    return scalar @out;
}

### ---
### Get SQL for UPDATE command in string
### ---
sub to_string_for_update {
    my ($self, $prefix) = @_;
    local $SQL::OOP::Base::quote_char = $self->quote_char;
    $self->generate(MODE_UPDATE);
    if ($self->{gen} && $prefix) {
        return $prefix. ' '. $self->{gen};
    } else {
        return $self->{gen};
    }
}

### ---
### Get SQL for INSERT command in string
### ---
sub to_string_for_insert {
    my ($self, $prefix) = @_;
    local $SQL::OOP::Base::quote_char = $self->quote_char;
    $self->generate(MODE_INSERT);
    if ($self->{gen} && $prefix) {
        return $prefix. ' '. $self->{gen};
    } else {
        return $self->{gen};
    }
}

sub generate {
    my ($self, $type) = @_;

    my @copy = @{$self->{array}};
    my @key;
    my @val;
    while (my($k, $v) = splice @copy, 0, 2) {
        push(@key, $k);
        push(@val, $v);
    }

    if ($type eq MODE_INSERT) {
        $self->{gen} = sprintf('(%s) VALUES (%s)',
            join(', ', @key),
            join(', ', map {blessed($_) ? $_->to_string : '?'} @val));
    } elsif ($type eq MODE_UPDATE) {
        $self->{gen} = '';
        for my $idx (0 .. (scalar @key) - 1) {
            $self->{gen} .= ', '. sprintf('%s = %s', $key[$idx],
                        blessed($val[$idx]) ? $val[$idx]->to_string : '?');
        }
        $self->{gen} =~ s{^, }{};
    }
    return $self;
}

sub retrieve {
    my ($self, $key) = @_;
    my %tmp = (@{$self->{array}});
    return $tmp{$key} || $tmp{$self->quote($key)};
}

1;

__END__